From 0c87e600944605498afd21c7f6dd58079d64616e Mon Sep 17 00:00:00 2001 From: Ethan Dalool Date: Mon, 3 Jun 2019 14:17:31 -0700 Subject: [PATCH] Fix threaded_dl thread_count being a string. --- ThreadedDL/threaded_dl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ThreadedDL/threaded_dl.py b/ThreadedDL/threaded_dl.py index 3789ce1..8fab6aa 100644 --- a/ThreadedDL/threaded_dl.py +++ b/ThreadedDL/threaded_dl.py @@ -49,7 +49,7 @@ def threaded_dl( for (index, url) in enumerate(clean_url_list(urls)): - while len(threads) == thread_count: + while len(threads) >= thread_count: threads = remove_finished(threads) time.sleep(0.1) @@ -96,7 +96,7 @@ def main(argv): parser = argparse.ArgumentParser(description=__doc__) parser.add_argument('url_file') - parser.add_argument('thread_count', nargs='?', default=None) + parser.add_argument('thread_count', nargs='?', type=int, default=None) parser.add_argument('filename_format', nargs='?', default='{now}_{index}_{basename}') parser.add_argument('--timeout', dest='timeout', default=15) parser.set_defaults(func=threaded_dl_argparse)