作业就是写一个简单 的图书信息管理系统，实现增删改查功能，附带了登录注册功能，就是把上一次的代码直接拿来用了。如下：
主函数：main.py
import pymysql import getpass import time from operate import manage #查询用户名是否存在 def check(uname): namelist = [] conn1 = pymysql.connect("localhost", "root", "admin", "login", charset=\'utf8\') cur = conn1.cursor() sql1 = "select uName from logintable" cur.execute(sql1) results = cur.fetchall() for row in results: namelist.append(row[0]) i = 0 for x in namelist: if (uname == x): i = 1 break return i # 注册函数 def register(): i = 1 username = "" passwd = "" while i == 1: username = input("请输入用户名：") if (check(username) == 1): print(\'该用户名已被占用，请重新输入\') continue if (username.isalnum()): break else: print("用户名只能包含数字与英文字母！！！") while i == 1: passwd = getpass.getpass("请输入密码：") if (len(passwd) > 5 and len(passwd) < 21): break else: print("密码为6至20位！！！") pswd = getpass.getpass(\'请再输入一遍密码：\') if (passwd == pswd): print(\'注册成功！\') else: print(\'两次密码不相同，请重新注册！！！\') register() # 打开数据库连接 conn = pymysql.connect("localhost", "root", "admin", "login", charset=\'utf8\') # 使用corsor()方法创建一个游标对象 cursor = conn.cursor() # 使用execute()方法执行sql语句，将注册的信息插入到数据库中 sql = "INSERT INTO logintable (uName,pw) VALUES (%s,%s)" val = (username, passwd) # 这两行执行sql语句 cursor.execute(sql, val) conn.commit()#提交数据库执行 cursor.close() conn.close() # 返回主界面 first() # 登录函数 def login(): sql = "select * from logintable" conn = pymysql.connect("localhost", "root", "admin", "login", charset=\'utf8\') cursor = conn.cursor() cursor.execute(sql) result = cursor.fetchall() i = 1 s = 0 while i == 1: uname = input(\'请输入用户名：\') if (check(uname) == 0): print(\'该用户不存在！！！\') continue else: while s < 3: passwd = getpass.getpass("请输入密码：") key = 0 for row in result: if (uname == row[0] and passwd == row[1]): key = 1 if (key == 1): print(\'登录成功！！！\') break else: print(\'密码错误！请重新登录！\') s = s + 1 if s == 3: print(\'账户已锁定！！！请 5 s 后重新登录！！！\') time.sleep(5) break manage() #主界面函数 def first(): print(\'--------------xxx图书信息管理系统------------\') print(\'输入序号，进行选择：\') print(\'1、注册\') print(\'2、登录\') print(\'3、退出\') select = input(\'请选择： \') if select == \'1\': register() elif select == \'2\': login() elif select == \'3\': exit(0) else: print(\'请输入正确序号：\') first() first()
增删改查模块 ：operate.py
import pymysql #from sqlConnect import addToMysql,check_bookname,check_author from mysqlOperate import* import mysqlOperate #管理函数 def manage(): m=1 while m == 1: print("-----------------------图书管理----------------------") print("1、查询图书信息") print("2、添加图书信息") print("3、删除图书信息") print("4、修改图书信息") print("5、退出") select = input("请输入编号选择功能：") if select == \'1\': check() elif select == \'2\': add() elif select == \'3\': delete() elif select == \'4\': update() elif select == \'5\': exit(0) else: print("请输入正确序号！！！") #添加函数 def add(): print("请输入图书信息：") bookname=input("请输入书籍名称：") author=input(\'请输入作者姓名：\') price=input(\'请输入价格：\') addToMysql(bookname,author,price) #查询函数 def check(): print("-----------------------查询---------------------") print("请选择：") print("1、按书名查询 2、按作者查询 3、查询所有图书 （其他任意键返回）") select=input(\'请输入编号：\') if select==\'1\': bookname = input("请输入书名：") check_bookname(bookname) elif select==\'2\': authorname=input("请输入作者姓名：") check_author(authorname) elif select==\'3\': check_all() #删除函数 def delete(): print("-----------------------删除----------------------") bookname=input("请输入要删除的书名：") check_bookname(bookname) select=input("确定删除？（请输入y/n）：") if select==\'y\': delete_book(bookname) elif select==\'n\': print("未删除！！！") else: print("输入错误！！已退出！！！") #修改函数 def update(): print("-----------------------修改------------------------") bookname=input("请输入要修改的书名：") check_bookname(bookname) select=input(\'确定修改？（y/n）：\') if select==\'n\': print(\'已返回\') elif select==\'y\': update_book(bookname)
数据库操作模块：mysqlOperate.py
import pymysql #创建数据库连接 conn=pymysql.connect("localhost","root","admin","login",charset=\'utf8\') #创建游标对象 cursor=conn.cursor() #添加图书 def addToMysql(name,author,price): str="INSERT INTO bookinfo (bookname,author,price) VALUES (%s,%s,%s)" val=(name,author,price) cursor.execute(str,val) conn.commit() print("添加成功") #按书名查询 def check_bookname(bookName): cursor.execute(\'SELECT * FROM bookinfo WHERE bookname = %s\',(bookName,)) result=cursor.fetchall() i = 0 if result==None: print (\'该书不存在！！！\') else: for row in result: print("-------第%s本书-------" % i) print("book_id:", row[0]) print("书名：", row[1]) print("作者：", row[2]) print("价格：", row[3]) i = i + 1 # 按作者查询 def check_author(author): cursor.execute(\'SELECT * FROM bookinfo WHERE author = %s\',(author,)) result=cursor.fetchall() i=1 if result==None: print(\'该作者不存在！！！\') else: for row in result: print(\'-------第%s本书--------\'% i) print("book_id:", row[0]) print("书名：", row[1]) print("作者：", row[2]) print("价格：", row[3]) i = i + 1 #查询所有图书 def check_all(): cursor.execute(\'SELECT * FROM bookinfo\') result=cursor.fetchall() i=1 for row in result: print(\'-------第%s本书--------\' % i) print("book_id:", row[0]) print("书名：", row[1]) print("作者：", row[2]) print("价格：", row[3]) i = i + 1 # 删除书籍 def delete_book(bookname): cursor.execute(\'DELETE FROM bookinfo WHERE bookname = %s\' ,(bookname,)) conn.commit() print(\'删除成功！！！\') #修改书籍信息 def update_book(bookname): new_bookname=input(\'请输入修改书名：\') new_author=input(\'请输入修改作者名称：\') new_price=input(\'请输入修改价格：\') cursor.execute(\'UPDATE bookinfo SET bookname=%s,author=%s,price=%s WHERE bookname=%s\',(new_bookname,new_author,new_price,bookname)) conn.commit() print(\'修改成功\')
数据库设计：
bookinfo表
logintable表：
另外，执行时跟上一次作业一样，通过cmd执行