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
|
from astconfigparser import MultiOrderedConfigParser
import MySQLdb
import traceback
class SqlConfigParser(MultiOrderedConfigParser):
_tablename = "sippeers"
def __init__(self,tablename="sippeers"):
self._tablename=tablename
MultiOrderedConfigParser.__init__(self)
def connect(self, user, password, host, port, database):
self.cnx = MySQLdb.connect(user=user,passwd=password,host=host,port=int(port),db=database)
def read(self, filename, sect=None):
MultiOrderedConfigParser.read(self, filename, sect)
# cursor = self.cnx.cursor(dictionary=True)
cursor = self.cnx.cursor(cursorclass=MySQLdb.cursors.DictCursor)
cursor.execute("SELECT * from `" + MySQLdb.escape_string(self._tablename) + "`")
rows = cursor.fetchall()
for row in rows:
sect = self.add_section(row['name'])
for key in row:
if (row[key] != None):
for elem in str(row[key]).split(";"):
sect[key] = elem
#sect[key] = str(row[key]).split(";")
def write_dicts(self, config_file, mdicts):
"""Write the contents of the mdicts to the specified config file"""
for section, sect_list in mdicts.iteritems():
# every section contains a list of dictionaries
for sect in sect_list:
sql = "INSERT INTO "
if (sect.get('type')[0] == "endpoint"):
sql += "ps_endpoints "
elif (sect.get('type')[0] == "aor" and section != "sbc"):
sql += "ps_aors "
elif (sect.get('type')[0] == "identify"):
sql += "ps_endpoint_id_ips"
else:
continue
sql += " SET `id` = " + "\"" + MySQLdb.escape_string(section) + "\""
for key, val_list in sect.iteritems():
if key == "type":
continue
# every value is also a list
key_val = " `" + key + "`"
key_val += " = " + "\"" + MySQLdb.escape_string(";".join(val_list)) + "\""
sql += ","
sql += key_val
config_file.write("%s;\n" % (sql))
def write(self, config_file):
"""Write configuration information out to a file"""
try:
self.write_dicts(config_file, self._sections)
except Exception,e:
print "Could not open file ", config_file, " for writing"
traceback.print_exc()
|
