didier/cogs/fun.py

from data import paginatedLeaderboard
from decorators import help
import discord
from discord.ext import commands
from enums.help_categories import Category
from functions import checks, mock, stringFormatters
from functions.database import memes, trump, dadjoke
import json
import os
import random
import requests


class Fun(commands.Cog):

    def __init__(self, client):
        self.client = client
        self.utilsCog = self.client.get_cog("Utils")

    # Don't allow any commands to work when locked
    def cog_check(self, ctx):
        return not self.client.locked

    @commands.command(name="Dadjoke", aliases=["Dj", "Dad"])
    @commands.check(checks.allowedChannels)
    @help.Category(category=Category.Fun)
    async def dadjoke(self, ctx):
        """
        Command that sends a random dad joke.
        :param ctx: Discord Context
        """
        await ctx.send(dadjoke.getRandomJoke())

    @commands.command(name="Stalin", aliases=["Ss", "StalinMotivation", "Motivate"])
    @commands.check(checks.allowedChannels)
    @help.Category(category=Category.Quotes)
    async def stalin(self, ctx):
        """
        Command that sends a random Stalin quote.
        :param ctx: Discord Context
        """
        with open("files/StalinMotivation.json", "r") as fp:
            file = json.load(fp)
            await ctx.send(file[random.randint(1, len(file))])

    @commands.command(name="Satan", aliases=["S8n", "SatanQuote"])
    @commands.check(checks.allowedChannels)
    @help.Category(category=Category.Quotes)
    async def satan(self, ctx):
        """
        Command that sends a random Satan quote.
        :param ctx: Discord Context
        """
        with open("files/SatanQuotes.json", "r") as fp:
            file = json.load(fp)
            await ctx.send(file[random.randint(1, len(file))])

    @commands.command(name="Trump")
    @commands.check(checks.allowedChannels)
    @help.Category(category=Category.Quotes)
    async def trump(self, ctx):
        """
        Command that sends a random Trump quote.
        :param ctx: Discord Context
        """
        quote = trump.getRandomQuote()
        await ctx.send("**\"{}**\"\n{} - {}".format(quote[1], quote[2], quote[3]))

    @commands.command(name="8-Ball", aliases=["8b", "8Ball"], ignore_extra=True)
    @help.Category(category=Category.Quotes)
    async def eightball(self, ctx):
        """
        Command that sends a random 8-ball response.
        :param ctx: Discord Context
        """
        with open("files/eightball.json", "r") as fp:
            file = json.load(fp)
            await ctx.send(file[random.randint(0, len(file) - 1)])

    @commands.command(name="Memegen", usage="[Meme] [Velden]")
    @commands.cooldown(1, 5, commands.BucketType.guild)
    @help.Category(category=Category.Fun)
    async def memegen(self, ctx, name="", *fields):
        """
        Command that generates memes.
        :param ctx: Discord Context
        :param name: the name of the meme
        :param fields: the fields to add to the meme
        """
        if len(fields) == 0:
            return await ctx.send("Controleer je argumenten.")

        # Get the meme info that corresponds to this name
        result = memes.getMeme(name)

        # No meme found
        if not result[0]:
            return await ctx.send(result[1])

        # Convert to list to support item assignment
        fields = list(fields)

        # If there's only one field, the user isn't required to use quotes
        if result[1][2] == 1:
            fields = [" ".join(fields)]

        # Apply mock to mocking spongebob memes
        if result[1][1] == "mocking spongebob":
            fields = list(map(mock.mock, fields))

        # X, X everywhere only takes X as an argument
        if result[1][1] == "x, x everywhere":
            fields[0] = " ".join(fields)
            fields.append(fields[0] + " everywhere")

        # List of fields to send to the API
        boxes = [{"text": ""}, {"text": ""}, {"text": ""}, {"text": ""}]

        # Add all fields required & ignore the excess ones
        for i in range(len(fields)):
            if i > 3:
                break
            boxes[i]["text"] = fields[i]

        # Check server status
        req = requests.get('https://api.imgflip.com/get_memes').json()

        if req["success"]:
            caption = {
                "template_id": result[1][0],
                "username": os.getenv("IMGFLIPNAME"),
                "password": os.getenv("IMGFLIPPASSWORD"),
                "boxes[0][text]": boxes[0]["text"],
                "boxes[1][text]": boxes[1]["text"],
                "boxes[2][text]": boxes[2]["text"],
                "boxes[3][text]": boxes[3]["text"]
            }

            # Send the POST to the API
            memeReply = requests.post('https://api.imgflip.com/caption_image', caption).json()

            if memeReply['success']:
                await ctx.send(str(memeReply['data']['url']))
                await self.utilsCog.removeMessage(ctx.message)
            else:
                await ctx.send(
                    "Error! Controleer of je de juiste syntax hebt gebruikt. Gebruik het commando "
                    "\"memes\" voor een lijst aan geaccepteerde meme-namen.")
        else:
            await ctx.send("Er is een fout opgetreden.")

    @commands.command(name="Memes")
    @commands.check(checks.allowedChannels)
    @help.Category(category=Category.Fun)
    async def memes(self, ctx):
        """
        Command that shows a list of memes in the database.
        :param ctx: Discord Context
        """
        memeList = memes.getAllMemes()

        # Turn the list into a list of [Name: fields]
        memeList = [": ".join([stringFormatters.titleCase(meme[1]),
                               str(meme[2])]) for meme in sorted(memeList, key=lambda x: x[1])]

        pages = paginatedLeaderboard.Pages(source=paginatedLeaderboard.Source(memeList, "Memes", discord.Colour.blue()),
                                           clear_reactions_after=True)
        await pages.start(ctx)

    @commands.command(name="Pjoke")
    @help.Category(category=Category.Fun)
    async def pjoke(self, ctx):
        """
        Command that sends a random programming joke.
        :param ctx: Discord Context
        """
        r = requests.get("https://official-joke-api.appspot.com/jokes/programming/random").json()
        await ctx.send(r[0]["setup"] + "\n" + r[0]["punchline"])


def setup(client):
    client.add_cog(Fun(client))