zhenxun_bot/plugins/genshin/query_user/utils/__init__.py

from configs.config import Config
import json
import time
import random
import hashlib
import string


def _md5(text):
    md5 = hashlib.md5()
    md5.update(text.encode())
    return md5.hexdigest()


def get_old_ds() -> str:
    n = Config.get_config("genshin", "n")
    i = str(int(time.time()))
    r = ''.join(random.sample(string.ascii_lowercase + string.digits, 6))
    c = _md5("salt=" + n + "&t=" + i + "&r=" + r)
    return i + "," + r + "," + c


def get_ds(q: str = "", b: dict = None) -> str:
    if b:
        br = json.dumps(b)
    else:
        br = ""
    s = Config.get_config("genshin", "salt")
    t = str(int(time.time()))
    r = str(random.randint(100000, 200000))
    c = _md5("salt=" + s + "&t=" + t + "&r=" + r + "&b=" + br + "&q=" + q)
    return t + "," + r + "," + c


def random_hex(length: int) -> str:
    result = hex(random.randint(0, 16 ** length)).replace("0x", "").upper()
    if len(result) < length:
        result = "0" * (length - len(result)) + result
    return result


element_mastery = {
    "anemo": "风",
    "pyro": "火",
    "geo": "岩",
    "electro": "雷",
    "cryo": "冰",
    "hydro": "水",
    "dendro": "草",
    "none": "无",
}