josh/video-thumbnail-generator
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Bug fix - worker was only taking one item and terminating after

Browse Source
This commit is contained in:
Jacob Zelek
2020-12-17 06:28:48 -08:00
parent 7a0cf18c94
commit 8d1223b646
1 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
generator
View File

@@ -93,18 +93,19 @@ def generate_video_thumbnails(args):
def worker(queue): def worker(queue):
work_unit = queue.get() while True:
work_unit = queue.get()
# If no work unit then quit # If no work unit then quit
if not work_unit: if not work_unit:
return return
input_file, output_file, interval, size, columns = work_unit input_file, output_file, interval, size, columns = work_unit
video_file_clip = VideoFileClip(input_file) video_file_clip = VideoFileClip(input_file)
output_prefix = get_output_prefix() output_prefix = get_output_prefix()
file_name = os.path.basename(input_file) file_name = os.path.basename(input_file)
generate_frames(file_name, video_file_clip, interval, output_prefix, size) generate_frames(file_name, video_file_clip, interval, output_prefix, size)
generate_sprite_from_frames(output_prefix, columns, size, output_file) generate_sprite_from_frames(output_prefix, columns, size, output_file)
def generate_frames(file_name, video_file_clip, interval, output_prefix, size): def generate_frames(file_name, video_file_clip, interval, output_prefix, size):