from oxi.interfaces import register_parser
from oxi.interfaces.base import BaseDevice


@register_parser(["vrp", "huawei"])
class Huawei(BaseDevice):
    template = "huawei.ttp"

    def vlans(self):
        vlan_ids = self.raw["vlans"].get("vlan_ids")
        return [{"vlan_id": vlan} for vlan in vlan_ids]


if __name__ == "__main__":
    with open("./test4.txt") as file:
        data = file.read()
    huawei = Huawei(data)
    print(huawei.parse())