python实现简易通讯录修改版

描述：


上一篇博客我写了一个简单的通讯录，但是还是觉得不够完美：


需要输入ID，虽然ID是主键，但是没有实现自增功能；


忘记加电话号码了；


如果插入用户名相同，则后续的查、改、删功能受到影响；


所以，我这个修改版主要修正了以上三点缺陷。具体如下


将用户ID设置为自增字段，即就是不用输入ID，程序会自动帮你将插入的用户进行排序并附上ID，默认从1开始；SQLite实现主键自增功能，请查看SQLiteAUTOINCREMENT/自动递增这篇博客。


创建表时增加了电话号码字段；


插入的用户名不能重复；


好了，废话不多说，附上新代码，哈哈——


#-*- coding:utf-8 -*-


import sqlite3


#打开本地数据库用于存储用户信息


conn = sqlite3.connect('mysql_person.db')


#在该数据库下创建表,创建表的这段代码在第一次执行后需要注释掉，否则再次执行程序会一直提示：该表已存在


conn.execute ('''CREATE TABLE MT_table1(


ID INTEGER PRIMARY KEY  AUTOINCREMENT,


NAME      TEXT   NOT NULL,


AGE      INT    NOT NULL,


ADDRESS    CHAR(50),


SALARY     REAL,


PhoneNumber  INT    NOT NULL);''')


print "Table created successfully";


#增加用户信息


def insert():


USER_NAME = raw_input('请输入用户昵称:')


cursor = conn.execute("SELECT name from MT_table1 where name = '%s';"%USER_NAME)


for row in cursor:


if USER_NAME == row[0]:


print "sorry,改用户名已存在，请重新输入用户名"


break


else:


AGE = input('请输入年龄:')


ADDRESS = raw_input('请输入用户地址:')


SALARY = input('请输入用户薪水:')


PhoneNumber = input("请输入联系方式:")


#为避免代码太长，以下我用字符串的连接特性将代码缩短


sql1 = 'INSERT INTO MT_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 "Records insert successfully"


#删除用户信息


def delete():


delete_name = raw_input("请输入所要删除的联系人姓名:")


cursor = conn.execute("SELECT name from MT_table1 where name = '%s';"%delete_name)


for row in cursor:


if delete_name == row[0]:


conn.execute("DELETE from MT_table1 where name = '%s';"%delete_name)


conn.commit()


print "Records delete successfully"


break


else:


print "sorry,不存在该用户"


#修改用户信息


def modify():


update_name = raw_input("请输入要修改用户的姓名:")


sql6 = "SELECT name from MT_table1 where name = '%s';"%update_name


cursor = conn.execute(sql6)


for row in cursor:


if update_name == row[0]:


New_addr = raw_input("请输入要修改用户的新地址:")


New_age = input("请输入要修改用户的新年龄:")


New_salary = input("请输入要修改用户的新薪水:")


New_num = input("输入要修改用户的新号码:")


sql3 = "UPDATE MT_table1 set address = '%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 = "SELECT id, name, age,address, salary ,PhoneNumber from MT_table1 where name = '%s';"%update_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]


print "PhoneNumber = ",row[5], "\n"


break


else:


print "sorry,不存在该用户信息"


#查询用户信息


def search():


conn = sqlite3.connect('mysql_person.db')


search_name = raw_input('请输入要查询的用户姓名')


sql2 = "SELECT id,name,age, address, salary ,PhoneNumber from MT_table1 where name= '%s';" % (search_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]


print "PhoneNumber = ",row[5], "\n"


break


else:


print "sorry,没有该用户信息"


#显示所有用户信息


def showall():


cursor = conn.execute("SELECT id, name,age, address, salary ,PhoneNumber from MT_table1")


for row in cursor:


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 "Operation done successfully";


cursor = conn.execute("select count(*) from MT_table1;")


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