Python文件操作-图书管理系统

简易版图书管理系统

# 用户注册
import time


def register():
    username = input('请输入用户名：')
    password = input('请输入密码：')
    repassword = input('请确认密码：')

    if password == repassword:
        # 保存信息
        with open(r'E:\book\user.txt', 'a') as wstream:
            wstream.write('{}   {}\n'.format(username, password))
        print('用户注册成功')
    else:
        print('密码不一致')


def login():
    username = input('请输入用户名：')
    password = input('请输入密码：')

    if username and password:
        with open(r'E:\book\user.txt') as rstream:
            while True:
                user = rstream.readline()
                if not user:
                    print('用户名或密码输入错误！')
                    break
                # print(user)
                input_user = '{}   {}\n'.format(username, password)
                if input_user == user:
                    print('登录成功')
                    break


def show_books():
    print('--------------图书馆里面的图书有-----------------')
    with open(r'E:\book\books.txt', 'r', encoding='utf-8') as restream:
        books = restream.readlines()
        for book in books:
            print(book, end='')


def borrow_book():
    username = input('请输入您的用户名：')
    book_name = input('请输入您需要借阅的书籍名称：')
    with open(r'E:\book\books.txt', 'r', encoding='utf-8') as restream:
        booklist = restream.readlines()
        # print(type(booklist[1]))
    for book in booklist:
        book0 = book.strip('\n')
        book = book0.split(' ')[0]
        # number = book0.split(' ')[1]
        # print(number)
        if book_name == book:
            print('借书成功')
            with open(r'E:\book\user_books.txt', 'a') as wstream:
                wstream.write(username)
                wstream.write(':')
                wstream.write(book_name)
                wstream.write('\n')
                #            with open(r'E:\book\books.txt','a') as
                break
    else:
        print('书籍不存在')


# 调用函数

while True:
    print('\n1.用户注册\n2.用户登录\n3.所有图书\n4.借书')
    n = input('请输入您的操作：')
    if n == '1':
        register()
    elif n == '2':
        login()
    elif n == '3':
        show_books()
    elif n == '4':
        borrow_book()
    else:
        print('正在退出本系统')
        time.sleep(3)
        print('退出成功！！！感谢您的使用')
        break

注意：特别注意strip()函数的应用（https://www.cnblogs.com/onebit123/p/16056793.html）