添加分享功能

main.py

@@ -150,6 +150,7 @@ TG_BASE_URL = "https://tg.alist.dpdns.org/bot"
 
 
 ###################TG机器人功能区###################
+# ❗❗❗❗❗❗❗❗❗注意TG机器人callbackdata不能超过64位，否则会报无效按钮的错误
 # 定义命令处理函数
 async def start(update: Update, context):
     commands = (
@@ -253,6 +254,10 @@ async def tg_show_files(update: Update, context: CallbackContext):
                             f"删除",
                             callback_data=f"del_f:{file['id']}:{file['parent_id']}",
                         ),
+                        InlineKeyboardButton(
+                            f"分享",
+                            callback_data=f"sh_f:{file['id']}:{file['parent_id']}",
+                        ),
                     ]
                 )
             else:
@@ -266,6 +271,10 @@ async def tg_show_files(update: Update, context: CallbackContext):
                             f"删除",
                             callback_data=f"del_f:{file['id']}:{file['parent_id']}",
                         ),
+                        InlineKeyboardButton(
+                            f"分享",
+                            callback_data=f"sh_f:{file['id']}:{file['parent_id']}",
+                        ),
                     ]
                 )
 
@@ -358,6 +367,10 @@ async def perform_file_action(
                                 f"删除",
                                 callback_data=f"del_f:{file['id']}:{file['parent_id']}",
                             ),
+                            InlineKeyboardButton(
+                                f"分享",
+                                callback_data=f"sh_f:{file['id']}:{file['parent_id']}",
+                            ),
                         ]
                     )
                 else:
@@ -371,6 +384,10 @@ async def perform_file_action(
                                 f"删除",
                                 callback_data=f"del_f:{file['id']}:{file['parent_id']}",
                             ),
+                            InlineKeyboardButton(
+                                f"分享",
+                                callback_data=f"sh_f:{file['id']}:{file['parent_id']}",
+                            ),
                         ]
                     )
 
@@ -392,6 +409,12 @@ async def perform_file_action(
             )
         else:
             await update.callback_query.edit_message_text(f"❌未找到文件下载地址!!")
+    elif action == "sh_f":
+        result = await THUNDERX_CLIENT.file_batch_share([file_id], False, -1)
+        if result["share_id"] is not None:
+            await update.callback_query.edit_message_text(f"✅分享码为:{result["share_id"]}")
+        else:
+            await update.callback_query.edit_message_text(f"❌分享失败!!")
 
 
 #################### 离线任务处理 ##########################
@@ -592,7 +615,9 @@ async def init_client():
         ########## 文件操作 ###############
 
         TG_BOT_APPLICATION.add_handler(
-            CallbackQueryHandler(handle_file_operation, pattern="^(del_f|ls_f|dw_f):")
+            CallbackQueryHandler(
+                handle_file_operation, pattern="^(del_f|ls_f|dw_f|sh_f):"
+            )
         )
         # 处理取消任务操作
         TG_BOT_APPLICATION.add_handler(