ModerateWinGuy/BDFR_Web
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects 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
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]: def _get_subreddits(self, args: argparse.Namespace) -> list[praw.models.ListingGenerator]:
if args.subreddit: if args.subreddit:
subreddits = [self.reddit_instance.subreddit(chosen_subreddit) for chosen_subreddit in args.subreddit] subreddits = [self.reddit_instance.subreddit(chosen_subreddit) for chosen_subreddit in args.subreddit]
if self.sort_filter is RedditTypes.SortType.NEW: if args.search:
sort_function = praw.models.Subreddit.new return [reddit.search(args.search, sort=self.sort_filter.name.lower()) for reddit in subreddits]
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: else:
sort_function = praw.models.Subreddit.hot if self.sort_filter is RedditTypes.SortType.NEW:
return [sort_function(reddit) for reddit in subreddits] 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: else:
return [] return []