Python—简单图书管理系统

最近还在学Python，想着写一个简易的图书管理系统巩固一下学到的知识。
实现的功能有注册，登陆，查看图书，借书，还书。(还真的是简易!)
还需要增加的功能有永久保留数据，修改密码，以及界面的优化等。

1.初始书籍和用户

books = [['<<西游记>>', '吴承恩'], ['<<水浒传>>', '施耐庵'], ['<<三国演义>>', '罗贯中'], ['<<红楼梦>>', '曹雪芹']]
users = [['zxz', '123'], ['th', '456']]
flag = 0  #用来判断程序执行

2.注册
将注册信息保存在 users列表 中

#注册
def register():
    user = input('请输入用户名: ')
    password = input('请输入密码: ')
    users.append([user, password])
    print('注册成功!')

3.登陆
在登陆函数中调用书籍信息相关函数( ui() ),并在输入的用户或密码不正确时提醒用户再次输入(while循环实现)

#登陆
def login():
    f = 1
    while f:
        user = input('请输入用户名: ')
        password = input('请输入密码: ')
        if [user, password] in users:
            ui()
            f = 0  #循环结束
        else:
            print('用户名或密码错误!')

4.书籍信息( ui() )
登陆成功后进入此界面，实现查看书籍，借书，还书，离开系统这些功能。

#登陆后的界面
def ui():
    global flag
    flag = 1
    while flag:
        print('**********************')
        print('**********************')
        print('   欢迎登陆图书管理系统   ')
        print('*** 1.view   books ***')
        print('*** 2.borrow books ***')
        print('*** 3.return books ***')
        print('*** 4.leave system ***')
        print('**********************')
        print('**********************')
        num = int(input('请输入你的操作: '))
        #查看书籍
        if num == 1:
            print('作者', '\t', '书籍')
            for i in books:
                print(i[0], '\t', i[1])
        #借书
        elif num == 2:
            book = input('请输入你要借的书名: ')
            author = input('请输入所要借的书的作者: ')
            if [book, author] in books:
                books.remove([book, author])
                print('出库成功!')
         #还书       
        elif num == 3:
            book = input('请输入你要还的书名: ')
            author = input('请输入所要还的书的作者: ')
            books.append([book, author])
            print('入库成功!')
         #离开图书管理系统
        elif num == 4:
            exits()
        else:
            print('输入错误，请在输入!')

5.主界面
main ，调用执行程序

#主界面
def main():
    global flag
    flag = 1
    while flag:
        print('******************')
        print('******************')
        print(' 欢迎登陆图书管理系统 ')
        print('*** 1.register ***')
        print('*** 2.login    ***')
        print('*** 3.exits    ***')
        print('******************')
        print('******************')
        num = int(input('请输入操作数字:'))
        try:
            if num == 1:
                register()
            elif num == 2:
                login()
            elif num == 3:
                exits()
            else:
                print('输入错误，请重新输入！')
        except ValueError:
            print('输入错误，请重新输入！')

6.完整的代码

books = [['<<西游记>>', '吴承恩'], ['<<水浒传>>', '施耐庵'], ['<<三国演义>>', '罗贯中'], ['<<红楼梦>>', '曹雪芹']]
users = [['zxz', '123'], ['th', '456']]
flag = 0


#注册
def register():
    user = input('请输入用户名: ')
    password = input('请输入密码: ')
    users.append([user, password])
    print('注册成功!')


#登陆
def login():
    f = 1
    while f:
        user = input('请输入用户名: ')
        password = input('请输入密码: ')
        if [user, password] in users:
            ui()
            f = 0
        else:
            print('用户名或密码错误!')


#登陆后的界面
def ui():
    global flag
    flag = 1
    while flag:
        print('**********************')
        print('**********************')
        print('   欢迎登陆图书管理系统   ')
        print('*** 1.view   books ***')
        print('*** 2.borrow books ***')
        print('*** 3.return books ***')
        print('*** 4.leave system ***')
        print('**********************')
        print('**********************')
        num = int(input('请输入你的操作: '))
        if num == 1:
            print('作者', '\t', '书籍')
            for i in books:
                print(i[0], '\t', i[1])
        elif num == 2:
            book = input('请输入你要借的书名: ')
            author = input('请输入所要借的书的作者: ')
            if [book, author] in books:
                books.remove([book, author])
                print('出库成功!')
        elif num == 3:
            book = input('请输入你要还的书名: ')
            author = input('请输入所要还的书的作者: ')
            books.append([book, author])
            print('入库成功!')
        elif num == 4:
            exits()
        else:
            print('输入错误，请在输入!')


#离开系统
def exits():
    global flag
    flag = 0
    print('欢迎再次使用图书管理系统!')


#主界面
def main():
    global flag
    flag = 1
    while flag:
        print('******************')
        print('******************')
        print(' 欢迎登陆图书管理系统 ')
        print('*** 1.register ***')
        print('*** 2.login    ***')
        print('*** 3.exits    ***')
        print('******************')
        print('******************')
        num = int(input('请输入操作数字:'))
        try:
            if num == 1:
                register()
            elif num == 2:
                login()
            elif num == 3:
                exits()
            else:
                print('输入错误，请重新输入！')
        except ValueError:
            print('输入错误，请重新输入！')


main()

以后还会再继续优化！