-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.py
70 lines (50 loc) · 2.06 KB
/
main.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
68
69
70
import discord
from discord.ext import commands
from dotenv import load_dotenv
import os
import praw
from datetime import datetime
import random
bot = commands.Bot(".", intents=discord.Intents.all())
flairwhitelist = ["Cat Picture", "Cat Picture - OC", "Adoption"]
filetypes = ["jpeg", "jpg", "webp", "png", "gif"]
load_dotenv()
reddit = praw.Reddit(client_id=os.getenv("CLIENTID"), client_secret=os.getenv("CLIENTSECRET"), user_agent="Discord Cat Bot Made by JugieNoob: github.com/JugieNoob/Discord-Cat-Bot", check_for_async=False)
def fetchCat():
embed = discord.Embed(color=discord.Color.random())
embed.set_footer(text="r/cats", icon_url=bot.user.avatar)
submissions = []
for submission in reddit.subreddit("Cats").hot(limit=100):
if not submission.stickied and submission.link_flair_text in flairwhitelist and not submission.is_video:
submissions.append(submission)
pass
image = ""
post = random.randint(0, len(submissions) - 1)
if "gallery" in submissions[post].url:
gallery = reddit.submission(url = submissions[post].url).media_metadata
for image_item in gallery.values():
image =image_item["s"]["u"]
break
else:
for file in filetypes:
if file in submissions[post].url:
image = submissions[post].url
try:
embed.title = submissions[post].title
except():
print("Title is too long! Attempting to shorten title...")
embed.title = f"{submissions[post].title[0:200]}..."
embed.url = f"https://www.reddit.com/r/cats/comments/{submissions[post]}"
embed.description = f"u/{str(submissions[post].author)}"
embed.set_image(url=image)
embed.timestamp = datetime.fromtimestamp(submissions[post].created)
return embed
@bot.event
async def on_ready():
print(f'Logged in as {bot.user}')
@bot.command(name="cat")
async def self(ctx):
catembed = fetchCat()
await ctx.send(embed=catembed)
# Start the bot.
bot.run(os.getenv("TOKEN"))