-
Notifications
You must be signed in to change notification settings - Fork 0
/
reddit.py
67 lines (54 loc) · 2.32 KB
/
reddit.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
# IMPORTS
import praw
import os
# REDDIT KEYS
CLIENT_ID = os.getenv('CLIENT_ID')
CLIENT_SECRET = os.getenv('CLIENT_SECRET')
USER_AGENT = os.getenv('USER_AGENT')
USERNAME = os.getenv('USERNAME')
PASSWORD = os.getenv('PASSWORD')
class RedditWrapper:
def __init__(self):
"""
Initialize the RedditWrapper instance with a PRAW Reddit object.
The constructor sets up a PRAW Reddit object using the provided
API credentials (client ID, client secret, user agent, username,
and password), which are imported from the config module.
"""
self.reddit = praw.Reddit(
client_id=CLIENT_ID,
client_secret=CLIENT_SECRET,
user_agent=USER_AGENT,
username=USERNAME,
password=PASSWORD,
)
def get_top_post_title(self, time: str, subreddit: str) -> str:
"""
Get the title of the top post for the specified time period in the given subreddit.
Args:
time (str): The time period for which the top post should be retrieved.
Valid values are 'hour', 'day', 'week', 'month', 'year', and 'all'.
subreddit (str): The name of the subreddit to fetch the top post from.
Returns:
str: The title of the top post for the specified time period in the given subreddit.
"""
subreddit = self.reddit.subreddit(subreddit)
top_post = subreddit.top(time_filter=time, limit=1)
for submission in top_post:
title = submission.title
return title
def get_top_post_url(self, time: str, subreddit: str) -> str:
"""
Get the URL of the top post for the specified time period in the given subreddit.
Args:
time (str): The time period for which the top post should be retrieved.
Valid values are 'hour', 'day', 'week', 'month', 'year', and 'all'.
subreddit (str): The name of the subreddit to fetch the top post from.
Returns:
str: The URL of the top post for the specified time period in the given subreddit.
"""
subreddit = self.reddit.subreddit(subreddit)
top_post = subreddit.top(time_filter=time, limit=1)
for submission in top_post:
url = submission.url
return url