Update main.py

main.py

@@ -173,6 +173,30 @@ async def get_file_info(file_id: str):
 async def emptytrash():
     return await THUNDERX_CLIENT.emptytrash()
 
+
+
+@api_router.post(
+    "/file_batch_share", summary="创建分享", description="创建分享", tags=["API"]
+)
+async def file_batch_share(ids: List[str] = None, need_password: bool | None = False,expiration_days:int | None=-1):
+    return await THUNDERX_CLIENT.file_batch_share(ids,need_password,expiration_days)
+
+
+
+@api_router.post(
+    "/get_share_folder", summary="获取分享信息", description="获取分享信息", tags=["API"]
+)
+async def get_share_folder(share_id: str, pass_code_token: str | None = None,parent_id:str | None=None):
+    return await THUNDERX_CLIENT.get_share_folder(share_id,pass_code_token,parent_id)
+
+
+@api_router.post(
+    "/restore", summary="转存分享文件", description="转存分享文件", tags=["API"]
+)
+async def restore(share_id: str, pass_code_token: str | None = None,file_ids:List[str] | None=None):
+    return await THUNDERX_CLIENT.restore(share_id,pass_code_token,file_ids)
+
+
 @api_router.get(
     "/offline", summary="离线任务列表", description="离线任务列表", tags=["API"]
 )
@@ -206,5 +230,14 @@ async def quota_info():
     return await THUNDERX_CLIENT.get_quota_info()
 
 
+
+@api_router.get(
+    "/invite_code", summary="查看邀请码", description="查看邀请码", tags=["API"]
+)
+async def get_invite_code():
+    return await THUNDERX_CLIENT.get_invite_code()
+
+
+
 app.include_router(front_router)
 app.include_router(api_router)