taskManager / api.py
rogerxavier's picture
Upload 6 files
b9d0462 verified
raw
history blame
1.73 kB
import uvicorn
from fastapi import FastAPI
from utils import *
from typing import *
from config import *
from apscheduler.schedulers.background import BackgroundScheduler#定时任务
scheduler = BackgroundScheduler()#这个不要用缓存的
def load_config():
return get_variables()
class configData():
def update(self):
self.data = load_config()
print(self.data)
return self.data
configData = configData()#复用实例
app = FastAPI()
def periodic_function():
data = configData.update()
print(f'定时执行的操作时间:{datetime.now()}')
print("是否允许定时任务",data['allow_scheduler'])
run_asp_task()#执行asp任务进行发送
#启动app时候的定时任务 - 每天根据manga_abs_dir路径是否有jpg文件来判断是否运行
@app.on_event("startup")
async def app_start():
#scheduler.add_job(periodic_function, 'interval', seconds=55)
scheduler.add_job(periodic_function, 'interval', seconds=18000)#间隔5小时执行一次
scheduler.start()
# 返回当前目录及其子目录下的所有信息
@app.get("/listFiles")
def list_files_func():
result = list_files()
return result
# 返回一个章节并删除
@app.get("/random_chapter")
def random_chapter_handler():
result = random_chapter()
return result
#两个blocks放到不同的地方,一个管理manga素材上传,一个管理任务处理
app = gr.mount_gradio_app(app, mangaManager, path="/gr")
#下面这个暂不更新->没想到太大作用
app = gr.mount_gradio_app(app, taskManager, path="/taskManager")
if __name__ == '__main__':
uvicorn.run(app, host='0.0.0.0', port=7860)