Python Python中的加密工具

Python中的加密工具

hashlib模块介绍

• 难破解
• 不可逆

hashlib模块中的常用方法

代码

# coding:utf-8

import hashlib
import time

base_sigh = 'muke'


def custom():
    a_timestamp = int(time.time())
    _token = '%s%s' % (base_sigh, a_timestamp)
    # print(_token, type(_token))
    hashobj = hashlib.sha1(_token.encode('utf-8'))
    a_token = hashobj.hexdigest()
    # print(a_token)
    return a_token, a_timestamp


def b_service_check(token, timestamp):
    _token = '%s%s' % (base_sigh, timestamp)
    b_token = hashlib.sha1(_token.encode('utf-8')).hexdigest()
    if token == b_token:
        return True
    else:
        return False


if __name__ == '__main__':
    need_help_token, timestamp = custom()  # 若只使用一个变量,则此处会生成一个元组传入其变量
    time.sleep(1)
    result = b_service_check(need_help_token, int(time.time()))
    if result == True:
        print('a合法,b服务可以进行帮助')
    else:
        print('a不合法,b不可进行帮助')

base64模块介绍

• 通用型
• 可解密

base64模块的常用方法

代码

# coding:utf-8

import base64

replace_one = '%'
replace_two = '$'


def encode(data):
    if isinstance(data, str):
        data = data.encode('utf-8')
    elif isinstance(data, bytes):
        data = data
    else:
        raise TypeError('data need bytes or str')

    _data = base64.encodebytes(data).decode('utf-8')
    # print(_data)
    _data = _data.replace('a', replace_one).replace('2', replace_two)
    # print(_data)
    return _data


def decode(data):
    if not isinstance(data, bytes):
        raise TypeError('data need bytes')
    replace_one_b = replace_one.encode('utf-8')
    replace_two_b = replace_two.encode('utf-8')
    data = data.replace(replace_one_b, b'a').replace(replace_two_b, b'2')
    return base64.decodebytes(data).decode('utf-8')


if __name__ == '__main__':
    result = encode('hello xiaomu')
    print(result)
    new_result = decode(result.encode('utf-8'))
    print(new_result)