用户登录管理

# -*- coding:utf-8 -*-
login_user = {'is_login': False}


def register():
    user = input('请输入您的用户名：')
    file = open('text','r',encoding='utf-8')
    for line in file:
        line_list = line.strip().split(' ')
        if line_list[0] == user:
            print('该用户已存在，请使用其它用户名进行注册！')
            register()
    register2(user)


def register2(user):
    pwd = input('请输入您的密码：')
    pwd_2 = input('请再次输入您的密码：')
    if pwd != pwd_2:
        print('两次密码输入不一致，请重新输入！')
        register2(user)
    register3(user, pwd)


def register3(user, pwd):
    f = open('text','a',encoding='utf-8')
    f.write('
' + user + ' ' + pwd)
    f.close()
    print('注册成功！')
    main()


def login():
    user = input('请输入用户名：')
    pwd = input('请输入密码：')
    f = open('text','r')
    lines = f.readlines()
    for line in lines:
        line_list = line.strip().split(' ')
        if line_list[0] == user and line_list[1] == pwd:
            login_user['current_user'] = user
            login_user['current_pwd'] = pwd
            return True
    return False


def manager():
    if login_user['is_login']:
        print('欢迎%s登陆！' % login_user['current_user'])
    else:
        print('请先登录！')
        main()


def changepwd():
    if login_user['is_login']:
        login_user['new_pwd'] = input('请输入新密码：')
        delete()
        print('密码修改成功！')
    else:
        print('请先登录！')
        main()


def delete():
    lines = open('text','r').readlines()
    for i in range(len(lines)):
        if login_user['current_user'] in lines[i]:
            lines[i] = lines[i].replace(login_user['current_user'] + ' ' + login_user['current_pwd'],login_user['current_user'] + ' ' + login_user['new_pwd'])
            open('text','w').writelines(lines)


def main():
    while True:
        print('【1】.登陆   【2】.注册   【3】后台管理   【4】修改密码')
        choice = input('请输入序号进行登陆或注册：')
        if choice == '1':
            ret = login()
            if ret:
                print("登陆成功！")
                login_user['is_login'] = True
            else:
                print("用户名或密码错误，请重新登陆！")
        if choice == '2':
            register()
        if choice == '3':
            manager()
        if choice == '4':
            changepwd()

main()