python图书管理系统

#coding=utf-8
import sys

books = [['红楼梦', '曹雪芹', 50]]
users = [['qwx', '123']]
#登录注册模块
def main():
    while True:
        print('''
            **********************'
             *欢迎来到图书管理系统*'
            **********************
        '*********************************'
        '******   登录-------1  **********'
        '******   注册-------2  **********'
        '******   退出-------0  **********'
        '*********************************\n''')
        try:
            v = int(input('请输入对应的数字：'))
            if v == 2:
                register()
            elif v == 1:
                login()
            elif v == 0:
                sys.exit(0)
        except:
            print('参数有误')
        finally:
            main()
#登录验证
flag = False
def login():
    print('请输入用户名：', end='')
    username = input()
    print('请输入用户密码：', end='')
    password = input()
    if [username, password] in users:
        print('登录成功！')
        flag = True
        menu()
    else:
        print('\n用户名不存在或密码错误！ 请重新登录！')
#注册账号
def register():
    username = input('请输入用户名：')
    password = input('请输入用户密码：')
    if [username, password] not in users：
        users.append([username, password])
        print('注册成功！')
    else：
        print（‘该账户已存在！’）
#主菜单
def menu():
    print(''''
    ***************************************
    ********  增加书籍--------1  **********
    ********  删除书籍--------2  **********
    ********  查找书籍--------3  **********
    ********  修改书籍--------4  **********
    ********  查看所有书籍----5  **********
    ********  返回主界面------6  **********
    ********  退出------------0  **********
    ***************************************\n''')
    try:
        i = int(input('请输入对应的数字：\n'))
        if i == 1:
            bk_add()
        elif i==2:
            bk_del()
        elif i == 3:
            bk_search()
        elif i == 4:
            bk_mod()
        elif i == 5:
            bk_view()
        elif i == 6:
            bk_menu()
        elif i == 0:
            bk_exit()
    except:
        print('参数有误1')
    finally:
        menu()
#添加书籍
def bk_add():
    flag = True
    while flag:
        bookname = input('请输入书名：')
        author = input('请输入作者：')
        price = int(input('请输入价格：'))
        ipt = input('是否添加该书籍信息[y/n]')
        if ipt=='y':
            if [bookname, author, price] not in books:
                books.append([bookname, author, price])
                print('\n添加书籍成功！')
            else:
                print('该书籍已存在，添加失败！')
        elif ipt == 'n':
            menu()
        else:
            print('参数有误')
        opt = input('是否继续添加？y/n：')
        if opt == 'y':
            continue
        elif opt == 'n':
            flag = False
            menu()
#删除书籍
def bk_del():
    flag = True
    while flag:
        bookname = input('请输入书名：')
        author = input('请输入作者：')
        price = int(input('请输入价格：'))
        if [bookname, author, price] in books:
            books.remove([bookname, author, price])
            print('\n删除书籍成功！')
        else:
            print('\n该书籍不存在！自动返回...')
        opt = input('是否继续删除？y/n：')
        if opt == 'y':
            continue
        elif opt == 'n':
            flag = False
            menu()
#查询书籍
def bk_search():
    flag = True
    while flag:
        bookname = input('请输入书名：')
        f=0
        for book in books:
            if bookname == book[0]:
                f=1
                print('书名：', book[0], '作者：', book[1], '价格：', book[2])
        if f == 0:
             print('该书籍不存在，自动返回...')
        print('\n查找结束!\n')
        opt = input('是否继续查询？y/n：')
        if opt == 'y':
            continue
        elif opt == 'n':
            flag = False
            menu()
#修改书籍
def bk_mod():
    flag=True
    while flag:
        bookname = input('请输入书名：')
        author = input('请输入作者：')
        price = int(input('请输入价格：'))
        if [bookname, author, price] in books:
            books.remove([bookname, author, price])
            bookname = input('请输入修改后的书名：')
            author = input('请输入修改后的作者：')
            price = int(input('请输入修改后的价格：'))
            books.append([bookname, author, price])
            print('\n修改书籍成功！')
        else:
            print('\n该书籍不存在！自动返回...')
        opt = input('是否继续修改？y/n')
        if opt == 'y':
            continue
        elif opt == 'n':
            flag = False
            menu()
#查看所有书籍
def bk_view():
    print('\n书名\t\t', '作者\t\t\t', '价格\n')
    for i in books:
        print(i[0], '\t\t', i[1], '\t\t', i[2])
#返回主界面
def bk_menu():
    main()
#退出系统
def bk_exit():
    sys.exit(0)

if __name__ == '__main__':
    main()