python实现注册登录

利用python实现简单的注册登录：

#注册
import json

def register():
    with open('user_db.json', 'r', encoding='utf-8')as f:
        user_data = json.load(f)
    user_list = {}
    print('please register new user')
    username = input('please input your username:')
    password = input('please input your password:')
    user_list['username'] = username
    user_list['password'] = password
    user_data.append(user_list)
    with open('user_db.json', 'w') as f:
        json.dump(user_data, f)


if __name__ == '__main__':
    register()

#登录
import json
import register_module

def login():
    user_flag = input('do you have a username? yes or no:')
    if user_flag == 'yes':
        username = input('username:')
        password = input('password:')
        with open('user_db.json', 'r') as f:
            user_info = json.load(f)
            for user in user_info:
                if user['username'] == username and user['password'] == password:
                    print('login is successful')
                else:
                    print('your information is wrong!')
    else:
        print('please register a username!')
        register_flag = input('do you want to register? yes or no:')
        if register_flag == 'yes':
            register_module.register()
        else:
            return 'login out'

if __name__ == '__main__':
    login()