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

Merge pull request #18 from Voodfy/master

Browse Source 
Adding support to python3
This commit is contained in:
Flavio Ribeiro
2020-09-26 11:49:30 -04:00
committed by GitHub
parent f1eb403c1f 1a1e047484
commit 542f0d2e36
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
build
View File

@@ -5,5 +5,5 @@ if [ "$UNAME_S" == "Darwin" ]; then
ln -s /usr/local/include/freetype2 /usr/local/include/freetype ln -s /usr/local/include/freetype2 /usr/local/include/freetype
fi fi
pip install -r requirements.txt pip3 install -r requirements.txt
chmod a+x generator chmod a+x generator

8
generator
View File

@@ -1,4 +1,4 @@
#!/usr/bin/env python #!/usr/bin/env python3
"""Video Thumbnail Generator """Video Thumbnail Generator
@@ -39,13 +39,13 @@ def generate_video_thumbnail(args):
generate_sprite_from_frames(outputPrefix, columns, size, output) generate_sprite_from_frames(outputPrefix, columns, size, output)
def generate_frames(videoFileClip, interval, outputPrefix, size): def generate_frames(videoFileClip, interval, outputPrefix, size):
print "Extracting", int(videoFileClip.duration / interval), "frames" print("Extracting", int(videoFileClip.duration / interval), "frames")
frameCount = 0 frameCount = 0
with progressbar(range(0, int(videoFileClip.duration), interval)) as items: with progressbar(range(0, int(videoFileClip.duration), interval)) as items:
for i in items: for i in items:
extract_frame(videoFileClip, i, outputPrefix, size, frameCount) extract_frame(videoFileClip, i, outputPrefix, size, frameCount)
frameCount += 1 frameCount += 1
print "Frames extracted." print("Frames extracted.")
def extract_frame(videoFileClip, moment, outputPrefix, size, frameCount): def extract_frame(videoFileClip, moment, outputPrefix, size, frameCount):
output = outputPrefix + ("%05d.png" % frameCount) output = outputPrefix + ("%05d.png" % frameCount)
@@ -88,7 +88,7 @@ def generate_sprite_from_frames(framesPath, columns, size, output):
finalImage.save(output) finalImage.save(output)
shutil.rmtree(TMP_FRAMES_PATH, ignore_errors=True) shutil.rmtree(TMP_FRAMES_PATH, ignore_errors=True)
print "Saved!" print("Saved!")
def get_output_prefix(): def get_output_prefix():
if not os.path.exists(TMP_FRAMES_PATH): if not os.path.exists(TMP_FRAMES_PATH):