lauchmelder/RedditWordCloud
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Added support for top posts

Browse source
This commit is contained in:
Robert 2020-07-14 00:06:21 +02:00
parent f99109791b
commit cbf586b362
1 changed files with 9 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
cloud.py
View file

@ -44,6 +44,8 @@ parser.add_argument("-boost", metavar="freq_boost", type=float,
help="The boost a word that isn't in the wordlist gets (Default 1)") help="The boost a word that isn't in the wordlist gets (Default 1)")
parser.add_argument("-blow", metavar="freq_blow", type=float, parser.add_argument("-blow", metavar="freq_blow", type=float,
help="The \"anti-boost\" a word that is in the wordlist gets (Default 1)") help="The \"anti-boost\" a word that is in the wordlist gets (Default 1)")
parser.add_argument("--top", action="store_true",
help="Use Top posts instead of Hot posts")
args = parser.parse_args() args = parser.parse_args()
@ -93,14 +95,19 @@ reddit = praw.Reddit(client_id=settings["client_id"],
user_agent="Windows10:RWC:1.0") user_agent="Windows10:RWC:1.0")
if args.sub: if args.sub:
posts = reddit.subreddit(args.id).hot(limit=args.p) if args.top:
posts = reddit.subreddit(args.id).top("all")
else:
posts = reddit.subreddit(args.id).hot(limit=args.p)
else: else:
posts = [reddit.submission(id=args.id)] posts = [reddit.submission(id=args.id)]
i = 1 i = 1
comments = [] comments = []
posts = list(posts)
length = len(posts)
for post in posts: for post in posts:
print(f"\rFetching comments... {i}/{args.p} ", end=" ", flush=True) print(f"\rFetching comments... {i}/{length} ", end=" ", flush=True)
post.comments.replace_more(limit=args.n) post.comments.replace_more(limit=args.n)
for top_level_comment in post.comments: for top_level_comment in post.comments:
comments.extend(fetch_comments(top_level_comment)) comments.extend(fetch_comments(top_level_comment))