allowing processing multiple files

This commit is contained in:
Paweł Płazieński 2016-02-20 16:32:10 +01:00
parent 79b2058747
commit 3db704fa51

15
napi
View file

@ -41,16 +41,18 @@ if len(args) < 1:
print "usage: %s <movie>" % (sys.argv[0],) print "usage: %s <movie>" % (sys.argv[0],)
sys.exit(0) sys.exit(0)
digest = hashlib.md5() for movie_filename in args:
movie_filename = args[0]
try: digest = hashlib.md5()
try:
with open(movie_filename, "r") as movie_file: with open(movie_filename, "r") as movie_file:
digest.update(movie_file.read(10 * 1024 * 1024)) digest.update(movie_file.read(10 * 1024 * 1024))
except IOError: except IOError:
print "%s: No such file" % movie_filename print "%s: No such file" % movie_filename
sys.exit(2) continue
movie_digest = digest.hexdigest()
def download_subtitles(movie_digest):
request_data = { request_data = {
"downloaded_subtitles_id" : movie_digest, "downloaded_subtitles_id" : movie_digest,
"mode" : "31", "mode" : "31",
@ -90,4 +92,3 @@ def download_subtitles(movie_digest):
print "%s: Subtitles fetched successfully to %s" % (movie_filename, nazwa) print "%s: Subtitles fetched successfully to %s" % (movie_filename, nazwa)
os.remove(input_filename) os.remove(input_filename)
download_subtitles(digest.hexdigest())