Update api.py

api.py

@@ -15,6 +15,7 @@ from bilibili_api import sync, video_uploader, Credential#bili上传部分
 import asyncio #保证后台任务的先后
 import json #对bili_meta_data进行dumps ->str和loads—>dict
 import bilibili_api
+import random #随机获取cover
 import requests #用于请求其他space的图片api
 
 sessdata = os.getenv('sessdata')
@@ -178,6 +179,8 @@ async def upload_video(meta:dict,cover_url:str):
     await uploader.start()
 
 
+
+
 #检测output.mp4下载查看生成效果(b不过审时用)
 @app.get("/get_video")
 async def get_video(dir:str = "/mp4_out/output.mp4")->"video file in header":
@@ -194,6 +197,53 @@ async def get_video(dir:str = "/mp4_out/output.mp4")->"video file in header":
 
 
 
+# 补零函数，将数字部分补齐为指定长度
+def zero_pad(s, length):
+    return s.zfill(length)
+
+
+#如果默认的cover不能过审,那么随机选取无水印的manga1下面的随便一个作为cover重新上传
+async def upload_video(meta:dict,cover_url:str):
+    credential = Credential(sessdata=sessdata,
+                            bili_jct=bili_jct,
+                            buvid3=buvid3)
+
+    # 获取存放去水印漫画图片的路径
+    img_path = 'manga1'
+    #获取漫画原图无水印的加入image_files，并排序
+    subdir_path = os.path.join(os.getcwd(), img_path)
+    # 对话图片经过加入list并补0确定顺序
+    image_files = []
+    for root, dirs, files in os.walk(subdir_path):
+        for file in files:
+            if file.endswith(".jpg") or file.endswith(".png"):
+                image_files.append(os.path.relpath(os.path.join(root, file)))
+    # 对对话框文件名中的数字部分进行补零操作-这样顺序会正常
+    image_files.sort(
+        key=lambda x: zero_pad(''.join(filter(str.isdigit, os.path.splitext(os.path.basename(x))[0])), 3))
+
+    # 获取随机一张图片图片存入cover/0.jpg作为cover图片
+    random_image = Image.open(random.choice(image_files))
+    # 定义要保存的文件路径
+    save_path = os.path.join("cover", "0.jpg")
+    # 保存图片文件
+    random_image.save(save_path)
+    
+    with open("cover/0.jpg", "rb") as f:
+        cover_data = f.read()
+    cover = bilibili_api.utils.picture.Picture.from_content(cover_data,"jpg")#直接读取本地图片content
+    
+    
+    
+    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)
+
+    @uploader.on("__ALL__")
+    async def ev(data):
+        print(data)
+
+    await uploader.start()