-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathdb.py
executable file
·61 lines (51 loc) · 1.84 KB
/
db.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
#!/usr/bin/env python
#coding:utf-8
import MySQLdb as mysql
conn=mysql.connect(user='root',passwd='jrh08231009',db='reboot10',charset='utf8')
curs=conn.cursor()
#获取用户列表
def get_userlist(userlist):
sql="select %s from users" %','.join(userlist)
users=[]
curs.execute(sql)
result=curs.fetchall()
users=[dict((k,row[i]) for i, k in enumerate(userlist)) for row in result]
# for row in result:
# user={}
# for k,v in enumerate(userlist):
# user[v]=row[k]
# users.append(user)
return users
#获取单个用户
def getuser(uid):
userlist=["name","name_cn","email","mobile","role","status"]
sql="select %s from users where id=%s" %(",".join(userlist),uid)
curs.execute(sql)
result=curs.fetchone()
conn.commit()
userinfo={}
for i,k in enumerate(userlist):
userinfo[k]=result[i]
return userinfo
#添加用户
def add_user(userlist):
# sql='insert into users(name,name_cn,password,email,mobile,role,status)values("%s")'%'","'.join(userlist)
sql="insert into users(%s)values('%s')"%(",".join(userlist.keys()),"','".join(userlist.values()))
curs.execute(sql)
conn.commit()
#删除用户
def del_user(uid):
sql="delete from users where id=%s"%uid
curs.execute(sql)
conn.commit()
#更新用户
def update_user(userdict):
sql="update users set email='%(email)s',mobile='%(mobile)s',role='%(role)s',status='%(status)s' where name='%(name)s'"%(userdict)
curs.execute(sql)
conn.commit()
#检查用户
def checkuser(name):
sql = "select password from users where name='%s'"%name
curs.execute(sql)
passwd = curs.fetchone()
return passwd[0]