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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
import logging
import random
import discord
from discord.ext import commands
from bot.bot import Bot
from bot.utils.pagination import ImagePaginator
log = logging.getLogger(__name__)
class Reddit(commands.Cog):
"""Fetches reddit posts."""
def __init__(self, bot: Bot):
self.bot = bot
async def fetch(self, url: str) -> dict:
"""Send a get request to the reddit API and get json response."""
session = self.bot.http_session
params = {
"limit": 50
}
headers = {
"User-Agent": "Iceman"
}
async with session.get(url=url, params=params, headers=headers) as response:
return await response.json()
@commands.command(name="reddit")
@commands.cooldown(1, 10, commands.BucketType.user)
async def get_reddit(self, ctx: commands.Context, subreddit: str = "python", sort: str = "hot") -> None:
"""
Fetch reddit posts by using this command.
Gets a post from r/python by default.
Usage:
--> .reddit [subreddit_name] [hot/top/new]
"""
pages = []
sort_list = ["hot", "new", "top", "rising"]
if sort.lower() not in sort_list:
await ctx.send(f"Invalid sorting: {sort}\nUsing default sorting: `Hot`")
sort = "hot"
data = await self.fetch(f"https://www.reddit.com/r/{subreddit}/{sort}/.json")
try:
posts = data["data"]["children"]
except KeyError:
await ctx.send("Subreddit not found!")
return
if not posts:
await ctx.send("No posts available!")
return
if posts[0]["data"]["over_18"] is True:
await ctx.send(
"You cannot access this Subreddit as it is meant for those who "
"are 18 years or older."
)
return
embed_titles = ""
# Chooses k unique random elements from a population sequence or set.
random_posts = random.sample(posts, k=min(len(posts), 5))
# -----------------------------------------------------------
# This code below is bound of change when the emojis are added.
upvote_emoji = self.bot.get_emoji(755845219890757644)
comment_emoji = self.bot.get_emoji(755845255001014384)
user_emoji = self.bot.get_emoji(755845303822974997)
text_emoji = self.bot.get_emoji(676030265910493204)
video_emoji = self.bot.get_emoji(676030265839190047)
image_emoji = self.bot.get_emoji(676030265734201344)
reddit_emoji = self.bot.get_emoji(676030265734332427)
# ------------------------------------------------------------
for i, post in enumerate(random_posts, start=1):
post_title = post["data"]["title"][0:50]
post_url = post["data"]["url"]
if post_title == "":
post_title = "No Title."
elif post_title == post_url:
post_title = "Title is itself a link."
# ------------------------------------------------------------------
# Embed building.
embed_titles += f"**{i}.[{post_title}]({post_url})**\n"
image_url = " "
post_stats = f"{text_emoji}" # Set default content type to text.
if post["data"]["is_video"] is True or "youtube" in post_url.split("."):
# This means the content type in the post is a video.
post_stats = f"{video_emoji} "
elif post_url.endswith("jpg") or post_url.endswith("png") or post_url.endswith("gif"):
# This means the content type in the post is an image.
post_stats = f"{image_emoji} "
image_url = post_url
votes = f"{upvote_emoji}{post['data']['ups']}"
comments = f"{comment_emoji}\u2002{ post['data']['num_comments']}"
post_stats += (
f"\u2002{votes}\u2003"
f"{comments}"
f"\u2003{user_emoji}\u2002{post['data']['author']}\n"
)
embed_titles += f"{post_stats}\n"
page_text = f"**[{post_title}]({post_url})**\n{post_stats}\n{post['data']['selftext'][0:200]}"
embed = discord.Embed()
page_tuple = (page_text, image_url)
pages.append(page_tuple)
# ------------------------------------------------------------------
pages.insert(0, (embed_titles, " "))
embed.set_author(name=f"r/{posts[0]['data']['subreddit']} - {sort}", icon_url=reddit_emoji.url)
await ImagePaginator.paginate(pages, ctx, embed)
def setup(bot: Bot) -> None:
"""Load the Reddit cog."""
bot.add_cog(Reddit(bot))
|
