nft.py

import time
import json
from pathlib import Path
import requests
from hashlib import md5
from typing import Union
from urllib.parse import urlencode
from requests_toolbelt.multipart.encoder import MultipartEncoder
import imghdr

CARD_INFO_PATH = Path(__file__).parent / "bili.json"  # 不要动


class Crypto:
    APPKEY = '4409e2ce8ffd12b8'
    APPSECRET = '59b43e04ad6965f34319062b478f83dd'

    @staticmethod
    def md5(data: Union[str, bytes]) -> str:
        '''generates md5 hex dump of `str` or `bytes`'''
        if type(data) == str:
            return md5(data.encode()).hexdigest()
        return md5(data).hexdigest()

    @staticmethod
    def sign(data: Union[str, dict]) -> str:
        '''salted sign funtion for `dict`(converts to qs then parse) & `str`'''
        if isinstance(data, dict):
            _str = urlencode(data)
        elif type(data) != str:
            raise TypeError
        return Crypto.md5(_str + Crypto.APPSECRET)


class SingableDict(dict):
    @property
    def sorted(self):
        '''returns a alphabetically sorted version of `self`'''
        return dict(sorted(self.items()))

    @property
    def signed(self):
        '''returns our sorted self with calculated `sign` as a new key-value pair at the end'''
        _sorted = self.sorted
        return {**_sorted, 'sign': Crypto.sign(_sorted)}


def get_image_type(file_path):
    with open(file_path, 'rb') as f:
        data = f.read()
    return imghdr.what(None, data)


def upload_image(file_path, ACCESS_KEY):
    url = "https://api.bilibili.com/x/upload/app/image?access_key=" + ACCESS_KEY

    payload = {'bucket': 'medialist', 'dir': 'nft'}

    with open(file_path, 'rb') as f:
        type = f'image/{imghdr.what(f)}'
        print(type)
        files = [
            (
                'file',
                (file_path, f, type),
            )
        ]
        response = requests.request("POST", url, data=payload, files=files)
        print(response.text)
        return response.json()['data']['location']


def get_one_card_id(UID, ACCESS_KEY):
    url = "https://api.bilibili.com/x/vas/nftcard/cardlist"
    params = SingableDict(
        {
            "access_key": ACCESS_KEY,
            "act_id": "14",  # 这里默认已经是三体数字藏品卡14，github下载的默认是4，也就是胶囊卡
            "appkey": "4409e2ce8ffd12b8",
            "disable_rcmd": "0",
            "ruid": UID,
            "statistics": "{\"appId\":1,\"platform\":3,\"version\":\"7.9.0\",\"abtest\":\"\"}",
            "ts": int(time.time()),
        }
    ).signed
    response = requests.request("GET", url, params=params)
    data = response.json()
    if data['code'] != 0:
        print(f"获取卡片信息出错，下面是 api 返回：\n{data}")
        return
    # print(data)  # 查询已添加无需再填
    with CARD_INFO_PATH.open("w", encoding="utf8") as f:
        json.dump(data, f, indent=4, ensure_ascii=False)
        print(f"已将卡片信息写入 {CARD_INFO_PATH}")
        for round in data['data']['round_list']:
            for card in round['card_list']:
                if card['card_type'] == 1 and card['card_id_list']:
                    print(card['card_id_list'][0]['card_id'])
                    return card['card_id_list'][0]['card_id']
        print('没有 R 级别胶囊计划的卡片')
        for i in data["data"]["pre_list"]:
            if card_id_list := i.get("card_id_list"):
                for j in card_id_list:
                    if card_id := j.get("card_id"):
                        print("=================================")
                        print(f"找到三体卡 card_id: {card_id}\n这个 id 属于 {i.get('card_name')}")
                        print("=================================")
                        return card_id
                    else:
                        print()  # 有值的 card_id_list 中没找到 card_id ，可能是没领三体卡吧
            else:
                print()  # 卡片信息中没找到有值的 card_id_list ，可能是没有这张三体卡吧


def set_face(card_id, ACCESS_KEY, img_data):
    api = "https://api.bilibili.com/x/member/app/face/digitalKit/update"
    params = SingableDict(
        {
            "access_key": ACCESS_KEY,
            "appkey": "4409e2ce8ffd12b8",
            "build": "7090300",
            "c_locale": "zh_CN",
            "channel": "xiaomi",
            "disable_rcmd": "0",
            "mobi_app": "android",
            "platform": "android",
            "s_locale": "zh_CN",
            "statistics": "{\"appId\":1,\"platform\":3,\"version\":\"7.9.0\",\"abtest\":\"\"}",
            "ts": int(time.time()),
        }
    ).signed
    m = MultipartEncoder(
        fields={
            'digital_kit_id': str(card_id),
            'face': ('face', img_data, 'application/octet-stream'),
        }
    )
    headers = {
        "Content-Type": m.content_type,
    }
    response = requests.request("POST", api, data=m, headers=headers, params=params)
    if response.json()['code'] != 0:
        return False, response.json()
    return True, '设置头像成功, 请等待审核'


def up_card(uid, key, img_data):
    UID = int(uid)
    ACCESS_KEY = str(key)
    result = False
    CODE = "超时"

    card_id = get_one_card_id(UID, ACCESS_KEY)  # 根据你所取得的卡card_id进行更改
    if not card_id:
        CODE = "没找到 card_id ,请确认是否已经领取卡片"
    # img_url = upload_image(BG_PATH)
    # set_bg_img(img_url, card_id)
    else:
        result, CODE = set_face(card_id, ACCESS_KEY, img_data)
    return result, CODE