llama-bot/bot/cogs/fun.py

from llama import Llama
from . import _util as util

import discord
from discord.ext import commands

import nekos
import random


class Fun(commands.Cog):
    def __init__(self, bot):
        self.bot: Llama = bot

        self.quotes = self.bot.VARS["settings"]["quotes"]
        random.shuffle(self.quotes)
        self.quote_index = 0

    async def cog_check(self, ctx: commands.Context):
        if exception_or_bool := await util.on_pm(ctx.message, self.bot):
            raise exception_or_bool
        return not exception_or_bool

    @commands.command(
        aliases=[
            "quote",
        ],
        help="Shows a random llama quote.",
    )
    async def llama(self, ctx):
        # the reason why I don't use random.choice is because it may give two of the same result consecutively.
        quote = self.quotes[self.quote_index]
        await ctx.send(
            embed=discord.Embed(
                title="Llama quote that'll make your day", description=quote
            )
        )

        self.quote_index += 1  # get next quote next time this command is called
        if self.quote_index > (len(self.quotes) - 1):  # reshuffle and reset index
            random.shuffle(self.quotes)
            self.quote_index = 0

    @commands.command(
        aliases=[
            "pp",
        ],
        help="""Detects user's penis length and arranges them from largest to smallest.
This is 101% accurate.""",
        usage="""> {prefix}{command} *<user>
ex:
Show penis length of <@501277805540147220> and <@641574882382970891>
> {prefix}{command} <@501277805540147220> <@641574882382970891>
""",
    )
    async def penis(self, ctx, *users: discord.Member):
        dongs = {}
        msg = ""
        state = random.getstate()

        for user in users:
            random.seed(user.id)
            random_size = random.randint(0, 30)
            dongs[user] = ["8{}D".format("=" * random_size), random_size]

        if not users:
            random.seed(ctx.author.id)
            random_size = random.randint(0, 30)
            dongs[ctx.author] = ["8{}D".format("=" * random_size), random_size]

        random.setstate(state)
        dongs = sorted(dongs.items(), key=lambda x: x[1][1])

        for user, dong in dongs:
            msg += "**%s's size: ( %s )**\n%s\n" % (user.mention, dong[1], dong[0])

        await ctx.send(
            embed=discord.Embed(title="Here's your pp list", description=msg)
        )

    @commands.command(
        help="Call for help.",
    )
    async def fix(self, ctx: discord.ext.commands.Context):
        """
        Bug report feature.
        Removed due to excessive pinging

        old code:
        await ctx.send(f"Yo {', '.join([f'<@{fixer_id}>' for fixer_id in self.bot.fixer_ids])} fix this shit")
        """

        await ctx.send(
            f"Fuck you {ctx.message.author.mention} <:coronalol:692323993419776020>"
        )

    @commands.command(
        help="Shows image matching search term",
        usage="""> {prefix}{command} <target>
Can chose from:
**NSFW**:
`feet`, `yuri`, `trap`, `futanari`, `hololewd`, `lewdkemo`, `solog`, `feetg`, `cum`, `erokemo`, `les`, `wallpaper`, `lewdk`, `tickle`, `lewd`, `eroyuri`, `eron`, `cum_jpg`, `bj`, 'nsfw_neko_gif', `solo`, `kemonomimi`, `nsfw_avatar`, `gasm`, `anal`, `hentai`, `avatar`, `erofeet`, `holo`, `keta`, `blowjob`, `pussy`, `tits`, `holoero`, `pussy_jpg`, `pwankg`, `classic`, `kuni`, `kiss`, `femdom`, `neko`, `spank`, `erok`, `boobs`, `random_hentai_gif`, `smallboobs`, `ero`
**non NSFW**:
`feed`, `gecg`, `poke`, `slap`, `lizard`, `waifu`, `pat`, `8ball`, `cuddle`, `fox_girl`, `hug`, `smug`, `goose`, `baka`, `woof`

ex:
> {prefix}{command} neko""",
    )
    async def img(self, ctx, target: str):
        non_nsfw = [
            "feed",
            "gecg",
            "poke",
            "slap",
            "lizard",
            "waifu",
            "pat",
            "8ball",
            "cuddle",
            "fox_girl",
            "hug",
            "smug",
            "goose",
            "baka",
            "woof",
        ]
        nsfw = [
            "feet",
            "yuri",
            "trap",
            "futanari",
            "hololewd",
            "lewdkemo",
            "solog",
            "feetg",
            "cum",
            "erokemo",
            "les",
            "wallpaper",
            "lewdk",
            "tickle",
            "lewd",
            "eroyuri",
            "eron",
            "cum_jpg",
            "bj",
            "nsfw_neko_gif",
            "solo",
            "kemonomimi",
            "nsfw_avatar",
            "gasm",
            "anal",
            "hentai",
            "avatar",
            "erofeet",
            "holo",
            "keta",
            "blowjob",
            "pussy",
            "tits",
            "holoero",
            "pussy_jpg",
            "pwankg",
            "classic",
            "kuni",
            "kiss",
            "femdom",
            "neko",
            "spank",
            "erok",
            "boobs",
            "random_hentai_gif",
            "smallboobs",
            "ero",
        ]
        in_nsfw = target in nsfw
        in_non_nsfw = target in non_nsfw
        if not (in_nsfw or in_non_nsfw):
            await ctx.send(
                embed=discord.Embed(
                    title="Oops",
                    description="Image category you're looking for doesn't exist. Here's all we got:",
                )
                .add_field(name="NSFW", value=f"`{'`, `'.join(nsfw)}`", inline=False)
                .add_field(
                    name="non NSFW", value=f"`{'`, `'.join(non_nsfw)}`", inline=False
                )
            )
            return

        if in_nsfw and not ctx.message.channel.is_nsfw():
            raise discord.ext.commands.errors.NSFWChannelRequired

        image_url = nekos.img(target.lower())
        await ctx.send(
            embed=discord.Embed(
                title="Image",
                description=f"requested by: {ctx.message.author.mention}\n**[Link if you don't see the image]({image_url})**",
            )
            .set_image(url=image_url)
            .set_footer(text=f"powered by nekos.life")
        )

    @commands.command(
        help="Shows useless facts.",
    )
    async def fact(self, ctx):
        await ctx.send(
            embed=discord.Embed(title="Fact of the day", description=nekos.fact())
        )

    @commands.command(
        aliases=[
            "clapify",
        ],
        help="does the karen clap thing. Does not work with external emojis",
    )
    async def clap(self, ctx, *args):
        await ctx.send(" :clap: ".join(args))


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