topic_spider/zgztb_cookie/utils/tools.py

import hashlib


def encrypt(text: str):
    """
    十六进制数字文本摘要值

    @param text: 字符串文本
    @return: 十六进制编码
    """
    _hash = hashlib.sha256()
    _hash.update(text.encode("utf-8"))
    return _hash.hexdigest()


def redis_exists(href: str, redis_cluster):
    href_sign = encrypt(href)
    list_sign = 'pylist_' + href_sign
    if redis_cluster.exists(href_sign) > 0 or redis_cluster.exists(list_sign) > 0:
        return True
    return False


def redis_set(href: str, redis_cluster, val=1):
    if not redis_exists(href, redis_cluster):
        sign = 'pylist_' + encrypt(href)
        expiration_time = 86400 * 365 * 2
        redis_cluster.set(sign, val, ex=expiration_time)
        return True
    return False