Upload app.py

app.py

@@ -0,0 +1,275 @@
+import requests
+import json
+import datetime
+import time # 用于时间转换
+import gradio as gr
+
+# --- 抖音配置信息 ---
+DOUYIN_CLIENT_KEY = 'awbeykzyos7kbidv'
+DOUYIN_CLIENT_SECRET = '4575440b156ecbe144284e4f69d284a2' # Keep secure!
+DOUYIN_ACCOUNT_ID = '7241078611527075855'
+DOUYIN_TOKEN_URL = 'https://open.douyin.com/oauth/client_token/'
+DOUYIN_ORDER_QUERY_URL = 'https://open.douyin.com/goodlife/v1/trade/order/query/'
+
+# --- 飞书配置信息 ---
+FEISHU_APP_ID = "cli_a6672cae343ad00e"
+FEISHU_APP_SECRET = "0J4SpfBMeIxJEOXDJMNbofMipRgwkMpV" # Keep secure!
+FEISHU_APP_TOKEN = "GcTMbpdDPaAxV5sb8gvcRVC3ndh"
+
+# 表格 1 (订单记录)
+FEISHU_TABLE1_ID = "tblaYUUrFT1rKXNB"
+FEISHU_TABLE1_ORDER_FIELD = "订单号"      # 文本类型
+FEISHU_TABLE1_STATUS_FIELD = "履约状态" # 文本类型 - 将被更新为 "已兑换"
+TABLE1_EXCHANGED_STATUS_CONTENT = "已兑换" # Table 1 更新后的状态
+
+# 表格 2 (激活码/发送记录)
+FEISHU_TABLE2_ID = "tbl0S1kl9FTBX8oT"
+FEISHU_TABLE2_ACTIVATION_CODE_FIELD = "激活码" # Table 2 中激活码字段名 (假设为文本)
+FEISHU_TABLE2_STATUS_FIELD = "发送状态"       # Table 2 中发送状态字段名 (文本类型)
+FEISHU_TABLE2_SORT_FIELD = "最后更新时间"     # Table 2 中用于排序的字段名
+FEISHU_TABLE2_UPDATE_TIME_FIELD = "最后更新时间" # Table 2 中最后更新日期字段名
+TABLE2_UNSENT_STATUS_CONTENT = "未发送" # Table 2 中查找的状态
+TABLE2_SENT_STATUS_CONTENT = "已发送"     # Table 2 更新后的状态
+
+# --- 抖音券（item）状态映射字典 ---
+ITEM_STATUS_MAP = {
+    0: "初始化", 1: "交易成功", 10: "待支付", 20: "支付成功",
+    100: "待使用", 101: "交易关闭", 200: "预约中", 201: "已预约",
+    300: "退款中", 301: "已退款", 400: "履约中", 401: "已履约"
+}
+
+# =========== 必须包含所有 API 函数定义 ===========
+
+def get_douyin_access_token(client_key, client_secret):
+    headers = {'Content-Type': 'application/json'}
+    payload = {"grant_type": "client_credential", "client_key": client_key, "client_secret": client_secret}
+    try:
+        response = requests.post(DOUYIN_TOKEN_URL, headers=headers, json=payload, timeout=10); response.raise_for_status(); data = response.json()
+        if data.get('data') and data['data'].get('error_code') == 0: return data['data'].get('access_token'), "成功获取抖音 Access Token！"
+        else: msg = f"获取抖音 Access Token 失败: Code={data.get('data', {}).get('error_code', 'N/A')}, Desc={data.get('data', {}).get('description', '未知错误')}"; return None, msg
+    except Exception as e: msg = f"请求抖音 Access Token 时发生错误: {e}"; return None, msg
+
+def query_douyin_order(access_token, account_id, order_id_str):
+    if not access_token: return None, "抖音 Token 无效"
+    headers = {'Content-Type': 'application/json', 'access-token': access_token}
+    params = {'account_id': account_id, 'order_id': order_id_str, 'page_num': 1, 'page_size': 1}
+    try:
+        response = requests.get(DOUYIN_ORDER_QUERY_URL, headers=headers, params=params, timeout=15); response.raise_for_status(); data = response.json()
+        extra_error_code = data.get('extra', {}).get('error_code', 0); data_error_code = data.get('data', {}).get('error_code', 0)
+        if extra_error_code == 0 and data_error_code == 0: return data, f"抖音订单 '{order_id_str}' 查询成功！"
+        else: error_code = extra_error_code if extra_error_code != 0 else data_error_code; description = data.get('extra', {}).get('description', '') or data.get('data', {}).get('description', '未知错误'); msg = f"抖音订单查询失败: Code={error_code}, Desc={description}"; return None, msg
+    except Exception as e: msg = f"查询抖音订单时发生错误: {e}"; return None, msg
+
+def parse_douyin_order_status(order_data):
+    try:
+        if not order_data or 'data' not in order_data or 'orders' not in order_data['data'] or not order_data['data']['orders']: return None, None, "抖音订单数据无效。"
+        order = order_data['data']['orders'][0]; certificates = order.get('certificate', [])
+        if not certificates: return None, None, "订单中未找到券信息。"
+        first_cert_status_code = certificates[0].get('item_status'); certificate_id = certificates[0].get('certificate_id', 'N/A')
+        if first_cert_status_code is not None: status_text = ITEM_STATUS_MAP.get(first_cert_status_code, f"未知状态码({first_cert_status_code})"); msg=f"解析到抖音订单状态: {status_text} (状态码: {first_cert_status_code}, 券ID: {certificate_id})"; return status_text, first_cert_status_code, msg
+        else: return None, None, "第一个券信息中未找到 'item_status'。"
+    except Exception as e: msg = f"解析抖音订单状态时出错: {e}"; return None, None, msg
+
+def get_feishu_tenant_access_token(app_id, app_secret):
+    auth_url = "https://open.feishu.cn/open-apis/auth/v3/tenant_access_token/internal"
+    payload = json.dumps({"app_id": app_id, "app_secret": app_secret})
+    headers = {'Content-Type': 'application/json; charset=utf-8'}
+    try:
+        response = requests.post(auth_url, headers=headers, data=payload, timeout=10); response.raise_for_status(); data = response.json()
+        if data.get("code") == 0: return data.get("tenant_access_token"), "获取飞书 tenant_access_token 成功！"
+        else: msg = f"获取飞书 tenant_access_token 失败: {data.get('msg')} (Code: {data.get('code')})"; return None, msg
+    except Exception as e: msg = f"请求飞书 tenant_access_token 时发生错误: {e}"; return None, msg
+
+def check_order_exists_in_feishu(access_token, app_token, table_id,
+                                 order_field_name, order_id_str_to_check):
+    if not access_token: return None, "飞书 Access Token 无效，无法检查记录。"
+    search_url = f"https://open.feishu.cn/open-apis/bitable/v1/apps/{app_token}/tables/{table_id}/records/search"
+    headers = {'Authorization': f'Bearer {access_token}', 'Content-Type': 'application/json; charset=utf-8'}
+    filter_payload = {"filter": {"conjunction": "and", "conditions": [{"field_name": order_field_name, "operator": "is", "value": [ order_id_str_to_check ]}]}, "page_size": 1}
+    msg = f"正在 Table 1 ({table_id}) 中检查订单号 '{order_id_str_to_check}'..."
+    try:
+        response = requests.post(search_url, headers=headers, json=filter_payload, timeout=15); response.raise_for_status(); data = response.json()
+        if data.get("code") == 0:
+            items = data.get("data", {}).get("items", [])
+            exists = bool(items)
+            msg += f" 检查结果: {'已存在' if exists else '不存在'}。"
+            return exists, msg # 返回布尔值和消息
+        else:
+            msg += f" 检查失败: Code={data.get('code')}, Msg={data.get('msg')}"
+            return None, msg # 返回 None 表示检查出错
+    except Exception as e:
+        msg += f" 检查时发生错误: {e}"
+        return None, msg
+
+def add_bitable_record(access_token, app_token, table_id,
+                       order_field_name, order_id_str,
+                       status_field_name, exchange_status):
+    if not access_token: return None, "飞书 Access Token 无效，无法写入数据。"
+    write_url = f"https://open.feishu.cn/open-apis/bitable/v1/apps/{app_token}/tables/{table_id}/records"
+    headers = {'Authorization': f'Bearer {access_token}', 'Content-Type': 'application/json; charset=utf-8'}
+    payload = {"fields": {order_field_name: order_id_str, status_field_name: exchange_status}}
+    msg = f"准备写入 Table 1 ({table_id})..."
+    try:
+        response = requests.post(write_url, headers=headers, json=payload, timeout=15); response.raise_for_status(); data = response.json()
+        if data.get("code") == 0:
+            record_info = data.get('data', {}).get('record')
+            record_id = record_info.get('record_id') if record_info else None
+            msg += f" Table 1 写入成功！Record ID: {record_id}"
+            return record_id, msg # 返回 record_id 和消息
+        else:
+            msg += f" Table 1 写入失败: Code={data.get('code')}, Msg={data.get('msg')}"
+            return None, msg
+    except Exception as e:
+        msg += f" Table 1 写入时发生错误: {e}"
+        return None, msg
+
+# --- 修改：查找 Table 2 记录，并只提取激活码文本 ---
+def find_first_unsent_record_and_code_text(access_token, app_token, table_id,
+                                           status_field, unsent_status_value,
+                                           activation_code_field, sort_field):
+    """
+    在 Table 2 查找第一个状态为 '未发送' 的记录，返回其 record_id 和激活码文本。
+    """
+    if not access_token: print("内部错误: 飞书 Token 无效"); return None, None
+    search_url = f"https://open.feishu.cn/open-apis/bitable/v1/apps/{app_token}/tables/{table_id}/records/search"
+    headers = {'Authorization': f'Bearer {access_token}', 'Content-Type': 'application/json; charset=utf-8'}
+    payload = {
+        "filter": {"conjunction": "and", "conditions": [{"field_name": status_field, "operator": "is", "value": [unsent_status_value]}]},
+        "sort": [{"field_name": sort_field, "order": "asc"}],
+        "page_size": 1,
+        # 【修改这里】: 直接传递 Python 列表，不要 json.dumps()
+        "field_names": [activation_code_field]
+    }
+    print(f"飞书API: 查找 Table 2 '{status_field}' 为 '{unsent_status_value}' 的记录...") # 后台日志
+    # print("构造的查找 Payload:", json.dumps(payload, indent=2, ensure_ascii=False)) # 调试时可以打印看看
+    try:
+        response = requests.post(search_url, headers=headers, json=payload, timeout=15); response.raise_for_status(); data = response.json()
+        if data.get("code") == 0:
+            items = data.get("data", {}).get("items", [])
+            if items:
+                # ... (后续解析激活码的逻辑不变) ...
+                record = items[0]; record_id = record.get('record_id')
+                activation_code_raw = record.get('fields', {}).get(activation_code_field)
+                activation_code_text = None
+                if isinstance(activation_code_raw, list) and len(activation_code_raw) > 0 and isinstance(activation_code_raw[0], dict) and 'text' in activation_code_raw[0]:
+                    activation_code_text = activation_code_raw[0]['text']
+                elif isinstance(activation_code_raw, str):
+                    activation_code_text = activation_code_raw
+                elif activation_code_raw is not None:
+                     try: activation_code_text = str(activation_code_raw); print(f"警告: 激活码字段值类型为 {type(activation_code_raw)}, 已尝试转为字符串。")
+                     except: print(f"警告: 无法将激活码字段值 {activation_code_raw} 转为字符串。")
+                else: print(f"警告: 激活码字段值为空或格式未知: {activation_code_raw}")
+
+                if record_id and activation_code_text is not None:
+                     print(f"飞书API: 找到记录 ID: {record_id}, 激活码文本: {activation_code_text}"); return record_id, activation_code_text
+                elif record_id: print(f"飞书API: 找到记录 ID: {record_id}, 但未能提取激活码文本。"); return record_id, None
+                else: return None, None
+            else:
+                print(f"飞书API: 未找到 '{status_field}' 为 '{unsent_status_value}' 的记录。"); return None, None
+        else:
+            print(f"飞书API: 查找 Table 2 记录失败: Code={data.get('code')}, Msg={data.get('msg')}"); return None, None
+    except Exception as e:
+        print(f"飞书API: 查找 Table 2 记录时发生错误: {e}"); return None, None
+
+def update_table2_record_to_sent(access_token, app_token, table_id, record_id,
+                                  status_field, sent_status_content,
+                                  update_time_field):
+    if not access_token or not record_id: return False, "飞书 Token 或 Table 2 Record ID 无效，无法更新。"
+    update_url = f"https://open.feishu.cn/open-apis/bitable/v1/apps/{app_token}/tables/{table_id}/records/batch_update"
+    headers = {'Authorization': f'Bearer {access_token}', 'Content-Type': 'application/json; charset=utf-8'}
+    now = datetime.datetime.now(); formatted_time_str = now.strftime("%Y/%m/%d %H:%M")
+    payload_fields = {status_field: sent_status_content, update_time_field: formatted_time_str}
+    payload = {"records": [{"record_id": record_id, "fields": payload_fields}]}
+    msg = f"准备更新 Table 2 ({table_id}, ID: {record_id}) 为 '{sent_status_content}'..."
+    try:
+        response = requests.post(update_url, headers=headers, json=payload, timeout=15); response.raise_for_status(); data = response.json()
+        if data.get("code") == 0:
+            failed_records = data.get("data", {}).get("failed_records", [])
+            if not failed_records: msg += " Table 2 更新成功！"; return True, msg
+            else: msg += f" Table 2 更新失败: {failed_records}"; return False, msg
+        else: msg += f" Table 2 更新失败: Code={data.get('code')}, Msg={data.get('msg')}"; return False, msg
+    except Exception as e: msg += f" Table 2 更新时发生错误: {e}"; return False, msg
+
+def update_table1_record_to_exchanged(access_token, app_token, table_id, record_id,
+                                      status_field, exchanged_status_content):
+    if not access_token or not record_id: return False, "飞书 Token 或 Table 1 Record ID 无效，无法更新。"
+    update_url = f"https://open.feishu.cn/open-apis/bitable/v1/apps/{app_token}/tables/{table_id}/records/batch_update"
+    headers = {'Authorization': f'Bearer {access_token}', 'Content-Type': 'application/json; charset=utf-8'}
+    payload = {"records": [{"record_id": record_id, "fields": {status_field: exchanged_status_content}}]}
+    msg = f"准备更新 Table 1 ({table_id}, ID: {record_id}) 为 '{exchanged_status_content}'..."
+    try:
+        response = requests.post(update_url, headers=headers, json=payload, timeout=15); response.raise_for_status(); data = response.json()
+        if data.get("code") == 0:
+            failed_records = data.get("data", {}).get("failed_records", [])
+            if not failed_records: msg += " Table 1 更新成功！"; return True, msg
+            else: msg += f" Table 1 更新失败: {failed_records}"; return False, msg
+        else: msg += f" Table 1 更新失败: Code={data.get('code')}, Msg={data.get('msg')}"; return False, msg
+    except Exception as e: msg += f" Table 1 更新时发生错误: {e}"; return False, msg
+
+
+# --- Gradio 主处理函数 (简化版) ---
+def process_order_simple(douyin_order_id_str):
+    final_output = "处理中..."
+    # --- 1. 获取抖音 Token ---
+    douyin_token, msg = get_douyin_access_token(DOUYIN_CLIENT_KEY, DOUYIN_CLIENT_SECRET)
+    if not douyin_token: return f"错误(抖音Token): {msg}"
+    # --- 2. 验证输入 ---
+    douyin_order_id_str = douyin_order_id_str.strip()
+    if not douyin_order_id_str: return "请输入抖音订单号"
+    # --- 3. 查询抖音订单 ---
+    order_details, msg = query_douyin_order(douyin_token, DOUYIN_ACCOUNT_ID, douyin_order_id_str)
+    if not order_details: return f"错误(查询订单): {msg}"
+    # --- 4. 解析抖音状态 ---
+    fulfillment_status_text, _, msg = parse_douyin_order_status(order_details)
+    if fulfillment_status_text is None: return f"错误(解析状态): {msg}"
+    # --- 5. 检查抖音状态 ---
+    if fulfillment_status_text != "已履约": return f"提示：订单状态为 '{fulfillment_status_text}'，不予兑换。"
+    # --- 6. 获取飞书 Token ---
+    feishu_token, msg = get_feishu_tenant_access_token(FEISHU_APP_ID, FEISHU_APP_SECRET)
+    if not feishu_token: return f"错误(飞书Token): {msg}"
+    # --- 7. 检查 Table 1 ---
+    exists_in_table1, msg = check_order_exists_in_feishu(feishu_token, FEISHU_APP_TOKEN, FEISHU_TABLE1_ID, FEISHU_TABLE1_ORDER_FIELD, douyin_order_id_str)
+    if exists_in_table1 is None: return f"错误(检查Table1): {msg}"
+    if exists_in_table1 is True: return f"提示：订单 '{douyin_order_id_str}' 已兑换过。"
+    # --- 8. 写入 Table 1 ---
+    table1_record_id, msg = add_bitable_record(feishu_token, FEISHU_APP_TOKEN, FEISHU_TABLE1_ID, FEISHU_TABLE1_ORDER_FIELD, douyin_order_id_str, FEISHU_TABLE1_STATUS_FIELD, fulfillment_status_text)
+    if not table1_record_id: return f"错误(写入Table1): {msg}"
+    print(f"后台日志: Table 1 写入成功, Record ID: {table1_record_id}")
+    # --- 9. 查找 Table 2 ---
+    table2_record_id, activation_code_text = find_first_unsent_record_and_code_text(feishu_token, FEISHU_APP_TOKEN, FEISHU_TABLE2_ID, FEISHU_TABLE2_STATUS_FIELD, TABLE2_UNSENT_STATUS_CONTENT, FEISHU_TABLE2_ACTIVATION_CODE_FIELD, FEISHU_TABLE2_SORT_FIELD)
+    if table2_record_id and activation_code_text:
+        final_output = f"{activation_code_text}" # 直接显示激活码
+        # --- 10. "发送" 激活码 (模拟) ---
+        print(f"后台日志: '发送' 激活码: {activation_code_text}")
+        sent_success = True # 假设发送成功
+        if sent_success:
+            # --- 11. 更新 Table 2 ---
+            update_t2_success, msg_t2 = update_table2_record_to_sent(feishu_token, FEISHU_APP_TOKEN, FEISHU_TABLE2_ID, table2_record_id, FEISHU_TABLE2_STATUS_FIELD, TABLE2_SENT_STATUS_CONTENT, FEISHU_TABLE2_UPDATE_TIME_FIELD)
+            if not update_t2_success: print(f"后台警告: {msg_t2}") # 只在后台打印警告
+            if update_t2_success: # 只有 T2 更新成功才更新 T1
+                 # --- 12. 更新 Table 1 ---
+                update_t1_success, msg_t1 = update_table1_record_to_exchanged(feishu_token, FEISHU_APP_TOKEN, FEISHU_TABLE1_ID, table1_record_id, FEISHU_TABLE1_STATUS_FIELD, TABLE1_EXCHANGED_STATUS_CONTENT)
+                if not update_t1_success: print(f"后台警告: {msg_t1}") # 只在后台打印警告
+        else: final_output = "错误：激活码 '发送' 失败 (模拟)"
+    elif table2_record_id and not activation_code_text: final_output = f"错误：未能获取激活码文本，请检查 Table 2 数据。"
+    else: final_output = "提示：无可用激活码，请联系管理员。"
+    return final_output # 返回最终结果
+
+# --- 创建 Gradio Interface (最终简化版) ---
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 竞潮玩游戏激活码兑换小助手")
+    gr.Markdown("请输入抖音订单号以兑换激活码。仅限状态为“已履约”且未兑换过的订单。")
+    with gr.Row():
+        order_id_input = gr.Textbox(label="输入抖音订单号", placeholder="在此填写订单号...")
+    process_button = gr.Button("兑换激活码")
+    with gr.Row():
+        activation_code_output = gr.Textbox(label="兑换结果 / 激活码", interactive=False)
+    process_button.click(
+        fn=process_order_simple,
+        inputs=order_id_input,
+        outputs=activation_code_output
+    )
+
+# --- 启动 Gradio 应用 ---
+if __name__ == "__main__":
+    demo.launch()