2 # -*- coding: utf-8 -*-
10 from linkpair.device import Device
11 from linkpair.port import Port
12 from linkpair.linkpair import LinkPair
13 from linkpair.formatter.grapheasy import GraphEasyFormatter
14 from linkpair.commonutils import CommonUtils
15 from linkpair.dbutils import DBUtils
16 from linkpair.collect.utils import CollectUtils
17 from linkpair.collect.collector_dataset import CollectorDataset
18 from linkpair.collect.agent.commandrunner import CommandRunner
19 from linkpair.collect.agent.linux_agent import LinuxAgent
20 from linkpair.collect.agent.namespace_agent import NamespaceAgent
21 from linkpair.collect.agent.ovs_agent import OVSAgent
22 from linkpair.collect.agent.linuxbridge_agent import LinuxBridgeAgent
23 from linkpair.collect.agent.libvirt_agent import LibvirtAgent
26 class Collector(object):
27 '''LinkpPair collector
29 This class gets the LinkPair information from LinuxBridge and Open vSwitch and libvirt
34 def __init__(self, remote_desc, dbu, formatter=GraphEasyFormatter()):
38 self._iface_to_nss = {}
39 self._u = CommonUtils()
41 self._db_enable = False
43 self._remote_desc = remote_desc
44 self._ssh_username = ""
45 self._ssh_hostname = ""
46 self._ssh_hostport = 22
47 self._remote_password = ""
48 self._remote_sshkey = ""
49 self._ssh_keyauth = False
50 self._ssh_passauth = False
52 ''' set parameters '''
53 [self._ssh_username, self._ssh_hostname,
54 self._ssh_hostport] = self._u.parse_remote_desc(remote_desc)
56 self._formatter = formatter
57 self._runner = CommandRunner(self._remote_desc)
60 self._cu = CollectUtils(
61 self._linkpairs, self._port_to_br, self._iface_to_nss, self._dbu, self._formatter)
62 cd = CollectorDataset(
63 self._os_info, self._linkpairs, self._port_to_br, self._iface_to_nss)
64 os_agent = LinuxAgent(
65 self._runner, self._cu, self._remote_desc, cd, self._formatter)
67 namespace_agent = NamespaceAgent(
68 self._runner, self._cu, self._remote_desc, cd, self._formatter)
71 self._runner, self._cu, self._remote_desc, cd, self._formatter)
73 bridge_agent = LinuxBridgeAgent(
74 self._runner, self._cu, self._remote_desc, cd, self._formatter)
76 libvirt_agent = LibvirtAgent(
77 self._runner, self._cu, self._remote_desc, cd, self._formatter)
80 def get_linkpairs(self):
81 return self._linkpairs
83 def set_remote_sshkey(self, remote_sshkey):
84 self._runner.set_remote_sshkey(remote_sshkey)
86 def set_remote_password(self, remote_password):
87 self._runner.set_remote_password(remote_password)