app.py

import gradio as gr

import io
import datetime
from pytz import timezone

from login import catch_qr, get_uid_key
from nft import having_card_id_list, card_id_set_ava

GLOBAL_STATE_SUCCESS = 0
SUCCESS_UIDS = []

sid_map = {
    "SHN48荣耀时刻": 1,
    "无": 2,
    "无2": 3,
    "胶囊计划": 4,
    "天官赐福": 5,
    "A - AKB48TSH四周年": 6,
    "B - AKB48TSH四周年": 7,
    "C - AKB48TSH四周年": 8,
    "D - AKB48TSH四周年": 9,
    "E - AKB48TSH四周年": 10,
    "F - AKB48TSH四周年": 11,
    "G - AKB48TSH四周年": 12,
    "H - AKB48TSH四周年": 13,
    "三体动画": 14
}


def qr(ava_image):
    if ava_image:
        qr_img, info = catch_qr(ava_image)
        return {image_qr: qr_img, login_info: info}
    return {image_qr: None, login_info: None}


def verify_login(info, u_verify, u_id, u_key, sid):
    sid_num = sid_map[sid]
    code = "请刷新页面并重新扫码登录"
    if not u_verify:
        result, u_id, u_key, code = get_uid_key(info)
        if not result:
            return False, "", "", sid_num, code
        else:
            u_verify = True
    return u_verify, u_id, u_key, sid_num, code


def back_card_id_list(info, u_verify, u_id, u_key, sid):
    if info:
        u_verify, u_id, u_key, sid_num, code = verify_login(info, u_verify, u_id, u_key, sid)
        if u_verify:
            result, had_card_id_list, code = having_card_id_list(u_id, u_key, sid_num)
            if result:
                print("获取卡牌列表成功")
                card_name_list = list(had_card_id_list.keys())
                return {verify: u_verify, uid: u_id, key: u_key, card_id_list: had_card_id_list,
                        card_list_drop: gr.update(choices=card_name_list, value=card_name_list[0]), code_output: code}
            else:
                print("获取卡牌列表失败")
                return {verify: u_verify, uid: u_id, key: u_key, card_list_drop: gr.update(choices=[]),
                        code_output: code}
    else:
        return {code_output: "请先上传头像和扫码登录"}


def name_get_card_id(card_id_list_dict, card_list_drop_name):
    if card_id_list_dict and card_list_drop_name:
        for card_name in card_id_list_dict.keys():
            if card_name == card_list_drop_name:
                cid = card_id_list_dict[card_name]
                return {card_id: cid}
    return {card_id: ""}


def login_up_nft(info, u_verify, u_id, u_key, sid, cid, ava_image, log_into_public):
    global GLOBAL_STATE_SUCCESS, SUCCESS_UIDS
    if info and ava_image:
        u_verify, u_id, u_key, sid_num, code = verify_login(info, u_verify, u_id, u_key, sid)
        if u_verify:
            if not cid:
                result, had_card_id_list, code = having_card_id_list(u_id, u_key, sid_num)
                if had_card_id_list:
                    cid = list(had_card_id_list.values())[0]
            bytes_object = io.BytesIO()
            ava_image.save(bytes_object, format='PNG')
            img_data = bytes_object.getvalue()
            result, code = card_id_set_ava(cid, u_key, img_data)
            tz = timezone('Asia/Shanghai')
            now = datetime.datetime.now(tz)
            print("UID: ", u_id, " 提交完成: ", now.strftime('%Y-%m-%d %H:%M:%S'))
            if result:
                GLOBAL_STATE_SUCCESS = GLOBAL_STATE_SUCCESS + 1
                SUCCESS_UIDS.append({"uid": u_id, "time": now})
            return {verify: u_verify, uid: u_id, key: u_key, code_output: code + "\n" + "自本次更新/系统重启以来，已成功为用户设置 " + str(GLOBAL_STATE_SUCCESS) + " 次头像。感谢您使用此服务！"}
        else:
            return {verify: u_verify, uid: u_id, key: u_key, code_output: code}
    return {code_output: "请先上传头像和扫码登录"}

def status():
    global GLOBAL_STATE_SUCCESS, SUCCESS_UIDS
    text = "自本次更新/系统重启以来，已成功为用户设置 " + str(GLOBAL_STATE_SUCCESS) + " 次头像。"
    for u in SUCCESS_UIDS:
        text = text + "\n" + "UID: " + u["uid"] + " 于 " + u["time"].strftime('%Y-%m-%d %H:%M:%S') + " 成功设置数字头像。"
    return text

with gr.Blocks() as application:
    login_info = gr.State([])
    card_id_list = gr.State({})
    card_id = gr.State("")
    verify = gr.State(False)
    key = gr.State("")
    uid = gr.State("")
    with gr.Tab("BiliBili自定义数字头像"):
        gr.Markdown("第一步:")
        gr.Markdown("""获取一个最低R级的数字周边，多关注官方活动。如果列表中没有最新的卡组，请耐心等待兼容更新。""")
        gr.Markdown("第二步：")
        with gr.Row():
            with gr.Column():
                gr.Markdown(
                    """成功领取后，下方上传新头像，头像尽量为正方形并长宽小于1024像素，右上角画笔按钮可裁剪,

然后等待生成二维码并扫码登录进行授权(左上传头像，右扫码登录)""")
            image_ava = gr.Image(type="pil", label="上传头像")
            image_qr = gr.Image(type="pil", label="客户端扫码登录")
        gr.Markdown("第三步:")
        with gr.Row():
            gr.Markdown(
                """第二步完成后，选择拥有的卡组，并等待加载出所拥有的卡，再指定底卡，务必重新选择，否则无法工作

最后点击按钮<验证并提交头像>等待完成显示结果(成功可能需要时间审核)""")
            with gr.Row():
                sid_drop = gr.Dropdown(label="卡组", choices=list(sid_map.keys()), value="无")
                card_list_drop = gr.Dropdown(label="指定已有底卡")
                set_button = gr.Button("验证并提交头像")
            code_output = gr.Textbox(label="消息输出", placeholder="点击提交按钮才会提交")

    with gr.Accordion("警告"):
        gr.Markdown("""本服务基于bilibili未公开API制作，因此稳定性无法保证，敬请谅解。

本服务要求也必须登录&访问您的账号，我们无法保证您的账号不会被第三方劫持或盗用，但我们将竭尽全力保证您的数据和隐私安全。建议您在使用后及时修改密码，以退出所有设备的登陆状态，防止盗号风险。

本服务由方糖网络科技开发。""")

    gr.Markdown("服务状态:")
    with gr.Row():
        status_button = gr.Button("获取服务状态")
        status_output = gr.Textbox(label="服务状态", placeholder="点击获取按钮才会显示")

    image_ava.change(qr, inputs=[image_ava], outputs=[image_qr, login_info])
    sid_drop.change(back_card_id_list, inputs=[login_info, verify, uid, key, sid_drop],
                    outputs=[card_id_list, verify, uid, key, card_list_drop, code_output])
    card_list_drop.change(name_get_card_id, inputs=[card_id_list, card_list_drop], outputs=[card_id])
    set_button.click(login_up_nft, inputs=[login_info, verify, uid, key, sid_drop, card_id, image_ava],
                     outputs=[verify, uid, key, code_output])
    status_button.click(status, inputs=[], outputs=[status_output])

application.launch()