from model.example import Example
from model.translation import Translation
from model.editable import Editable
from model.tags import TAGS


class Sense(Editable):
    def __init__(self, sense_xml):
        definition = sense_xml.querySelector("definitionList definition")
        
        self.labels = []
        for tag_xml in sense_xml.querySelectorAll("sense > labelList label"):
            t_type = tag_xml.getAttribute("type")
            t_value = tag_xml.textContent
            
            if t_type not in TAGS:
                # using some default
                t_type = TAGS.keys()[0]
                
            self.labels.append((t_type, t_value))
        
        self.definition = definition.textContent if definition else ""
        
        self.examples = [Example(example_xml) for example_xml in
                         sense_xml.querySelectorAll("exampleContainerList exampleContainer")]

        translations = []
        max_num_cluster = 0
        
        for translation_xml in sense_xml.querySelectorAll("translationContainerList translationContainer"):
            num_cluster = int(translation_xml.getAttribute("cluster"))
            max_num_cluster = max(max_num_cluster, num_cluster)
            translations.append((num_cluster, Translation(translation_xml)))
        
        self.translations = [[] for _ in range(max_num_cluster)]
        for clusterNum, translation in translations:
            self.translations[clusterNum - 1].append(translation)
    
    
    def merge_labels(self):
        return ", ".join(val for _, val in self.labels)