From cc4bd22fab3ee3fa645262b706663bd14e3c5061 Mon Sep 17 00:00:00 2001 From: stockbsd Date: Thu, 14 Nov 2019 17:26:24 +0800 Subject: [PATCH] add --subdir switch --- README.md | 1 + twitter_dl/__init__.py | 2 +- twitter_dl/__main__.py | 6 +++++- 3 files changed, 7 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index d3a1d70..8af88aa 100644 --- a/README.md +++ b/README.md @@ -41,6 +41,7 @@ optional arguments: --thread-number THREAD_NUMBER --coro-number CORO_NUMBER --since SID + --subdir used with --file to save media to different dirs for each user ``` ``` diff --git a/twitter_dl/__init__.py b/twitter_dl/__init__.py index 5d2a741..280d67f 100644 --- a/twitter_dl/__init__.py +++ b/twitter_dl/__init__.py @@ -1,4 +1,4 @@ from .downloader import Downloader from .threaded_aio_dlder import AioDownloader -version = "0.1.8" +version = "0.1.9" diff --git a/twitter_dl/__main__.py b/twitter_dl/__main__.py index 5970320..f2762ac 100644 --- a/twitter_dl/__main__.py +++ b/twitter_dl/__main__.py @@ -70,6 +70,9 @@ def main(): parser.add_argument("--thread-number", type=int, default=2) parser.add_argument("--coro-number", type=int, default=5) parser.add_argument('-v', '--verbose', action='count', default=0) + parser.add_argument( + "--subdir", help="add subdir for each user", default=False, action="store_true" + ) args = parser.parse_args() level = logging.DEBUG if args.verbose>1 else ( @@ -106,7 +109,8 @@ def main(): id = line.strip() if id and not id.startswith('#'): try: - downloader.download_media_of_user(id, args.dest, args.size, + dest = os.path.join(args.dest, id) if args.subdir else args.dest + downloader.download_media_of_user(id, dest, args.size, args.limit, args.rts, args.video, args.photo, args.since) except Exception as e: pass