from Flowdat.Catv import Catv
from ModelOltBase import ModelOltBase
from Util import Strings


class ZTEv2(object, ModelOltBase):
    file_name = None
    conf_file_parse = None

    def __init__(self, file_name):
        ModelOltBase.__init__(self, file_name)

    def get_pon_onu_mng(self, olt):
        """
        Complete
        :param olt: Olt object
        :return: Return object with child
        """
        for obj_olt in olt.child:
            for obj_onu in obj_olt.child:
                # paso el nombre a minuscula
                search = obj_olt.name.lower()
                # reemplazo el valor olt por onu
                search = "pon-onu-mng " + search.replace("olt", "onu") + ":" + str(obj_onu.id)
                obj = self.search_object(search)
                if obj is not None and obj.text is not None:
                    # recorro el objeto para obtener los datos
                    for obj in obj.children:
                        val = obj.text.lower().strip()
                        index = Strings.index_value(val, "video")
                        if index >= 0:
                            obj_onu.catv = Catv()
                            obj_onu.catv.name = obj.text

        return olt

    def process_file(self, olt):
        super(ZTEv2, self).process_file(olt)
        self.get_pon_onu_mng(olt)