device_selector.py

# Python library import
from netscud.devices.alcatel.alcatel_aos import AlcatelAOS
from netscud.devices.cisco.cisco_ios import CiscoIOS
from netscud.devices.cisco.cisco_s300 import CiscoS300
from netscud.devices.mikrotik.mikrotik_routeros import MikrotikRouterOS

#  The supported device_types are the keys of this dictionary
ALL_DEVICE_TYPE_CLASS = {
    "alcatel_aos": AlcatelAOS,
    "cisco_ios": CiscoIOS,
    "cisco_s300": CiscoS300,
    "mikrotik_routeros": MikrotikRouterOS,
}


def ConnectDevice(**kwargs):
    """
    Async function to be used for connnecting a device

    This async function allows to use a class with an async method.
    This function includes a connection to a device.

    :param kwargs: dictionary with the device connection parameters
    :type kwargs: dict

    :return: Return the object with the
    :rtype: object with the device class
    """

    # Device type provided in the device dictionary?
    if "device_type" in kwargs:

        # Yes

        # Get device type
        device_type = kwargs["device_type"]
    else:

        # No

        # Display error message
        raise Exception("ConnectDevice: no device type specified")

    # Check if device type is amongst the supported ones
    if device_type not in ALL_DEVICE_TYPE_CLASS:

        # Not a supported device

        # Display error message
        raise Exception(f"ConnectDevice: device type unknown: {device_type}")

    # Create a class
    my_class = ALL_DEVICE_TYPE_CLASS[device_type](**kwargs)

    # Return the class
    return my_class