python通讯录

今天是个简易的通讯录

#作者@南方人啦
import sqlite3
# 打开数据库
def opendb():
        conn = sqlite3.connect("d:\mydb.db")
        cur = conn.execute("""create table if not exists tongxunlu(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()
        #执行sql语句
        cur.execute("select * from tongxunlu")
        for row in cur:
            print(row[0],row[1],row[2],row[3],row[4])
        cur.close()
        hel[1].close()
'''        #fetchall()返回结果集的剩余行     (fetchone()返回下一行)
        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 tongxunlu(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 tongxunlu where usernum ="+delchoice)
        hel[1].commit()
        print ("-----------------恭喜你，数据删除成功----------------")
        showalldb()
        hel[1].close()
#  修改数据库的内容
def alterdb():
        welcome = "--------------------欢迎使用修改数据库功能-----------------"
        print(welcome)
        changechoice = input("请输入想要修改的学生的学号:")
        hel =opendb()
        person = into()
        hel[1].execute("update tongxunlu 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 tongxunlu 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 = """
请选择您的进一步选择：
(1)往数据库里面添加内容
(2)删除数据库中内容
(3)修改书库的内容
（4）查询数据的内容
选择您想要的进行的操作：
"""
                choice = input(choiceshow)
                if choice == "1":
                        adddb()
                        conti(flag)
                elif choice == "2":
                        deldb()
                        conti(flag)
                elif choice == "3":
                        alterdb()
                        conti(flag)
                elif choice == "4":
                        searchdb()
                        conti(flag)
                else:
                        print("你输入错误，请重新输入")

运行