|
14 | 14 | # @formatter:off |
15 | 15 | # The keys of this dictionary are the supported device_types |
16 | 16 | CLASS_MAPPER = { |
17 | | - 'arista_eos': AristaEOS, |
18 | | - 'aruba_aos_6': ArubaAOS6, |
19 | | - 'aruba_aos_8': ArubaAOS8, |
20 | | - 'cisco_asa': CiscoASA, |
21 | | - 'cisco_ios': CiscoIOS, |
22 | | - 'cisco_ios_xe': CiscoIOS, |
23 | | - 'cisco_ios_xr': CiscoIOSXR, |
24 | | - 'cisco_nxos': CiscoNXOS, |
25 | | - 'fujitsu_switch': FujitsuSwitch, |
26 | | - 'hp_comware': HPComware, |
27 | | - 'hp_comware_limited': HPComwareLimited, |
28 | | - 'juniper_junos': JuniperJunOS, |
29 | | - 'mikrotik_routeros': MikrotikRouterOS, |
30 | | - 'ubiquity_edge': UbiquityEdgeSwitch, |
31 | | - 'terminal': Terminal, |
| 17 | + "arista_eos": AristaEOS, |
| 18 | + "aruba_aos_6": ArubaAOS6, |
| 19 | + "aruba_aos_8": ArubaAOS8, |
| 20 | + "cisco_asa": CiscoASA, |
| 21 | + "cisco_ios": CiscoIOS, |
| 22 | + "cisco_ios_xe": CiscoIOS, |
| 23 | + "cisco_ios_xr": CiscoIOSXR, |
| 24 | + "cisco_nxos": CiscoNXOS, |
| 25 | + "fujitsu_switch": FujitsuSwitch, |
| 26 | + "hp_comware": HPComware, |
| 27 | + "hp_comware_limited": HPComwareLimited, |
| 28 | + "juniper_junos": JuniperJunOS, |
| 29 | + "mikrotik_routeros": MikrotikRouterOS, |
| 30 | + "ubiquity_edge": UbiquityEdgeSwitch, |
| 31 | + "terminal": Terminal, |
32 | 32 | } |
33 | 33 |
|
34 | 34 | # @formatter:on |
|
40 | 40 |
|
41 | 41 | def create(*args, **kwargs): |
42 | 42 | """Factory function selects the proper class and creates object based on device_type""" |
43 | | - if kwargs['device_type'] not in platforms: |
44 | | - raise ValueError('Unsupported device_type: ' |
45 | | - 'currently supported platforms are: {0}'.format(platforms_str)) |
46 | | - connection_class = CLASS_MAPPER[kwargs['device_type']] |
| 43 | + if kwargs["device_type"] not in platforms: |
| 44 | + raise ValueError( |
| 45 | + "Unsupported device_type: " |
| 46 | + "currently supported platforms are: {0}".format(platforms_str) |
| 47 | + ) |
| 48 | + connection_class = CLASS_MAPPER[kwargs["device_type"]] |
47 | 49 | return connection_class(*args, **kwargs) |
0 commit comments