CodeMirror/bulk-downloader-for-reddit
1
0
Fork 0
mirror of synced 2024-06-03 10:54:38 +12:00
Code Issues Releases Wiki Activity

Add search functionality

Browse source
This commit is contained in:
Serene-Arc 2021-02-11 16:31:28 +10:00 committed by Ali Parlakci
parent a7f1db14e5
commit 91ae9924c3
1 changed files with 11 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
bulkredditdownloader/downloader.py
View file

@ -90,15 +90,18 @@ class RedditDownloader:
def _get_subreddits(self, args: argparse.Namespace) -> list[praw.models.ListingGenerator]:
if args.subreddit:
subreddits = [self.reddit_instance.subreddit(chosen_subreddit) for chosen_subreddit in args.subreddit]
if self.sort_filter is RedditTypes.SortType.NEW:
sort_function = praw.models.Subreddit.new
elif self.sort_filter is RedditTypes.SortType.RISING:
sort_function = praw.models.Subreddit.rising
elif self.sort_filter is RedditTypes.SortType.CONTROVERSIAL:
sort_function = praw.models.Subreddit.controversial
if args.search:
return [reddit.search(args.search, sort=self.sort_filter.name.lower()) for reddit in subreddits]
else:
sort_function = praw.models.Subreddit.hot
return [sort_function(reddit) for reddit in subreddits]
if self.sort_filter is RedditTypes.SortType.NEW:
sort_function = praw.models.Subreddit.new
elif self.sort_filter is RedditTypes.SortType.RISING:
sort_function = praw.models.Subreddit.rising
elif self.sort_filter is RedditTypes.SortType.CONTROVERSIAL:
sort_function = praw.models.Subreddit.controversial
else:
sort_function = praw.models.Subreddit.hot
return [sort_function(reddit) for reddit in subreddits]
else:
return []