瀏覽代碼

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))
             logger.info("Assembling video segment files from specified folder: {}".format(ass_segment_dir))
 
 
         for segment in all_segments:
         for segment in all_segments:
+            segment = re.sub('\?.*$', '', segment)
             if not os.path.isfile(segment.replace('.m4v', '.m4a')):
             if not os.path.isfile(segment.replace('.m4v', '.m4a')):
                 logger.warn('Audio segment not found: {0!s}'.format(segment.replace('.m4v', '.m4a')))
                 logger.warn('Audio segment not found: {0!s}'.format(segment.replace('.m4v', '.m4a')))
                 continue
                 continue

+ 2 - 2
pyinstalive/dlfuncs.py

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