summaryrefslogtreecommitdiff
path: root/haircontrol/data.py
blob: 599d322cda584e040d8f4f1759999385f3549cbf (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
class Equipment:
    def __init__(self, name=None, mgmtip=None):
        self.name = name
        self.mgmtip = mgmtip
        self.ifaces = {}

    def __repr__(self):
        return repr( (self.name, self.mgmtip, self.ifaces.values()) )

    def add_seen_mac(self, ifname, mac):
        iface = self.ifaces.get(ifname)
        if not iface:
            iface = Interface()
            iface.name = ifname
            self.ifaces[ifname] = iface
        iface.mac_seen.append(mac)

class Interface:
    def __init__(self, name=None, mac=None):
        self.name = name
        self.mac = mac
        self.mac_seen = []
        self.remote = []

    def __repr__(self):
        return repr( (self.mac, self.name) )