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
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持毛票票。