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

Fix for Queue when empty

Browse Source
This commit is contained in:
Jacob Zelek
2020-12-26 15:23:13 -08:00
parent 81dd25829f
commit 7ca81d03bd
1 changed files with 18 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
generator
View File

@@ -94,18 +94,27 @@ def generate_video_thumbnails(args):
def worker(queue): def worker(queue):
while True: while True:
work_unit = queue.get() try:
work_unit = queue.get(False)
# If no work unit then quit # If no work unit then quit
if not work_unit: if not work_unit:
return break
except:
break
input_file, output_file, interval, size, columns = work_unit input_file, output_file, interval, size, columns = work_unit
video_file_clip = VideoFileClip(input_file)
output_prefix = get_output_prefix() try:
file_name = os.path.basename(input_file) video_file_clip = VideoFileClip(input_file)
generate_frames(file_name, video_file_clip, interval, output_prefix, size) output_prefix = get_output_prefix()
generate_sprite_from_frames(output_prefix, columns, size, output_file) file_name = os.path.basename(input_file)
generate_frames(file_name, video_file_clip,
interval, output_prefix, size)
generate_sprite_from_frames(output_prefix, columns,
size, output_file)
except:
print("Error occurred with file: {}".format(file_name))
def generate_frames(file_name, video_file_clip, interval, output_prefix, size): def generate_frames(file_name, video_file_clip, interval, output_prefix, size):