Python加密字符串或者文件

#!/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
    ```