python大作业-图书管理系统

#缺少循环执行和错误处理
#add（）函数 添加了循环执行
#错误处理：regist（）函数 登录和退出选择的时候添加了错误处理
import sys
import importlib
importlib.reload(sys)
Book=[['高数','高斯',30],['龙王传说','唐家三少',30],['霸道总裁爱上我','熏紫樱桃',30]]#列表1：书籍数据系统
User=[['admin',1],['manager',1]] #列表2：用户数据系统
def regist():#监测注册用户名，然后注册
        UserName1=input('请输入你的注册用户名：')
        for i in User:
                if i[0]==UserName1:
                        print('该用户名已被注册，请重新输入！')
                else:
                        Pwd1=int(input('恭喜你，用户名可以使用，请输入你的密码:'))
                        User.append([UserName1,Pwd1])
                        print("恭喜你注册成功，您的用户名为：",UserName1,"密码为",Pwd1)
                        try:#这里加错误处理了哦

                                j=int(input("请选择 1.登录 2.退出"))
                                if j==1:
                                        login()
                                if j==2:
                                        SystemExit()
                        except ValueError  as e:
                                print('输入的不正确哦，只能输入整数！')
                        finally:
                                begin()
def login():#先输入用户名然后输入密码，循环列表，找到登录成功没有登录失败
        UserName2=input('请输入你的登录用户名：')
        for i in User:
                if i[0]==UserName2:
                        pwds=i[1]
                        pwd2=int(input('请输入你的登录密码：'))
                        if pwd2==pwds:
                                print('亲爱的%s ，很高兴见到你，今天又是元气满满充满希望的一天！'%UserName2)
                                opt()

def opt():#登录后选择操作项
        s=input('请选择你要执行的操作，1=增 2=删 3=改 4=查 5=退出')
        if s==1:
                add()
        elif s==2:
                delete()
        elif s==3:
                edit()
        elif s==4:
                query()
        else:
                SystemExit()

def add():#输入书名、作者、价格，用append（）方法将参数添加到列表结尾，添加成功，失败错误处理
        j=1
        while (j!=2):#循环执行添加书本

                Bookname=input('请输入要添加的书本名：')
                author=input('请输入要添加书本的作者：')
                price=int(input('请输入要添加书本的价格：'))
                AddBook=[Bookname,author,price]
                Book.append(AddBook)
                print('添加书本成功了哦！')
                j=int(input('小宝贝，还要继续添加添加书本吗？ 1.继续 2.退出'))
                if j==2:
                        opt()

def delete():#输入书名，选择是否删除
        Bookname=input('请输入要删除的书本名：')
        j=-1
        for i in Book:
                j+=1
                if i[0]==Bookname:
                        print("已找到书本：%s"%Bookname)
                        break
        i=int(input('真的要删除这本书吗？ 1：确定 2：取消'))
        if i==1:
                        del Book[j]
                        print('ok')
                        
                

def edit():#只能修改书本的价格，先输入书名然后查询，查到了输入要修改的价格，没有查到循环继续输入或退出
        Bookname=input('请输入要编辑书本名称：')
        j=-1
        for i in Book:
                j+=1
                if i[0]==Bookname:
                        print("已找到书本：%s"%Bookname)
                        break
        i=int(input('真的要修改这本书吗？ 1：确定 2：取消'))
        if i==1:
                price=int(input('请输入修改后的价格：'))
                Book[j][2]=price
                print('ok')
        

def query():#查询，输入所有的书本信息
        for i in Book:
                print("序号：%s 书本信息：%s"%(Book.index(i)+1,i))

def begin():
        n=int(input('欢迎来到图书管理系统，1=注册 2=登录 3=关闭'))#3关闭  开始
        print('
')
        if n==1:
                regist()
        if n==2:
                login()
        if n==3:
                SystemExit()
#begin()
add()