diff --git a/sau_backend.py b/sau_backend.py index aff4186..54ee393 100644 --- a/sau_backend.py +++ b/sau_backend.py @@ -15,7 +15,6 @@ from myUtils.auth import check_cookie from flask import Flask, request, jsonify, Response, render_template, send_from_directory from werkzeug.utils import secure_filename from conf import BASE_DIR -from myUtils.login import get_tencent_cookie, douyin_cookie_gen, get_ks_cookie, xiaohongshu_cookie_gen from myUtils.postVideo import ( post_video_tencent, post_video_DouYin, @@ -38,6 +37,10 @@ from uploader.alipay_uploader.main import alipay_setup from uploader.weibo_uploader.main import weibo_setup from uploader.hupu_uploader.main import hupu_setup from uploader.youtube_uploader.main import youtube_setup +from uploader.douyin_uploader.main import douyin_setup +from uploader.ks_uploader.main import ks_setup +from uploader.xiaohongshu_uploader.main import xiaohongshu_setup +from uploader.tencent_uploader.main import tencent_setup from uploader.bilibili_uploader.runtime import ensure_biliup_binary from loguru import logger from utils.settings import get_setting, set_setting, get_bool @@ -879,7 +882,15 @@ def _login_new_platform(platform_type, account_name, status_queue): try: status_queue.put("正在打开浏览器登录,请稍候...") t = int(platform_type) - if t == 6: + if t == 1: + result = loop.run_until_complete(xiaohongshu_setup(str(account_file), handle=True, return_detail=True, qrcode_callback=cb, headless=login_headless)) + elif t == 2: + result = loop.run_until_complete(tencent_setup(str(account_file), handle=True, return_detail=True, qrcode_callback=cb, headless=login_headless)) + elif t == 3: + result = loop.run_until_complete(douyin_setup(str(account_file), handle=True, return_detail=True, qrcode_callback=cb, headless=login_headless)) + elif t == 4: + result = loop.run_until_complete(ks_setup(str(account_file), handle=True, return_detail=True, qrcode_callback=cb, headless=login_headless)) + elif t == 6: result = loop.run_until_complete(baijiahao_setup(str(account_file), handle=True, return_detail=True, qrcode_callback=cb, headless=login_headless)) elif t == 7: result = loop.run_until_complete(alipay_setup(str(account_file), handle=True, return_detail=True, qrcode_callback=cb, headless=login_headless)) @@ -914,25 +925,13 @@ def _login_new_platform(platform_type, account_name, status_queue): def run_async_function(type,id,status_queue): match type: case '1': - loop = asyncio.new_event_loop() - asyncio.set_event_loop(loop) - loop.run_until_complete(xiaohongshu_cookie_gen(id, status_queue)) - loop.close() + _login_new_platform('1', id, status_queue) case '2': - loop = asyncio.new_event_loop() - asyncio.set_event_loop(loop) - loop.run_until_complete(get_tencent_cookie(id,status_queue)) - loop.close() + _login_new_platform('2', id, status_queue) case '3': - loop = asyncio.new_event_loop() - asyncio.set_event_loop(loop) - loop.run_until_complete(douyin_cookie_gen(id,status_queue)) - loop.close() + _login_new_platform('3', id, status_queue) case '4': - loop = asyncio.new_event_loop() - asyncio.set_event_loop(loop) - loop.run_until_complete(get_ks_cookie(id,status_queue)) - loop.close() + _login_new_platform('4', id, status_queue) case '5': _login_new_platform('5', id, status_queue) case '6':