说实话，第一次写这么长的Python代码，期间遇到了很多问题，但是，最终还是完成了，花了我一天半的时间。
该程序实现了用户的增，删，改，查，主要用到sqlite3模块。对于该模块的知识点，请查看我的另一篇文章SQLite-Python
如有更好的建议，请私信，本人将不胜荣幸，让我们一起来学习Python！
|
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
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
|
#-*- coding:utf-8 -*-
import sqlite3
#打开本地数据库用于存储用户信息
conn = sqlite3.connect('mysql_person.db')
#在该数据库下创建表，创建表的这段代码在第一次执行后需要注释掉，否则再次执行程序会一直提示：该表已存在
conn.execute('''CREATE TABLE MT
(ID INT PRIMARY KEY NOT NULL,
NAME TEXT NOT NULL,
AGE INT NOT NULL,
ADDRESS CHAR(50),
SALARY REAL);''')
print "Table created successfully";
conn.close()
#增加用户信息
def insert():
ID = input('请输入用户ID:\n')
NAME = raw_input('请输入用户昵称:\n')
AGE = input('请输入年龄:\n')
ADDRESS = raw_input('请输入用户地址:\n')
SALARY = input('请输入用户薪水:\n')
sql1 = 'insert into MT(ID,NAME,AGE,ADDRESS,SALARY)'
sql1 += 'values("%d","%s","%d","%s","%d");'%(ID,NAME,AGE,ADDRESS,SALARY)
conn.execute(sql1)
conn.commit()
print "Records insert successfully"
#删除用户信息
def delete():
name = raw_input("请输入所要删除的联系人姓名:")
cursor = conn.execute("SELECT name from MT where name = '%s';"%name)
for row in cursor:
if name == row[0]:
conn.execute("DELETE from MT where name = '%s';"%name)
conn.commit()
print "Records delete successfully"
break
else:
print "sorry,不存在该用户"
#修改用户信息
def modify():
name = raw_input("请输入要修改用户的姓名:")
print search()
sql4 = "SELECT id, name, age,address, salary from MT where name = '%s';"%name
cursor = conn.execute(sql4)
x = raw_input("请输入要修改用户的新地址:")
y = input("请输入要修改用户的新年龄:")
z = input("请输入要修改用户的新薪水:")
sql3 = "UPDATE MT set address = '%s',age = '%d',\
salary = '%d' where name = '%s';"%(x,y,z,name)
conn.execute(sql3)
conn.commit()
print "修改成功"
sql5 = "SELECT id, name, age,address, salary from MT where name = '%s';"%name
cursor = conn.execute(sql5)
for row in cursor:
print "ID = ", row[0]
print "NAME = ", row[1]
print "AGE = ",row[2]
print "ADDRESS = ", row[3]
print "SALARY = ", row[4], "\n"
#查询用户信息
conn = sqlite3.connect('mysql_person.db')
def search():
conn = sqlite3.connect('mysql_person.db')
name = raw_input('请输入要查询的用户姓名')
sql2 = "SELECT id,name,age, address, salary from MT where name= '%s';" % (name)
cursor = conn.execute(sql2)
for row in cursor:
print "ID = ", row[0]
print "NAME = ", row[1]
print "AGE = ",row[2]
print "ADDRESS = ", row[3]
print "SALARY = ", row[4], "\n"
break
else:
print "sorry,没有该用户信息"
#显示所有用户信息
def showall():
cursor = conn.execute("SELECT id, age, name, address, salary from MT")
for row in cursor:
print "ID = ", row[0]
print "NAME = ", row[1]
print "AGE = ",row[2]
print "ADDRESS = ", row[3]
print "SALARY = ", row[4], "\n"
print "Operation done successfully";
cursor = conn.execute("select count(*) from MT;")
for row in cursor:
print "一共有%d个用户"%row[0]
def menu():
print '1.新增联系人'
print '2.删除联系人'
print '3.修改联系人'
print '4.查询联系人'
print '5.显示所有联系人'
print '6.退出程序'
print 'What do you want to do?'
while True:
menu()
x = raw_input('请输入您的选择菜单号:')
if x == '1':
insert()
continue
if x == '2':
delete()
continue
if x == '3':
modify()
continue
if x == '4':
search()
continue
if x == '5':
showall()
continue
if x == '6':
print "谢谢使用！"
exit()
continue
else:
print "输入的选项不存在，请重新输入！"
continue
|
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持服务器之家。
原文链接：http://blog.csdn.net/MTbaby/article/details/58599864