move Link and Node to seperate files
parent
967eba4c13
commit
d23736d0f5
27
bat2nodes.py
27
bat2nodes.py
|
@ -8,6 +8,9 @@ import json
|
||||||
import fileinput
|
import fileinput
|
||||||
import argparse
|
import argparse
|
||||||
|
|
||||||
|
from node import Node
|
||||||
|
from link import Link
|
||||||
|
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
|
|
||||||
parser.add_argument('-a', '--aliases',
|
parser.add_argument('-a', '--aliases',
|
||||||
|
@ -27,30 +30,6 @@ aliases = dict()
|
||||||
links = set()
|
links = set()
|
||||||
nodes = []
|
nodes = []
|
||||||
|
|
||||||
class Node():
|
|
||||||
def __init__(self):
|
|
||||||
self.name = ""
|
|
||||||
self.macs = set()
|
|
||||||
self.group = 0
|
|
||||||
self.online = False
|
|
||||||
# groups:
|
|
||||||
# 0 normal node
|
|
||||||
# 1 aftermath
|
|
||||||
# 2 gateways
|
|
||||||
# 3 TT
|
|
||||||
|
|
||||||
def add_mac(self, mac):
|
|
||||||
self.macs.add(mac)
|
|
||||||
|
|
||||||
def __repr__(self):
|
|
||||||
return self.macs.__repr__()
|
|
||||||
|
|
||||||
class Link():
|
|
||||||
def __init__(self):
|
|
||||||
self.pair = None
|
|
||||||
self.distance = None
|
|
||||||
self.strength = None
|
|
||||||
|
|
||||||
def maybe_node_by_mac(nodes, macs):
|
def maybe_node_by_mac(nodes, macs):
|
||||||
for node in nodes:
|
for node in nodes:
|
||||||
for mac in macs:
|
for mac in macs:
|
||||||
|
|
|
@ -0,0 +1,6 @@
|
||||||
|
class Link():
|
||||||
|
def __init__(self):
|
||||||
|
self.pair = None
|
||||||
|
self.distance = None
|
||||||
|
self.strength = None
|
||||||
|
|
|
@ -0,0 +1,19 @@
|
||||||
|
class Node():
|
||||||
|
def __init__(self):
|
||||||
|
self.name = ""
|
||||||
|
self.macs = set()
|
||||||
|
self.group = 0
|
||||||
|
self.online = False
|
||||||
|
# groups:
|
||||||
|
# 0 normal node
|
||||||
|
# 1 aftermath
|
||||||
|
# 2 gateways
|
||||||
|
# 3 TT
|
||||||
|
|
||||||
|
def add_mac(self, mac):
|
||||||
|
self.macs.add(mac)
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return self.macs.__repr__()
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue