-
Notifications
You must be signed in to change notification settings - Fork 0
/
config_parser.py
73 lines (51 loc) · 2.08 KB
/
config_parser.py
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
import configparser
class config_parser(object):
def __init__(self, config_file):
self.config_file = config_file
self.config = configparser.ConfigParser()
def getParam(self, name, default):
self.config.read(self.config_file)
if "config" in self.config:
if name in self.config["config"]:
return self.config["config"][name]
return default
def getScanRange(self):
return self.getParam("scan_range", "192.168.2.0/24")
def getUsername(self):
return self.getParam("username", "username")
def getPassword(self):
return self.getParam("password", "password")
def getNumbers(self):
return self.getParam("numbers", "041928491").split(",")
def getClients(self):
self.config.read(self.config_file)
clients = []
if "config" in self.config:
if "clients" in self.config["config"]:
for client in self.config["config"]["clients"].split(","):
if not client in self.config:
continue
if not "mac" in self.config[client]:
continue
clients.append((client, self.config[client]))
return clients
def addClient(self, client, mac):
self.config.read(self.config_file)
conf = self.config["config"]
if not client in self.config:
if not "clients" in conf:
conf["clients"] = ""
self.config.add_section( client )
conf["clients"] = ",".join(conf["clients"].split(",") + [client])
self.config[client]["mac"] = mac
self.save()
def removeClient(self, client):
self.config.read(self.config_file)
conf = self.config["config"]
if client in self.config:
self.config.remove_section(client)
conf["clients"] = ",".join([x for x in conf["clients"].split(",") if x!=client])
self.save()
def save(self):
with open(self.config_file, 'w') as configfile:
self.config.write(configfile)