فهرست منبع

Add extra sanitation for segment name in json

dvingerh 4 سال پیش
والد
کامیت
25db0847ec
2فایلهای تغییر یافته به همراه3 افزوده شده و 2 حذف شده
  1. 1 0
      pyinstalive/assembler.py
  2. 2 2
      pyinstalive/dlfuncs.py

+ 1 - 0
pyinstalive/assembler.py

@@ -90,6 +90,7 @@ def assemble(user_called=True, retry_with_zero_m4v=False):
             logger.info("Assembling video segment files from specified folder: {}".format(ass_segment_dir))
 
         for segment in all_segments:
+            segment = re.sub('\?.*$', '', segment)
             if not os.path.isfile(segment.replace('.m4v', '.m4a')):
                 logger.warn('Audio segment not found: {0!s}'.format(segment.replace('.m4v', '.m4a')))
                 continue

+ 2 - 2
pyinstalive/dlfuncs.py

@@ -150,11 +150,11 @@ def merge_segments():
                 logger.info('Merging downloaded files into video.')
                 pil.broadcast_downloader.stitch(live_mp4_file, cleartempfiles=pil.clear_temp_files)
                 logger.info('Successfully merged downloaded files into video.')
+                if pil.clear_temp_files:
+                    helpers.remove_temp_folder()
             else:
                 logger.binfo("Merging of downloaded files has been disabled.")
                 logger.binfo("Use --assemble command to manually merge downloaded segments.")
-            if pil.clear_temp_files:
-                helpers.remove_temp_folder()
             helpers.remove_lock()
         except ValueError as e:
             logger.separator()