python实现简易通讯录修改版
描述:
上一篇博客我写了一个简单的通讯录,但是还是觉得不够完美:
需要输入ID,虽然ID是主键,但是没有实现自增功能;
忘记加电话号码了;
如果插入用户名相同,则后续的查、改、删功能受到影响;
所以,我这个修改版主要修正了以上三点缺陷。具体如下
将用户ID设置为自增字段,即就是不用输入ID,程序会自动帮你将插入的用户进行排序并附上ID,默认从1开始;SQLite实现主键自增功能,请查看SQLiteAUTOINCREMENT/自动递增这篇博客。
创建表时增加了电话号码字段;
插入的用户名不能重复;
好了,废话不多说,附上新代码,哈哈~~~
#-*-coding:utf-8-*-
importsqlite3
#打开本地数据库用于存储用户信息
conn=sqlite3.connect('mysql_person.db')
#在该数据库下创建表,创建表的这段代码在第一次执行后需要注释掉,否则再次执行程序会一直提示:该表已存在
conn.execute('''CREATETABLEMT_table1(
IDINTEGERPRIMARYKEYAUTOINCREMENT,
NAMETEXTNOTNULL,
AGEINTNOTNULL,
ADDRESSCHAR(50),
SALARYREAL,
PhoneNumberINTNOTNULL);''')
print"Tablecreatedsuccessfully";
#增加用户信息
definsert():
USER_NAME=raw_input('请输入用户昵称:')
cursor=conn.execute("SELECTnamefromMT_table1wherename='%s';"%USER_NAME)
forrowincursor:
ifUSER_NAME==row[0]:
print"sorry,改用户名已存在,请重新输入用户名"
break
else:
AGE=input('请输入年龄:')
ADDRESS=raw_input('请输入用户地址:')
SALARY=input('请输入用户薪水:')
PhoneNumber=input("请输入联系方式:")
#为避免代码太长,以下我用字符串的连接特性将代码缩短
sql1='INSERTINTOMT_table1(NAME,AGE,ADDRESS,SALARY,PhoneNumber)'
sql1+='VALUES("%s","%d","%s","%d","%d");'%(USER_NAME,AGE,ADDRESS,SALARY,PhoneNumber)
conn.execute(sql1)
conn.commit()
print"Recordsinsertsuccessfully"
#删除用户信息
defdelete():
delete_name=raw_input("请输入所要删除的联系人姓名:")
cursor=conn.execute("SELECTnamefromMT_table1wherename='%s';"%delete_name)
forrowincursor:
ifdelete_name==row[0]:
conn.execute("DELETEfromMT_table1wherename='%s';"%delete_name)
conn.commit()
print"Recordsdeletesuccessfully"
break
else:
print"sorry,不存在该用户"
#修改用户信息
defmodify():
update_name=raw_input("请输入要修改用户的姓名:")
sql6="SELECTnamefromMT_table1wherename='%s';"%update_name
cursor=conn.execute(sql6)
forrowincursor:
ifupdate_name==row[0]:
New_addr=raw_input("请输入要修改用户的新地址:")
New_age=input("请输入要修改用户的新年龄:")
New_salary=input("请输入要修改用户的新薪水:")
New_num=input("输入要修改用户的新号码:")
sql3="UPDATEMT_table1setaddress='%s',age='%d',salary='%d',PhoneNumber='%d'where\
name='%s';"%(New_addr,New_age,New_salary,New_num,update_name)
conn.execute(sql3)
conn.commit()
print"修改成功"
sql5="SELECTid,name,age,address,salary,PhoneNumberfromMT_table1wherename='%s';"%update_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]
print"PhoneNumber=",row[5],"\n"
break
else:
print"sorry,不存在该用户信息"
#查询用户信息
defsearch():
conn=sqlite3.connect('mysql_person.db')
search_name=raw_input('请输入要查询的用户姓名')
sql2="SELECTid,name,age,address,salary,PhoneNumberfromMT_table1wherename='%s';"%(search_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]
print"PhoneNumber=",row[5],"\n"
break
else:
print"sorry,没有该用户信息"
#显示所有用户信息
defshowall():
cursor=conn.execute("SELECTid,name,age,address,salary,PhoneNumberfromMT_table1")
forrowincursor:
print"ID=",row[0]
print"NAME=",row[1]
print"AGE=",row[2]
print"ADDRESS=",row[3]
print"SALARY=",row[4]
print"PhoneNumber=",row[5],"\n"
print"Operationdonesuccessfully";
cursor=conn.execute("selectcount(*)fromMT_table1;")
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
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。