python3----练习题(....)

import re
import hashlib
import json

def get_md5(pwd):     # md5加密
    m = hashlib.md5()
    m.update(pwd.encode('utf-8'))
    return m.hexdigest()

def get_name_md5():    # ‘加盐’加密
    hash = hashlib.md5('python'.encode('utf-8'))
    hash.update('admin'.encode('utf-8'))
    return hash.hexdigest()

def write_to_file(content):
    with open('dic.txt', 'a', encoding='utf-8') as f:
        f.write(json.dumps(content, ensure_ascii=False) + '
')
        f.close()


def regist():
    dd = {}
    print('{0:*^50}'.format('用户注册界面'))
    name = input("请输入你的用户名(以字母开头):")
    if re.match(r'^[a-zA-Z].{2,9}', name):
        pwd = input("请设置密码:")
        pwd = get_md5(pwd)
        dd[name] = pwd
        write_to_file(dd)
        print(dd)
    else:
        print("您输入的用户名格式不正确,请小于10位")
        regist()


def login():
    print('{0:*^50}'.format('用户登录界面'))
    name = input('帐号:')
    with open('dic.txt') as f:
        for line in iter(f):
            dic1 = json.loads(line)
            if name in dic1:
                pwd = input('密码:')
                pwd = get_md5(pwd)
                if dic1[name] == pwd:
                    print('登录成功')
                    return True
        print("帐号或密码错误!")
        login()

def main():
     login()

if __name__ == '__main__':
    main()