mirror of https://github.com/sgoudham/Enso-Bot.git
Making ~doggo it's own python file
parent
96ec925bd3
commit
2c39563895
@ -0,0 +1,129 @@
|
||||
import asyncio
|
||||
import random
|
||||
import string
|
||||
|
||||
from aiohttp import request
|
||||
from discord import Colour, Embed
|
||||
from discord.ext import commands
|
||||
from discord.ext.commands import BucketType, cooldown, command
|
||||
|
||||
import settings
|
||||
from cogs.anime.interactive import error_function
|
||||
from settings import colour_list, time
|
||||
|
||||
|
||||
# Set up the cog
|
||||
class Doggo(commands.Cog):
|
||||
def __init__(self, bot):
|
||||
self.bot = bot
|
||||
|
||||
@command(name="doggo", aliases=["Doggo"])
|
||||
@cooldown(1, 1, BucketType.user)
|
||||
async def doggo(self, ctx, breed=None):
|
||||
"""Allows an API to grab images of dogs"""
|
||||
|
||||
# If the channel that the command has been sent is in the list of accepted channels
|
||||
if str(ctx.channel) in settings.channels:
|
||||
|
||||
# Set member as the author
|
||||
member = ctx.message.author
|
||||
# Get the member avatar
|
||||
userAvatar = member.avatar_url
|
||||
|
||||
# Initialise array to store doggo pics
|
||||
b_list = []
|
||||
|
||||
# If a breed if specified
|
||||
if breed:
|
||||
# Get the lowercase string input
|
||||
lowercase_breed = breed.lower()
|
||||
|
||||
# If the user wants to know what breeds there are
|
||||
if lowercase_breed == "breeds":
|
||||
# Get the list of breeds
|
||||
breed_url = "https://dog.ceo/api/breeds/list/all"
|
||||
|
||||
# Using API, retrieve the full list of breeds available
|
||||
async with request("GET", breed_url, headers={}) as response:
|
||||
if response.status == 200:
|
||||
data = await response.json()
|
||||
breed_link = data["message"]
|
||||
|
||||
# Store every Doggo in an array
|
||||
for doggo in breed_link:
|
||||
b_list.append(doggo)
|
||||
|
||||
# Join together all the breeds into a string
|
||||
doggo_string = string.capwords(", ".join(b_list))
|
||||
|
||||
# Tell the user to try the breeds listed below
|
||||
await ctx.send(f"Try the Breeds listed below!\n{doggo_string}")
|
||||
|
||||
# If no breed has been specified
|
||||
else:
|
||||
|
||||
# Grab a random image of a doggo with the breed specified
|
||||
image_url = f"https://dog.ceo/api/breed/{lowercase_breed}/images/random"
|
||||
|
||||
# Using API, retrieve the image of a doggo of the breed specified
|
||||
async with request("GET", image_url, headers={}) as response:
|
||||
if response.status == 200:
|
||||
data = await response.json()
|
||||
image_link = data["message"]
|
||||
|
||||
# Set up the embed for a doggo image
|
||||
doggo_embed = Embed(
|
||||
title=f"**It's a {lowercase_breed.capitalize()} Doggo!!** ",
|
||||
colour=Colour(random.choice(colour_list)),
|
||||
timestamp=time)
|
||||
doggo_embed.set_image(url=image_link)
|
||||
doggo_embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar))
|
||||
|
||||
# Send the doggo image
|
||||
await ctx.send(embed=doggo_embed)
|
||||
|
||||
else:
|
||||
|
||||
# Send error message that Doggo was not found!
|
||||
await ctx.send(
|
||||
"Doggo Not Found! Please do **~doggo `breeds`** to see the full list of Doggos!")
|
||||
else:
|
||||
|
||||
# Grab a random image of a doggo of any breed
|
||||
image_url = "https://dog.ceo/api/breeds/image/random"
|
||||
|
||||
# Using API, retrieve the image of a doggo of any breed
|
||||
async with request("GET", image_url, headers={}) as response:
|
||||
if response.status == 200:
|
||||
data = await response.json()
|
||||
image_link = data["message"]
|
||||
|
||||
# Set up the embed for a random doggo image
|
||||
doggo_embed = Embed(
|
||||
title=f"**Doggo!** ",
|
||||
colour=Colour(random.choice(colour_list)),
|
||||
timestamp=time)
|
||||
doggo_embed.set_image(url=image_link)
|
||||
doggo_embed.set_footer(text=f"Requested by {member}", icon_url='{}'.format(userAvatar))
|
||||
|
||||
# Send random doggo image to the channel
|
||||
await ctx.send(embed=doggo_embed)
|
||||
|
||||
else:
|
||||
|
||||
# Send error message that Doggo was not found!
|
||||
await ctx.send(
|
||||
"Doggo Not Found! Please do **~doggo `breeds`** to see the full list of Doggos!")
|
||||
else:
|
||||
|
||||
# Call error_function() and display it to the user
|
||||
message = await ctx.send(error_function())
|
||||
|
||||
# Let the user read the message for 2.5 seconds
|
||||
await asyncio.sleep(2.5)
|
||||
# Delete the message
|
||||
await message.delete()
|
||||
|
||||
|
||||
def setup(bot):
|
||||
bot.add_cog(Doggo(bot))
|
Loading…
Reference in New Issue