release bot with cache

src/code/combine.py

@@ -0,0 +1,37 @@
+import os
+from scripts import make_full_clip
+
+async def combine_all_videos():
+    corpora = ["pr", "pk", "mi", "av", "bs"]
+
+    for corpus in corpora:
+        base_path = os.path.join("../videos", corpus)
+        offices_path = os.path.join(base_path, "offices")
+        if not os.path.exists(offices_path):
+            continue
+
+        for filename in os.listdir(offices_path):
+            if filename.endswith(".mp4"):
+                office_path = os.path.join(offices_path, filename)
+                name, _ = os.path.splitext(filename)
+                parts = name.split("-")
+                if len(parts) != 4:
+                    continue
+                corpus_id, building, floor, office = parts
+                floor_filename = f"{corpus_id}-{building}-{floor}.mp4"
+                building_filename = f"{corpus_id}-{building}.mp4"
+                floor_path = os.path.join(base_path, "floors", floor_filename)
+                building_path = os.path.join(base_path, "buildings", building_filename)
+                if os.path.exists(floor_path) and os.path.exists(building_path):
+                    clips = [building_path, floor_path, office_path]
+
+                    # ПОЛНОЕ ВИДЕО
+                    await make_full_clip(clips)
+
+                    # УРЕЗАННОЕ ВИДЕО
+                    await make_full_clip(clips[1:])
+                else:
+                    print(f"Missing files for {filename} in {corpus}")
+
+import asyncio
+asyncio.run(combine_all_videos())

src/code/scripts.py

@@ -68,23 +68,24 @@ async def make_full_clip(paths):
         print("Некоторые файлы не найдены")
         return None
 
-    clips = [VideoFileClip(path) for path in paths]  # cоздаем список клипов
+    full_clip_name = f"{paths[-1][21:].replace('.mp4', '')}-{'all' if len(paths) == 3 else 'small'}.mp4"
 
-    full_clip = concatenate_videoclips(clips)  # cклеиваем все клипы
+    if not os.path.exists(f"../data/cache/{full_clip_name}"):
+        clips = [VideoFileClip(path) for path in paths]  # cоздаем список клипов
 
-    full_clip = full_clip.without_audio() # удаляем звук
+        full_clip = concatenate_videoclips(clips)  # cклеиваем все клипы
-    full_clip = full_clip.time_transform(lambda t: t * 2).with_duration(full_clip.duration / 2)    # ускоряем в 2 раз
 
-    full_clip_name = f"{paths[-1][21:].replace('.mp4', '')}-{'all' if len(paths) == 3 else 'small'}.mp4" # генерируем рандомный 5-ти значный ключ
+        full_clip = full_clip.without_audio() # удаляем звук
-
+        full_clip = full_clip.time_transform(lambda t: t * 2).with_duration(full_clip.duration / 2)    # ускоряем в 2 раз
-    # рендерим видео с параметрами
+        full_clip = full_clip.resized(height=400)
-    full_clip.write_videofile(f"../data/cache/{full_clip_name}",
-                            fps=30,
-                            codec="libx264",
-                            bitrate="1500k",
-                            preset="fast",
-                            ffmpeg_params=["-crf", "23"])
 
+        # рендерим видео с параметрами
+        full_clip.write_videofile(f"../data/cache/{full_clip_name}",
+                                fps=30,
+                                codec="libx264",
+                                bitrate="1500k",
+                                preset="fast",
+                                ffmpeg_params=["-crf", "23"])
 
     return f"../data/cache/{full_clip_name}"