File size: 7,021 Bytes
8fcd983
d38fe54
 
 
 
 
 
 
9f7f597
cd84989
b628481
 
7ca96aa
 
c242b8c
a3e21b5
 
184af71
 
 
 
26fc5e2
d38fe54
184af71
26fc5e2
 
 
d38fe54
26fc5e2
 
d38fe54
26fc5e2
d38fe54
26fc5e2
 
 
9f7f597
26fc5e2
9f7f597
 
d38fe54
2c03cb6
679ffd8
 
 
06a0cd9
679ffd8
 
 
 
 
 
2c03cb6
49567b1
2c03cb6
582ad23
 
49567b1
 
 
 
582ad23
2c03cb6
49567b1
b628481
2c03cb6
 
 
 
b628481
bc7dec9
96888b9
 
 
 
 
 
 
 
 
 
 
 
 
 
bc7dec9
49f01e6
b628481
 
bc7dec9
49f01e6
b628481
2c03cb6
 
 
 
 
 
 
 
 
 
1ad0d49
184af71
a3e21b5
 
 
 
e472c61
a3e21b5
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1ad0d49
a3e21b5
 
 
 
e472c61
 
944b5f7
a5b527c
 
a3e21b5
e472c61
a3e21b5
bc7dec9
 
 
 
 
b628481
a3e21b5
 
2c03cb6
7ca96aa
33e8fe3
 
7ca96aa
 
 
 
 
 
 
2c03cb6
c242b8c
 
 
 
184af71
a5b527c
c242b8c
 
 
 
 
 
 
a5b527c
c242b8c
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
from fastapi import FastAPI, File, UploadFile,Body
from fastapi.staticfiles import StaticFiles
from fastapi.responses import FileResponse
from PIL import Image
import numpy as np
import urllib.request
import io
import os
from moviepyTest import test
from typing import *
from fastapi.responses import PlainTextResponse #执行其他py的plaintext返回
import subprocess
from fastapi import BackgroundTasks
import time
from bilibili_api import sync, video_uploader, Credential#bili上传部分
import concurrent.futures #线程保证后台任务的先后


sessdata = os.getenv('sessdata')
bili_jct = os.getenv('bili_jct')
buvid3 = os.getenv('buvid3')
app = FastAPI()


@app.get("/inference")
def inference():
    return "<p>Hello, World!</p>"
    
@app.get("/infer_t5")
def t5(input):
    
    return {"output": input}

@app.get("/moviepyTest")
def t5():
    result = test()
    
    return {"output": result}


    
##这个比较快不用异步
@app.post("/getOriginalMangaList") #增加指定保存路径功能,默认是/manga路径
def getOriginalMangaList(images: List[UploadFile] = File(...), save_path: str = "/manga"):
    saved_files = []
    for idx, image in enumerate(images):
        img = image.file.read()
        image_data = Image.open(io.BytesIO(img)).convert("L").convert("RGB")
        path_to_image = os.path.join(save_path, f"{idx}.jpg")
        image_data.save(path_to_image)
        saved_files.append(path_to_image)
    return {"message": "Images saved successfully", "saved_files": saved_files}
##这个比较快不用异步

##这个比较快不用异步
@app.delete("/deleteFiles")
async def delete_files(directory: str):
    for filename in os.listdir(directory):
        file_path = os.path.join(directory, filename)
        if os.path.isfile(file_path):
            os.remove(file_path)
    return {"message": f"成功删除{directory}目录下的所有文件"}
##这个比较快不用异步




########异步处理py文件执行接口
def file_executer(file_name:str)->"执行返回":
    try:
        print("开始执行py任务",file_name)
        # result = subprocess.check_output(["python", f"{file_name}.py"], stderr=subprocess.STDOUT).decode("utf-8")#执行完成后显示运行py的print
        ##########test 边执行py边显示print+++++++++++++++++++++++++++++++++++++
        # 开始执行Python 脚本
        output = []
        process = subprocess.Popen(["python", f"{file_name}.py"], stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
        # 逐行读取脚本的输出并显示
        for line in iter(process.stdout.readline, b''):
            line_text = line.decode('utf-8').strip()
            print(line_text)
            output.append(line_text)
        process.wait()
        result = '\n'.join(output)
        ##########test 边执行py边显示print+++++++++++++++++++++++++++++++++++++
        
        print("执行完成py任务:",file_name,"结果如下")
        print(result)
        return PlainTextResponse(result)
    except subprocess.CalledProcessError as e:
        print("执行py任务失败",file_name,"原因如下")
        print(f"Error executing {file_name}.py: {e}")
        return PlainTextResponse(f"Error executing {file_name}.py: {e}")
    
@app.get("/execute_py_file/{file_name}")
async def execute_py_file(file_name: str,background_tasks: BackgroundTasks):
    result = "接受到了请求{filename}任务".format(filename = file_name)
    background_tasks.add_task(file_executer,file_name)
    return result
########异步处理py文件执行接口




#收到图片后为了后台任务的顺序,单独设定一个函数依次将自定义执行顺序加入队列 ,有默认参数的写后面-并且和bili合并,不然还要等output.mp4生成后才能发起提交请求
# @app.get("/execute_all_task")
# async def execute_all_task(background_tasks: BackgroundTasks,file_list: list = ["1removeMask", "2magiDialogCut", "3mergeDialogToVideo"],
#                            bili_meta_data:dict = None,cover_image: UploadFile = File(...),mp4_out_file:str='mp4_out/output.mp4',
#                            allow_submit:bool=False,cover_path:str ='/cover'
                           
#                           ):
#     for file_name in file_list:
#         background_tasks.add_task(file_executer, file_name)
#     if allow_submit:
#         #如果请求中设置允许上传output.mp4
#         cover_img = cover_image.file.read()
#         cover_img_data = Image.open(io.BytesIO(cover_img)).convert("L").convert("RGB")
#         cover_path_to_image = os.path.join(cover_path, f"cover.jpg")
#         cover_img_data.save(cover_path_to_image)
#         #video数据从meta用户请求获取,至于
#         background_tasks.add_task(upload_video, bili_meta_data,cover_path_to_image)
#         return {"message": "提交video任务已加入队列"}
        
#     return {"message": "Tasks added to the queue"}


#保证既要提交队列后返回给用户,又要先完成前面的video生成步骤再执行后面的submit函数,还有兼顾allow_submit的值
@app.get("/execute_all_task")
async def execute_all_task(background_tasks: BackgroundTasks, file_list: list = ["1removeMask", "2magiDialogCut", "3mergeDialogToVideo"],
                           bili_meta_data: dict = None, cover_image: UploadFile = File(...), mp4_out_file: str = 'mp4_out/output.mp4',
                           allow_submit: bool = False, cover_path: str = '/cover'
                           ):
    tasks = []
    for file_name in file_list:
        tasks.append(file_executer(file_name))

    await asyncio.gather(*tasks)#使用asyncio.gather(*tasks)来等待所有file_executer任务完成后再执行upload_video任务。

    if allow_submit:
        cover_img = cover_image.file.read()
        cover_img_data = Image.open(io.BytesIO(cover_img)).convert("L").convert("RGB")
        cover_path_to_image = os.path.join(cover_path, f"cover.jpg")
        cover_img_data.save(cover_path_to_image)
        background_tasks.add_task(upload_video, bili_meta_data, cover_path_to_image)
        return {"message": "提交video任务已加入队列"}

    return {"message": "Tasks added to the queue"}







##########异步样例
def someTask():
    time.sleep(20)
    print("睡眠20s结束")
    
@app.get("/backTaskTest")
def returnRandomSubscribeUrl(background_tasks: BackgroundTasks)->str:
    #返回
    result = "先返回"
    background_tasks.add_task(someTask)
    return result
##########异步样例





async def upload_video(meta:dict,cover_path_to_image:str):
    credential = Credential(sessdata=sessdata,
                            bili_jct=bili_jct,
                            buvid3=buvid3)
    
    
    page = video_uploader.VideoUploaderPage(path='mp4_out/output_video.mp4', title=meta['title'], description=meta['desc'])
    
    uploader = video_uploader.VideoUploader([page], meta, credential, cover=cover_path_to_image)

    @uploader.on("__ALL__")
    async def ev(data):
        print(data)

    await uploader.start()