|
@@ -7,7 +7,8 @@ from datetime import timedelta
|
|
from io import BytesIO
|
|
from io import BytesIO
|
|
from typing import Optional, List
|
|
from typing import Optional, List
|
|
from urllib.parse import quote
|
|
from urllib.parse import quote
|
|
-
|
|
|
|
|
|
+import os
|
|
|
|
+import sys
|
|
import pandas as pd
|
|
import pandas as pd
|
|
import uvicorn
|
|
import uvicorn
|
|
import yaml
|
|
import yaml
|
|
@@ -17,13 +18,18 @@ from fastapi.responses import StreamingResponse
|
|
from loguru import logger
|
|
from loguru import logger
|
|
from pydantic import BaseModel, Field
|
|
from pydantic import BaseModel, Field
|
|
|
|
|
|
|
|
+curr_path = os.path.abspath(os.path.dirname(__file__))
|
|
|
|
+project_root_path = curr_path[:curr_path.find("video_to_word") + len("video_to_word")]
|
|
|
|
+sys.path.append(project_root_path)
|
|
|
|
+
|
|
from asr_client import send_asr_request, send_task_request
|
|
from asr_client import send_asr_request, send_task_request
|
|
from common_func import get_db_engine
|
|
from common_func import get_db_engine
|
|
from config.url import toutiao_static_video_url
|
|
from config.url import toutiao_static_video_url
|
|
from database import insert, update, query, Task
|
|
from database import insert, update, query, Task
|
|
|
|
+from time_task.get_material_and_script_by_query_word import get_material_and_script
|
|
|
|
|
|
logger.remove() # 删去 import logger之后自动产生的handler,不删除的话会出现重复输出的现象
|
|
logger.remove() # 删去 import logger之后自动产生的handler,不删除的话会出现重复输出的现象
|
|
-logger.add("logs/loguru.{time:YYYY-MM-DD}.log",
|
|
|
|
|
|
+logger.add("/data/pythonProject/video_to_word/logs/main_server.{time:YYYY-MM-DD}.log",
|
|
rotation="00:00",
|
|
rotation="00:00",
|
|
format="{time:YYYY-MM-DD HH:mm:ss,SSS} [{process}] [{thread}] {level} {file} {line} - {message}",
|
|
format="{time:YYYY-MM-DD HH:mm:ss,SSS} [{process}] [{thread}] {level} {file} {line} - {message}",
|
|
level="INFO")
|
|
level="INFO")
|
|
@@ -367,6 +373,26 @@ def add_script_config(item: AddScriptConfig):
|
|
"success": False}
|
|
"success": False}
|
|
|
|
|
|
|
|
|
|
|
|
+@logger.catch
|
|
|
|
+@app.post('/get_material_and_script_time_task/',
|
|
|
|
+ response_model=BaseResponse,
|
|
|
|
+ tags=['back-end task'],
|
|
|
|
+ description="获取素材和脚本任务",
|
|
|
|
+ summary='获取素材和脚本任务')
|
|
|
|
+def get_material_and_script_time_task():
|
|
|
|
+ try:
|
|
|
|
+ get_material_and_script()
|
|
|
|
+ logger.info(f"{date.today().strftime('%Y-%m-%d')}, 获取素材和脚本任务执行完成.")
|
|
|
|
+ return {"code": 0,
|
|
|
|
+ "success": True,
|
|
|
|
+ "message": f"{date.today().strftime('%Y-%m-%d')},获取素材和脚本任务执行完成."}
|
|
|
|
+ except:
|
|
|
|
+ logger.error(f"{date.today().strftime('%Y-%m-%d')}, 获取素材和脚本任务执行发生异常. {traceback.format_exc()}")
|
|
|
|
+ return {"code": -1,
|
|
|
|
+ "success": False,
|
|
|
|
+ "message": f"{date.today().strftime('%Y-%m-%d')},获取素材和脚本任务执行发生异常 .{traceback.format_exc()}"}
|
|
|
|
+
|
|
|
|
+
|
|
if __name__ == '__main__':
|
|
if __name__ == '__main__':
|
|
uvicorn.run(app='main:app', host="0.0.0.0", port=31013, reload=True, debug=True)
|
|
uvicorn.run(app='main:app', host="0.0.0.0", port=31013, reload=True, debug=True)
|
|
# gunicorn main:app -w 4 -k uvicorn.workers.UvicornWorker #线上启动命令
|
|
# gunicorn main:app -w 4 -k uvicorn.workers.UvicornWorker #线上启动命令
|