#!/usr/bin/env python
# Version = 3.8.1
# -*- coding: utf-8 -*-
#Author: lumia98@vip.qq.com
"""
加盐加密, 判断传输的文件还是str字符串进行加密
"""
import hashlib
def encry_hash(args):
SALT = b'dtgrt345s90' # 盐
try:
#如果传人的是文件
read_file = open(args, 'rb')
print('我是文件啊')
obj = hashlib.sha256() # 实例化对象
while 1:
read_data = read_file.read(8096)
if read_data: # 如何读取到了数据,则校验md5
obj.update(read_data)
else: # 如果读取为空b'',退出循环
break
read_file.close()
data = obj.hexdigest()
except:
print('我是字符串')
obj = hashlib.sha256(SALT) # 实例化对象
obj.update(args.encode('utf-8'))
data = obj.hexdigest()
return data
```