"""
Base Switch Object Definition
Authors: james.luhrsen@hp.com
Created: 2014-09-20
"""
import importlib
import xml.etree.ElementTree as ElementTree


class BaseSwitch(object):
    """
    Switch Base Class
    """

    make = ""
    model = ""

    mgmt_protocol = ""
    ssh_key = ""
    mgmt_ip = ""
    mgmt_port = ""
    mgmt_user = ""
    mgmt_password = ""
    mgmt_prompt = ""

    connection_index = ""

    initialization_type = ""

    of_controller_ip = ""

    connection_configs = []

    initialization_cmds = []

    base_openflow_config = []

    openflow_enable_config = []

    openflow_enable_validations = []

    openflow_disable_config = []
    openflow_disable_validations = []

    dump_all_flows = []

    src_mac = ""
    dst_mac = ""
    ip_src = ""
    ip_dst = ""
    table_id = ""
    action = ""

    datapath_id_output_string = ""
    datapath_id_output_command = ""
    datapath_id = ""

    def set_connection_index(self, idx):
        self.connection_index = idx

    def set_controller_ip(self, ip):
        self.of_controller_ip = ip

    def set_mgmt_ip(self, ip):
        self.mgmt_ip = ip

    def set_mgmt_user(self, user):
        self.mgmt_user = user

    def set_mgmt_prompt(self, prompt):
        self.mgmt_prompt = prompt

    def set_ssh_key(self, key):
        self.ssh_key = key

    def update_datapath_id(self):
        raise NotImplementedError("Please implement this method")

    def create_flow_match_elements(self, flow_xml):
        flow_tree = ElementTree.fromstring(flow_xml)
        self.table_id = flow_tree.find("{urn:opendaylight:flow:inventory}table_id").text
        instructions_element = flow_tree.find(
            "{urn:opendaylight:flow:inventory}instructions"
        )
        instruction_element = instructions_element.find(
            "{urn:opendaylight:flow:inventory}instruction"
        )
        apply_actions = instruction_element.find(
            "{urn:opendaylight:flow:inventory}apply-actions"
        )
        action = apply_actions.find("{urn:opendaylight:flow:inventory}action")
        output_action = action.find("{urn:opendaylight:flow:inventory}output-action")
        output_node_connector = output_action.find(
            "{urn:opendaylight:" "flow:inventory}output-node-connector"
        )
        self.action = output_node_connector.text
        match_element = flow_tree.find("{urn:opendaylight:flow:inventory}match")
        ethernet_match_element = match_element.find(
            "{urn:opendaylight:flow:inventory}ethernet-match"
        )
        ethernet_source = ethernet_match_element.find(
            "{urn:opendaylight:flow:inventory}ethernet-source"
        )
        ethernet_source_address = ethernet_source.find(
            "{urn:opendaylight:flow:inventory}address"
        )
        self.src_mac = ethernet_source_address.text
        ethernet_destination = ethernet_match_element.find(
            "{urn:opendaylight:flow:inventory}ethernet-destination"
        )
        ethernet_destination_address = ethernet_destination.find(
            "{urn:opendaylight:flow:inventory}address"
        )
        self.dst_mac = ethernet_destination_address.text
        self.ip_src = match_element.find(
            "{urn:opendaylight:flow:inventory}ipv4-source"
        ).text
        self.ip_dst = match_element.find(
            "{urn:opendaylight:flow:inventory}ipv4-destination"
        ).text

    def convert_hex_to_decimal_as_string(self, hex_string):
        # TODO: need to add error checking in case the hex_string is
        # not fully hex
        return str(int(hex_string, 16))

    def get_switch(self, switch_type):
        """
        Generic method that will allow Robot Code to pass a string
        to this "keyword - Get Switch" and create an object of that
        type.  (EX: Get Switch  OVS)
        """

        # TODO: what if the module "switch_type" does not exist.  Need some
        # error checking for that.
        module = importlib.import_module(switch_type)
        return getattr(module, switch_type)()