import mytools
import pymysql
from mytools import select_many,select_one,add_update_delete
class Book():
#显示图书列表
def many(self):
sql = "select * from book"
s1 = select_many(sql)
print('图书编号\t\t书名\t\t\t描述\t\t\t\t价格/元')
for i in range(len(s1)):
for j in range(len(s1[i])):
print(s1[i][j], end='\t\t')
print()
#书籍添加
def add(self):
name = input('请输入书籍名字:')
sql = f"select * from book where name='{name}'"
s2 = select_one(sql)
if s2 is None:
descp = input('请输入书籍描述:')
price = float(input('请输入书籍价格:'))
sql = f"insert into book(name,descp,price) values('{name}','{descp}','{price}')"
add_update_delete(sql)
print('书籍添加成功')
else:
print('图书已存在,请勿重复添加')
#书籍更新
def update(self):
id = int(input('请输入要修改的书籍编号:'))
sql = f"select * from book where id='{id}'"
s4 = select_one(sql)
if s4 is None:
print('您输入的图书编号不存在,则无法进行修改')
else:
print('您要修改的信息如下')
print('书名:', s4[1])
print('描述:', s4[2])
print('价格:', s4[3])
print('请输入新的信息:')
name = input('书名:')
descp = input('描述:')
price = float(input('价格:'))
sql = f"update book set name='{name}',descp='{descp}',price='{price}' where id='{id}'"
add_update_delete(sql)
print('恭喜您,修改成功')
sql = "select * from book"
s5 = select_many(sql)
print('图书编号\t\t书名\t\t\t描述\t\t\t\t价格/元')
for i in range(len(s5)):
for j in range(len(s5[i])):
print(s5[i][j], end='\t\t')
print()
#书籍删除
def delete(self):
id = int(input('请输入要删除的书籍编号:'))
sql = f"select * from book where id='{id}'"
s6 = select_one(sql)
if s6 is None:
print('您输入的图书编号不存在,则无法进行删除')
else:
print('您要删除的信息如下:')
print('书名:', s6[1])
print('描述:', s6[2])
print('价格:', s6[3])
num1 = input(f'您确定删除图书编号为{id}的图书,是请输入Y/y:')
if num1 == 'Y' or 'y':
sql = f"delete from book where id='{id}'"
add_update_delete(sql)
print('恭喜您,删除成功')
sql = "select * from book"
s7 = select_many(sql)
print('图书编号\t\t书名\t\t\t描述\t\t\t\t价格/元')
for i in range(len(s7)):
for j in range(len(s7[i])):
print(s7[i][j], end='\t\t')
print()
else:
print('您已取消删除')
def caidan(self):
while True:
print('a.显示图书列表\tb.书籍添加\tc.书籍更新\td.书籍删除\tn.退出系统')
num=input('请选择操作的功能项:')
if num=='a':
s0.many()
elif num=='b':
s0.add()
elif num=='c':
s0.update()
elif num=='d':
s0.delete()
elif num=='n':
print('退出系统,欢迎下次使用')
break
else:
print('输入无效,请重新选择')
if __name__ == '__main__':
s0=Book()
print('图书管理系统'.center(60,'='))
s0.caidan()