save DS movie if not s3

WiiLink24 · Jun 2, 2024 · 807eea2 · 807eea2
1 parent c7fa773
commit 807eea2
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/theunderground/mobiclip.py b/theunderground/mobiclip.py
@@ -153,6 +153,15 @@ def save_movie_data(movie_id: int, thumbnail_data: bytes, movie_data: bytes, ds_
         movie.write(movie_data)
         movie.close()
 
+        if ds_movie_data:
+            if not os.path.isdir(ds_movie_dir):
+                os.makedirs(ds_movie_dir)
+
+            # Write DS movie
+            ds_movie = open(f"{ds_movie_dir}/{movie_id}.enc", "wb")
+            ds_movie.write(ds_movie_data)
+            ds_movie.close()
+
 
 def save_pay_movie_data(
     movie_id: int, thumbnail_data: bytes, movie_data: bytes, poster_data: bytes