From 945116063f3b5e545621b5ac06881c5ccab13b4b Mon Sep 17 00:00:00 2001 From: Serene-Arc Date: Thu, 11 Mar 2021 11:26:10 +1000 Subject: [PATCH] Allow for ID or URL to be provided for submissions --- bulkredditdownloader/downloader.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/bulkredditdownloader/downloader.py b/bulkredditdownloader/downloader.py index 06cf8e3..154865b 100644 --- a/bulkredditdownloader/downloader.py +++ b/bulkredditdownloader/downloader.py @@ -175,7 +175,10 @@ class RedditDownloader: def _get_submissions_from_link(self) -> list[list[praw.models.Submission]]: supplied_submissions = [] for sub_id in self.args.link: - supplied_submissions.append(self.reddit_instance.submission(id=sub_id)) + if len(sub_id) == 6: + supplied_submissions.append(self.reddit_instance.submission(id=sub_id)) + else: + supplied_submissions.append(self.reddit_instance.submission(url=sub_id)) return [supplied_submissions] def _determine_sort_function(self):