import hashlib #导入模块

######## md5 ###########

m = hashlib.md5() #创建一个对象

m.update(b"hhf") #加密字符串 b代表byte，是把字符串转换成byte类型，也可以用bytes()强制转换

#m.update(bytes("hhf",encoding='utf-8')) #不指定encoding会报错，TypeError: Unicode-objects must be encoded before hashing

#bytes() 强制转换为字节类型

print(m.digest()) #取加密后字符串 转换成2进制格式hash

print(m.hexdigest()) #16进制格式hash，比较常用

###########-sha1 ########

hash = hashlib.sha1()

hash.update(b'admin')

print(hash.hexdigest())

# # ######## sha256 ########

hash = hashlib.sha256()

hash.update(b'admin')

print(hash.hexdigest())

# # ######## sha384 ########

hash = hashlib.sha384()

hash.update(b'admin')

print(hash.hexdigest())

# # ######## sha512 ########

hash = hashlib.sha512()

hash.update(b'admin')

print(hash.hexdigest())

#以上的加密方法都无法解密

import hashlib

hash = hashlib.md5()#md5对象，md5不能反解，但是加密是固定的，就是关系是一一对应，所以有缺陷，可以被对撞出来

hash.update(bytes('admin',encoding='utf-8'))#要对哪个字符串进行加密，就放这里

print(hash.hexdigest())#拿到加密字符串

# hash2=hashlib.sha384()#不同算法，hashlib很多加密算法

# hash2.update(bytes('admin',encoding='utf-8'))

# print(hash.hexdigest())

hash3 = hashlib.md5(bytes('abd',encoding='utf-8'))

''' 如果没有参数，所以md5遵守一个规则，生成同一个对应关系，如果加了参数，

就是在原先加密的基础上再加密一层，这样的话参数只有自己知道，防止被撞库，

因为别人永远拿不到这个参数

'''

hash3.update(bytes('admin',encoding='utf-8'))

print(hash3.hexdigest())

#输出

96270535571b70a9c42cedda1851d57c

827a2329d20080baa7f53bdfc7031425

#hashlib简单使用

def md5(arg):#这是加密函数，将传进来的函数加密

    md5_pwd = hashlib.md5(bytes('abd',encoding='utf-8'))

    md5_pwd.update(bytes(arg,encoding='utf-8'))

    return md5_pwd.hexdigest()#返回加密的数据

def log(user,pwd):#登陆时候时候的函数，由于md5不能反解，因此登陆的时候用正解

    with open('db','r',encoding='utf-8') as f:

        for line in f:

            u,p=line.strip().split('|')

            if u ==user and p == md5(pwd):#登陆的时候验证用户名以及加密的密码跟之前保存的是否一样

                return True

def register(user,pwd):#注册的时候把用户名和加密的密码写进文件，保存起来

    with open('db','a',encoding='utf-8') as f:

        temp = user+'|'+md5(pwd)

        f.write(temp)

i=input('1表示登陆，2表示注册：')

if i=='2':

    user = input('用户名：')

    pwd =input('密码：')

    register(user,pwd)

elif i=='1':

    user = user = input('用户名：')

    pwd =input('密码：')

    r=log(user,pwd)#验证用户名和密码

    if r ==True:

        print('登陆成功')

    else:

        print('登陆失败')

else:

    print('账号不存在')