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
|
import logging
import random
import discord
from discord.ext import commands
from discord.ext.commands.cooldowns import BucketType
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, 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:
return await ctx.send('Subreddit not found!')
if not posts:
return await ctx.send('No posts available!')
if posts[0]["data"]["over_18"] is True:
return await ctx.send(
"You cannot access this Subreddit as it is meant for those who "
"are 18 years or older."
)
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))
|
