Mercurial > hg > config
changeset 754:f011ec45b8e8
add example load type interface
author | Jeff Hammel <k0scist@gmail.com> |
---|---|
date | Fri, 03 Jul 2015 21:07:03 -0700 |
parents | 05fef8e5b8a9 |
children | 14eee9200854 |
files | python/ffslice.py python/url.py |
diffstat | 2 files changed, 190 insertions(+), 0 deletions(-) [+] |
line wrap: on
line diff
--- a/python/ffslice.py Fri Jul 03 16:39:33 2015 -0700 +++ b/python/ffslice.py Fri Jul 03 21:07:03 2015 -0700 @@ -140,6 +140,55 @@ return format_seconds_to_hhmmss(seconds) return '{:.2}'.format(seconds) + def slice(self): + """ + iterates over all the specified clips and slices them as per input params. + The sliced clips are stored under the provided coommand line destinati + on directory or current working dir + """ + + for clip_path in self.options.clip_paths: + print "***** Processing {0}".format(clip_path) + + if not os.path.exists(clip_path): + print "File not found! skipping {0}".format(clip_path) + continue + + #making sure the slice time is within bounds of the clip duration + duration = self.duration(clip_path) + print "Duration: {0}".format(duration) + if self.options.slice_length_sec > duration: + print "Skipping {0}, slice_time {1} is GREATER than file duration {2} ".format(clip_path,self.options.slice_length_sec ,duration) + continue + + #calculating the number slices to create + num_slices = 0 + if self.options.num_slices: + num_slices = min (self.options.num_slices, int(duration/(self.options.slice_length_sec))) + else: #number of slice were not specified + num_slices = int(duration/(self.options.slice_length_sec)) + print "Slicing in {0} parts, {1} seconds each".format(num_slices,self.options.slice_length_sec) + + hh = 0 + mm = 0 + ss = 0 + start_time_secs = 0 + [out_filename,out_file_ext] = clip_path.split("/")[-1].split(".") #assuming the file path to be something like /df/dsf/dsf/dsf.mp4 + ensure_dir(self.options.out_directory) + #creating slices + for i in range(num_slices): + [hh,mm,ss] = self.format_seconds_to_hhmmss(start_time_secs) + out_file_path = "{0}/{1}_{2}.{3}".format(self.options.out_directory,out_filename,i,out_file_ext) + command = "ffmpeg -ss {0}:{1}:{2} -t {3} -i {4} -q {5} -strict -2 {6}".format(hh,mm,ss,self.options.slice_length_sec, + clip_path,1, out_file_path) + + try: + output = subprocess.call(command, shell=True) + except subprocess.CalledProcessError as e: + print (e.output) + raise + start_time_secs += self.options.slice_length_sec + def main(args=sys.argv[1:]): """CLI""" @@ -163,6 +212,8 @@ parser.format_seconds(_duration))) sys.exit(returncode) + parser.slice() + if __name__ == '__main__': main()
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/python/url.py Fri Jul 03 21:07:03 2015 -0700 @@ -0,0 +1,139 @@ +#!/usr/bin/env python +# -*- coding: utf-8 -*- + +""" +url manipulation +""" + +import argparse +import os +import shutil +import subprocess +import sys +import tempfile +import urlparse +import urllib2 + +__all__ = ['load', 'main'] +string = (str, unicode) + +def ensure_dir(directory): + """ensure `directory` is a directory""" + if os.path.exists(directory): + assert os.path.isdir(directory) + return directory + os.makedirs(directory) + return directory + +def isURL(url): + return '://' in url + +def read_s3(url): + name = tempfile.mktemp() + try: + subprocess.check_output(['s3cmd', 'get', url, name]) + with open(name) as f: + read = f.read() + os.remove(name) + return read + finally: + if os.path.exists(name): + os.remove(name) + +def read_http(url): + return urllib2.urlopen(url).read() + +def read_file(url): + scheme = 'file://' + if url.startswith(scheme): + url = url[len(scheme):] + return open(url).read() + +loaders = {'s3': read_s3, + 'http': read_http, + 'https': read_http, + 'file': read_file + } + +def scheme(url): + if '://' in url: + parsed = urlparse.urlsplit(url) + return parsed.scheme + return 'file' + +def parent(url): + if '://' in url: + return url.rsplit('/', 1)[0] + else: + # file + return os.path.abspath(os.path.dirname(url)) + +def basename(url): + if '://' in url: + return url.rsplit('/', 1)[-1] + else: + # file + return os.path.basename(url) + +def loader(url): + return loaders[scheme(url)] + +def load(url): + """returns the contents of a URL""" + return loader(url)(url) + +def get_file(src, dest): + shutil.copy2(src, dest) + +def get_s3(src, dest): + subprocess.check_output(['s3cmd', 'get', src, dest]) + +def default_getter(src, dest): + assert not os.path.isURL(dest) + dirname = parent(dest) + ensure_dir(dirname) + with open(dest, 'w') as f: + f.write(load(url)) + +getters = {'file': get_file, + 's3': get_s3 + } + +def get(src, dest): + """get a thing to a local file""" + if os.path.isdir(dest): + dest = os.path.join(dest, basename(src)) + return getters.get(scheme(src), default_getter)(src, dest) + +def rel(base, path): + """ + relative path to base + otherwise, return None + """ + + if path.startswith(base): + return path[len(base):] + +def main(args=sys.argv[1:]): + """CLI""" + + # parse command line + parser = argparse.ArgumentParser(description=__doc__) + parser.add_argument('url', help='URL to read') + parser.add_argument('-o', '--output', dest='output', + help="get to this location") + options = parser.parse_args(args) + + if options.output: + # copy src to this location + get(options.url, options.output) + sys.exit() + + # read location + contents = load(options.url) + + # output + print (contents) + +if __name__ == '__main__': + main()