import sqlite3
# 打开数据库
def opendb():
conn = sqlite3.connect("e:\mydb.db")
cur = conn.execute("""create table if not exists tongxinlu(usernum integer primary key,username varchar(128), passworld varchar(128),address varchar(125), telnum varchar(128))""")
return cur, conn
#查询全部信息
def showalldb():
print("-------------------处理后的数据-------------------")
hel = opendb()
cur = hel[1].cursor()
cur.execute("select * from tongxinlu")
res = cur.fetchall()
for line in res:
for h in line:
print(h),
print
cur.close()
#输入信息
def into():
usernum=input("请输入学号：")
username1 = input("请输入姓名：")
passworld1 = input("请输入密码：")
address1 = input("请输入地址：")
telnum1 = input("请输入联系电话：")
return usernum,username1, passworld1, address1, telnum1
# 往数据库中添加内容
def adddb():
welcome = """-------------------欢迎使用添加数据功能---------------------"""
print(welcome)
person = into()
hel = opendb()
hel[1].execute("insert into tongxinlu(usernum,username, passworld, address, telnum)values (?,?,?,?,?)",(person[0], person[1], person[2], person[3],person[4]))
hel[1].commit()
print ("-----------------恭喜你,数据添加成功----------------")
showalldb()
hel[1].close()
# 删除数据库中的内容
def deldb():
welcome = "------------------欢迎使用删除数据库功能------------------"
print(welcome)
delchoice = input("请输入想要删除学号：")
hel = opendb() # 返回游标conn
hel[1].execute("delete from tongxinlu where usernum ="+delchoice)
hel[1].commit()
print ("-----------------恭喜你，数据删除成功----------------")
showalldb()
hel[1].close()
# 修改数据库的内容
def alter():
welcome = "--------------------欢迎使用修改数据库功能-----------------"
print(welcome)
changechoice = input("请输入想要修改的学生的学号:")
hel =opendb()
person = into()
hel[1].execute("update tongxinlu set usernum=?,username=?, passworld= ?,address=?,telnum=? where usernum="+changechoice,(person[0], person[1], person[2], person[3],person[4]))
hel[1].commit()
showalldb()
hel[1].close()
# 查询数据
def searchdb():
welcome = "--------------------欢迎使用查询数据库功能-----------------"
print(welcome)
choice = input("请输入要查询的学生的学号：")
hel = opendb()
cur = hel[1].cursor()
cur.execute("select * from tongxinlu where usernum="+choice)
hel[1].commit()
print("-------------------恭喜你，你要查找的数据如下---------------------")
for row in cur:
print(row[0],row[1],row[2],row[3],row[4])
cur.close()
hel[1].close()
# 是否继续
def conti(a):
choice = input("是否继续？（y or n):")
if choice == 'y':
a = 1
else:
a = 0
return a
if __name__ == "__main__":
flag = 1
while flag:
welcome = "---------欢迎使用数据库通讯录---------"
print(welcome)
choiceshow = """
请选择您的进一步选择：
(添加)往数据库里面添加内容
(删除)删除数据库中内容
(修改)修改书库的内容
（查询）查询数据的内容
选择您想要的进行的操作：
"""
choice = input(choiceshow)
if choice == "添加":
adddb()
conti(flag)
elif choice == "删除":
deldb()
conti(flag)
elif choice == "修改":
alter()
conti(flag)
elif choice == "查询":
searchdb()
conti(flag)
else:
print("你输入错误，请重新输入")
