Create and clean up a directory for temporary images (#12)

This commit is contained in:
Christian Ebert
2018-04-18 00:33:23 +01:00
parent ed56b0912b
commit 4161b485f2

View File

@@ -25,7 +25,7 @@ from click import progressbar
import glob, os, random, shutil, math, tempfile import glob, os, random, shutil, math, tempfile
TMP_FRAMES_PATH = tempfile.mkstemp()[1] TMP_FRAMES_PATH = tempfile.mkdtemp()
def generate_video_thumbnail(args): def generate_video_thumbnail(args):
videoFileClip = VideoFileClip(args['<video>']) videoFileClip = VideoFileClip(args['<video>'])
@@ -88,7 +88,7 @@ def generate_sprite_from_frames(framesPath, columns, size, output):
def get_output_prefix(): def get_output_prefix():
if not os.path.exists(TMP_FRAMES_PATH): if not os.path.exists(TMP_FRAMES_PATH):
os.makedirs(TMP_FRAMES_PATH) os.makedirs(TMP_FRAMES_PATH)
return TMP_FRAMES_PATH + ("%032x_" % random.getrandbits(128)) return TMP_FRAMES_PATH + os.sep + ("%032x_" % random.getrandbits(128))
if __name__ == "__main__": if __name__ == "__main__":
arguments = docopt(__doc__, version='0.0.2') arguments = docopt(__doc__, version='0.0.2')