Class InviteAPI

tianyi.py:193–435

class InviteAPI:
    BASE_URL = "https://ai.imusic.cn"

    def __init__(self, token: str, cookies: dict = None):
        self.token = token
        self.session = requests.Session()
        self.session.mount('https://', DESAdapter())
        if cookies:
            self.session.cookies.update(cookies)
        self.session.headers.update({
            "User-Agent": "CtClient;11.3.0;Android;12;Redmi K30 Pro;ODAwODUw!#!MTg2MDg",
            "Accept": "application/json, text/plain, */*",
            "Accept-Encoding": "gzip, deflate, br, zstd",
            "Origin": self.BASE_URL,
            "Authorization": f"Bearer {token}",
            "Referer": f"{self.BASE_URL}/h5v/fusion/ai-luck-flow?ca=AP3V&cc={CHANNEL_ID}&utm_scha=utm_ch-010001002009.utm_sch-hg_xx_qlxx-1-104705800001-105782800001.utm_af-1000000037.utm_as-158492900001.utm_sd1-default",
            "X-Requested-With": "com.ct.client",
            "sec-ch-ua": '"Chromium";v="140", "Not=A?Brand";v="24", "Android WebView";v="140"',
            "sec-ch-ua-mobile": "?1",
            "sec-ch-ua-platform": '"Android"',
            "Sec-Fetch-Site": "same-origin",
            "Sec-Fetch-Mode": "cors",
            "Sec-Fetch-Dest": "empty",
            "Accept-Language": "zh-CN,zh;q=0.9,en-US;q=0.8,en;q=0.7"
        })



    def request_plain(self, endpoint: str, params: dict = None) -> dict:
        url = f"{self.BASE_URL}{endpoint}"
        response = self.session.post(url, params=params, timeout=30)
        return response.json()

    def request_encrypted(self, endpoint: str, params: dict) -> dict:
        self.session.headers["imencrypt"] = "1"
        imrandomnum = generate_random_string(16)
        imtimestamp = str(int(time.time() * 1000))
        imencryptkey = encryptmd5(imtimestamp, imrandomnum)

        form_data = {"channelId": CHANNEL_ID, "portal": "45"}
        form_data.update(params)
        encrypted_form = encrypt_request(form_data, imrandomnum, imtimestamp, imencryptkey)

        headers = {
            "imencryptkey": imencryptkey,
            "imrandomnum": imrandomnum,
            "imtimestamp": imtimestamp,
        }
        url = f"{self.BASE_URL}{endpoint}"
        response = self.session.post(url, params={"formData": encrypted_form}, headers=headers, timeout=30)

        new_auth = response.headers.get("Authorization")
        if new_auth:
            self.token = new_auth
            self.session.headers["Authorization"] = f"Bearer {new_auth}"

        try:
            return decrypt_response(response.text, imrandomnum, imtimestamp, imencryptkey)