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('\n')
        if n==1:
                regist()
        if n==2:
                login()
        if n==3:
                SystemExit()
#begin()
add()