-
Notifications
You must be signed in to change notification settings - Fork 11
/
Database.py
47 lines (42 loc) · 1.4 KB
/
Database.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
import MySQLdb
class Mysqldb:
def __init__(self, host = "localhost", user = "root", passwd = ""):
self.Host = host
self.User = user
self.Passwd = passwd
self.Connecting()
def Connecting(self):
try:
self.db = MySQLdb.connect(host=self.Host, user=self.User,passwd=self.Passwd, db="Iptables" )
self.cursor = self.db.cursor()
except:
pass
def Execute(self,sql):
try:
tmp = self.cursor.execute(sql)
info = self.cursor.fetchmany(tmp)
self.db.commit()
return info
except Exception as e:
print e
pass
def Createtable(self,table):
sql = ("CREATE TABLE IF NOT EXISTS %s (\
`id` int(11) NOT NULL AUTO_INCREMENT,\
`SIP` varchar(20) NOT NULL,\
`Dport` INT DEFAULT NULL,\
`Flags` varchar(6) DEFAULT NULL,\
`Dt` datetime DEFAULT NULL,\
PRIMARY KEY (`id`)\
) ENGINE=InnoDB AUTO_INCREMENT=0 ;")%(table)
self.cursor.execute(sql)
self.db.commit()
if __name__ == '__main__':
a=Mysqldb()
sql="INSERT INTO a2017 VALUES ('', '127.0.0.1%', '123', 'S')"
a.Execute(sql)
"""
def Analysising(self,t):
Data=self.cursor.fetchmany(self.cursor.execute( "SELECT COUNT(*) FROM a2017" ))
print Data
"""