【python】hashlib.shasha256练习注册 --笔记

'''
用户注册和登录
'''
import csv
import hashlib

#注册将用户信息填写到csv文件中
def register():
    username = input('用户名:')
    password = input('密码:')
    user = []
    user.append(username)
    user.append(hashlib.sha256(password.encode('utf-8')).hexdigest())
    with open('t2/users.csv', 'a', newline='') as ws:
        csv_ws = csv.writer(ws)
        csv_ws.writerow(user)
    print('注册成功！')

# 登陆 将密码加密转换  将文件打开  对比用户名and密码
def login(): 
    username = input('用户名:')
    password = input('密码:')

    password = hashlib.sha256(password.encode('utf-8')).hexdigest()

    with open('t2/users.csv', 'r') as rs:
        csv_rs = csv.reader(rs)
        for user in csv_rs:
            if username == user[0] and password == user[1]:
                print('登录成功！')
                break
        else:
            print('用户名或者密码有误~！')


if __name__ == '__main__':
    # register()
    login()