Python实现通讯录功能
说实话，第一次写这么长的Python代码，期间遇到了很多问题，但是，最终还是完成了，花了我一天半的时间。
该程序实现了用户的增，删，改，查，主要用到sqlite3模块。对于该模块的知识点，请查看我的另一篇文章SQLite-Python
如有更好的建议，请私信，本人将不胜荣幸，让我们一起来学习Python！
#-*-coding:utf-8-*-
importsqlite3
#打开本地数据库用于存储用户信息
conn=sqlite3.connect('mysql_person.db')
#在该数据库下创建表，创建表的这段代码在第一次执行后需要注释掉，否则再次执行程序会一直提示：该表已存在
conn.execute('''CREATETABLEMT
(IDINTPRIMARYKEYNOTNULL,
NAMETEXTNOTNULL,
AGEINTNOTNULL,
ADDRESSCHAR(50),
SALARYREAL);''')
print"Tablecreatedsuccessfully";
conn.close()
#增加用户信息
definsert():
ID=input('请输入用户ID:\n')
NAME=raw_input('请输入用户昵称:\n')
AGE=input('请输入年龄:\n')
ADDRESS=raw_input('请输入用户地址:\n')
SALARY=input('请输入用户薪水:\n')
sql1='insertintoMT(ID,NAME,AGE,ADDRESS,SALARY)'
sql1+='values("%d","%s","%d","%s","%d");'%(ID,NAME,AGE,ADDRESS,SALARY)
conn.execute(sql1)
conn.commit()
print"Recordsinsertsuccessfully"
#删除用户信息
defdelete():
name=raw_input("请输入所要删除的联系人姓名:")
cursor=conn.execute("SELECTnamefromMTwherename='%s';"%name)
forrowincursor:
ifname==row[0]:
conn.execute("DELETEfromMTwherename='%s';"%name)
conn.commit()
print"Recordsdeletesuccessfully"
break
else:
print"sorry,不存在该用户"
#修改用户信息
defmodify():
name=raw_input("请输入要修改用户的姓名:")
printsearch()
sql4="SELECTid,name,age,address,salaryfromMTwherename='%s';"%name
cursor=conn.execute(sql4)
x=raw_input("请输入要修改用户的新地址:")
y=input("请输入要修改用户的新年龄:")
z=input("请输入要修改用户的新薪水:")
sql3="UPDATEMTsetaddress='%s',age='%d',\
salary='%d'wherename='%s';"%(x,y,z,name)
conn.execute(sql3)
conn.commit()
print"修改成功"
sql5="SELECTid,name,age,address,salaryfromMTwherename='%s';"%name
cursor=conn.execute(sql5)
forrowincursor:
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')
defsearch():
conn=sqlite3.connect('mysql_person.db')
name=raw_input('请输入要查询的用户姓名')
sql2="SELECTid,name,age,address,salaryfromMTwherename='%s';"%(name)
cursor=conn.execute(sql2)
forrowincursor:
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,没有该用户信息"
#显示所有用户信息
defshowall():
cursor=conn.execute("SELECTid,age,name,address,salaryfromMT")
forrowincursor:
print"ID=",row[0]
print"NAME=",row[1]
print"AGE=",row[2]
print"ADDRESS=",row[3]
print"SALARY=",row[4],"\n"
print"Operationdonesuccessfully";
cursor=conn.execute("selectcount(*)fromMT;")
forrowincursor:
print"一共有%d个用户"%row[0]
defmenu():
print'1.新增联系人'
print'2.删除联系人'
print'3.修改联系人'
print'4.查询联系人'
print'5.显示所有联系人'
print'6.退出程序'
print'Whatdoyouwanttodo?'
whileTrue:
menu()
x=raw_input('请输入您的选择菜单号:')
ifx=='1':
insert()
continue
ifx=='2':
delete()
continue
ifx=='3':
modify()
continue
ifx=='4':
search()
continue
ifx=='5':
showall()
continue
ifx=='6':
print"谢谢使用！"
exit()
continue
else:
print"输入的选项不存在，请重新输入！"
continue
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持毛票票。