add roleplay cog

roleplay/__init__.py

@@ -0,0 +1,7 @@
+import os
+from .roleplay import RolePlay
+from redbot.core import Config
+
+
+def setup(bot):
+    bot.add_cog(RolePlay())

roleplay/roleplay.py

@@ -0,0 +1,392 @@
+import discord
+from discord.ext import commands
+from redbot.core.utils.chat_formatting import escape, italics, pagify, box
+from redbot.core import Config, checks, commands
+import random
+from random import randint
+from random import choice
+from enum import Enum
+from urllib.parse import quote_plus
+import datetime
+import time
+import aiohttp
+import asyncio
+import os
+import json
+
+class RolePlay(commands.Cog):
+    def __init__(self):
+        super().__init__()
+
+        mass_mentions = True
+        self.config = Config.get_conf(self, identifier=3674895735)
+        self.default_guild = {
+            "slap_items": [
+                "a floppy disk",
+                "a book",
+                "a nuke",
+                "a loaf of bread",
+                "my hand",
+                "a pack of ramen",
+                "lotsa spaghetti",
+                "a brick",
+                "a slice of cheese",
+                "my foot"],
+            "high_iq_msgs": [
+                "wow!",
+                "that's pretty big.",
+                ":wink:",
+                "niiicceee.",
+                "someone here is actually smart.",
+                "thats a dab.",
+                "<:aureliawink:549481308519399425>",
+                "you must of watched Rick and Morty."],
+            "low_iq_msgs": [
+                ":rofl:",
+                "oof.",
+                "you must have a lot of trouble in life.",
+                "damn.",
+                "awww you're special aren't you.",
+                ":crying_cat_face:",
+                "god I'm sorry (not).",
+                "I didn't know people could have IQ that low."]
+        }
+
+        self.config.register_guild(**self.default_guild)
+
+    @commands.command()
+    @commands.guild_only()
+    async def hug(self, ctx, hug_target, *, intensity):
+        """Hugs a user with optional intensity!
+        Example: .hug *username* 4
+
+        Up to 10 intensity levels."""
+        user = ctx.guild.get_member_named(hug_target + " " + intensity)
+        if user is not None:
+            name = italics(user.display_name)
+            msg = "(っ´▽｀)っ" + name
+            await ctx.send(msg)
+            return
+        args = intensity.split(" ")
+        try:
+            intensity = int(args[-1])
+        except:
+            await ctx.send("Member not found.")
+            return
+        args.insert(0, hug_target)
+        user = ctx.guild.get_member_named(" ".join(args[:-1]))
+        if user is not None:
+            name = italics(user.display_name)
+            if intensity <= 0:
+                msg = "(っ˘̩╭╮˘̩)っ" + name
+            elif intensity <= 3:
+                msg = "(っ´▽｀)っ" + name
+            elif intensity <= 6:
+                msg = "╰(*´︶`*)╯" + name
+            elif intensity <= 9:
+                msg = "(つ≧▽≦)つ" + name
+            elif intensity >= 10:
+                msg = "(づ￣ ³￣)づ{} ⊂(´・ω・｀⊂)".format(name)
+            await ctx.send(msg)
+        else:
+            await ctx.send("Member not found.")
+
+    @commands.group(invoke_without_command=True)
+    async def slap(self, ctx, *, user: discord.Member = None):
+        """Slap a user"""
+        guild = ctx.guild
+        slap_items = await self.config.guild(guild).slap_items()
+        botid = ctx.bot.user.id
+        if user is None:
+            user = ctx.message.author
+            await ctx.send("Don't make me slap you instead " + user.display_name)
+            return
+        elif user.id == botid:
+            user = ctx.message.author
+            botname = ctx.bot.user.name
+            await ctx.send("`-" + botname + " slaps " + user.display_name +
+                               " multiple times with " +
+                               (choice(slap_items) + "-`"))
+        else:
+            await ctx.send("`-slaps " + user.display_name + " with " +
+                               (choice(slap_items) + "-`"))
+
+    @slap.command(name="add")
+    @checks.admin()
+    async def _add_slap(self, ctx, *, slap_item):
+        """Adds an item to use for slaps!"""
+        guild = ctx.guild
+        slap_items = await self.config.guild(guild).slap_items()
+        if slap_item not in slap_items:
+            async with self.config.guild(guild).slap_items() as current_slaps:
+                current_slaps.append(slap_item)
+            await ctx.send("Item '{}' added to the server's slap items list.".format(slap_item))
+        else:
+            await ctx.send("Item '{}' is already in the server's slap item list.".format(slap_item))
+
+    @slap.command(name="remove")
+    @checks.admin()
+    async def _remove_slap(self, ctx, slap_item: str=""):
+        """Removes item to use for slaps!"""
+        guild = ctx.guild
+        slap_items = await self.config.guild(guild).slap_items()
+        if slap_item in slap_items:
+            async with self.config.guild(guild).slap_items() as current_slaps:
+                current_slaps.remove(slap_item)
+            await ctx.send("Item '{}' deleted from the server's slap items list.".format(slap_item))
+        else:
+            await ctx.send("Item '{}' does not exist in the server's slap items list.".format(slap_item))
+
+    @slap.command(name="list")
+    @checks.admin()
+    async def _list_slap(self, ctx):
+        """Prints list of slaps"""
+        guild = ctx.guild
+        slap_items = await self.config.guild(guild).slap_items()
+        msg = ""
+        for item in slap_items:
+            msg += "+ {}\n".format(item)
+        pages = pagify(msg) # pages is an iterator of pages
+
+        for page in pages:
+            await ctx.send(box(page, lang="diff"))
+
+    @slap.command(name="import")
+    @checks.is_owner()
+    async def _import_slap(self, ctx, path_to_import):
+        """Imports slaps from jsons.
+
+        Specifiy the **path** to the json to import slaps from.
+
+        *i.e.: /path/containing/json/*"""
+        bot = ctx.bot
+        guild = ctx.guild
+        path_to_slaps = os.path.join(path_to_import, "items.json")
+
+        try:
+            with open(path_to_slaps) as raw_slaps:
+                import_slaps = json.load(raw_slaps)
+                await self.config.guild(guild).slap_items.set(import_slaps)
+                await ctx.send("Slaps imported.")
+        except FileNotFoundError:
+            await ctx.send("Invalid path to json file.")
+        except json.decoder.JSONDecodeError:
+            await ctx.send("Invalid or malformed json file.")
+
+
+    @commands.group(invoke_without_command=True)
+    async def iq(self, ctx, *users : discord.Member):
+        """
+        Gets IQ of a user. Use multiple users to compare IQs
+        """
+        if not users:
+            await ctx.bot.send_help_for(ctx, self.iq)
+            return
+        guild = ctx.guild
+        high_iq_msgs = await self.config.guild(guild).high_iq_msgs()
+        low_iq_messages = await self.config.guild(guild).low_iq_messages()
+        state = random.getstate()
+        iqs = {}
+        msg = ""
+        for user in users:
+            if user.id == 216319397944492033 or user.id == 213027180865781761 or user.id == 559915721627402241:
+                iqs[user] = "0"
+            else:
+                random.seed(user.id)
+                iqs[user] = "{}".format(random.randint(0, 250))
+
+        random.setstate(state)
+        iqs = sorted(iqs.items(), key=lambda x: x[1])
+
+        for user, iq in iqs:
+            msg += "{}'s iq is {}, {}\n".format(user.display_name, iq, choice(high_iq_msgs) if int(iq) > 130 else choice(low_iq_messages))
+
+        await ctx.send(msg)
+
+    @iq.command(name="list")
+    @checks.admin()
+    async def _list_iq(self, ctx):
+        """Prints a list of all IQ phrases."""
+        guild = ctx.guild
+        high_iq_msgs = await self.config.guild(guild).high_iq_msgs()
+        low_iq_msgs = await self.config.guild(guild).low_iq_msgs()
+        msg1 = "HIGH IQ PHRASES:\n"
+        msg2 = "LOW IQ PHRASES:\n"
+
+        for high_phrase in high_iq_msgs:
+            msg1 += "+ {}\n".format(high_phrase)
+        high_pages = pagify(msg1) # pages is an iterator of pages
+
+        for low_phrase in low_iq_msgs:
+            msg2 += "+ {}\n".format(low_phrase)
+        low_pages = pagify(msg2) # pages is an iterator of pages
+
+        for high_page in high_pages:
+            await ctx.send(box(high_page, lang="diff"))
+
+        for low_page in low_pages:
+            await ctx.send(box(low_page, lang="diff"))
+
+    @iq.command(name="addhigh")
+    @checks.admin()
+    async def _addhigh_iq(self, ctx, *, new_high_iq_msg: str):
+        """Adds a postive phrase for high IQ results!"""
+        guild = ctx.guild
+        high_iq_msgs = await self.config.guild(guild).high_iq_msgs()
+        if new_high_iq_msg not in high_iq_msgs:
+            async with self.config.guild(guild).high_iq_msgs() as current_hi_iq_msgs:
+                current_hi_iq_msgs.append(new_high_iq_msg)
+            await ctx.send("Phrase '{}' added to the server's High IQ list.".format(new_high_iq_msg))
+        else:
+            await ctx.send("Phrase '{}' is already in the server's High IQ list.".format(new_high_iq_msg))
+
+    @iq.command(name="addlow")
+    @checks.admin()
+    async def _addlow_iq(self, ctx, *, new_low_iq_msg: str):
+        """Adds a derogatory phrase for low IQ results!"""
+        guild = ctx.guild
+        low_iq_msgs = await self.config.guild(guild).low_iq_msgs()
+        if new_low_iq_msg not in low_iq_msgs:
+            async with self.config.guild(guild).low_iq_msgs() as current_low_iq_msgs:
+                current_low_iq_msgs.append(new_low_iq_msg)
+            await ctx.send("Phrase '{}' added to the server's Low IQ list. Heh, nice one.".format(new_low_iq_msg))
+        else:
+            await ctx.send("Phrase '{}' is already in the server's Low IQ list.".format(new_low_iq_msg))
+
+    @iq.command(name="removehigh")
+    @checks.admin()
+    async def _removehigh_iq(self, ctx, high_phrase: str=""):
+        """Removes phrases for high IQ's!"""
+        guild = ctx.guild
+        high_iq_msgs = await self.config.guild(guild).high_iq_msgs()
+        if high_phrase in high_iq_msgs:
+            async with self.config.guild(guild).high_iq_msgs() as current_phrases:
+                current_phrases.remove(high_phrase)
+            await ctx.send("Phrase '{}' deleted from the server's high IQ messages.".format(high_phrase))
+        elif high_phrase == "":
+            if self.default_guild["high_iq_msgs"] != high_iq_msgs:
+                await self.config.guild(guild).high_iq_msgs.set(self.default_guild["high_iq_msgs"])
+                await ctx.send("Reverted the server to the default high IQ messages.")
+            else:
+                await ctx.send("Server is already using the default high IQ messages.")
+        else:
+            await ctx.send("Phrase '{}' does not exist in the server's high IQ messages.".format(high_phrase))
+
+    @iq.command(name="removelow")
+    @checks.admin()
+    async def _removelow_iq(self, ctx, low_phrase: str=""):
+        """Removes phrases for low IQ's!"""
+        guild = ctx.guild
+        low_iq_msgs = await self.config.guild(guild).low_iq_msgs()
+        if low_phrase in low_iq_msgs:
+            async with self.config.guild(guild).low_iq_msgs() as current_phrases:
+                current_phrases.remove(low_phrase)
+            await ctx.send("Phrase '{}' deleted from the server's low IQ messages.".format(low_phrase))
+        elif low_phrase == "":
+            if self.default_guild["low_iq_msgs"] != low_iq_msgs:
+                await self.config.guild(guild).low_iq_msgs.set(self.default_guild["low_iq_msgs"])
+                await ctx.send("Reverted the server to the default low IQ messages.")
+            else:
+                await ctx.send("Server is already using the default low IQ messages.")
+        else:
+            await ctx.send("Phrase '{}' does not exist in the server's low IQ messages.".format(low_phrase))
+
+    @commands.command()
+    @commands.cooldown(1, 6, commands.BucketType.guild)
+    async def army(self, ctx, horses: int):
+        """
+        Summon an army of Aurelias. Max 20
+        """
+        army_emoji = "<a:trottingaurelia:568577886164877312>"
+        if horses > 20:
+            await ctx.send("Too many Aurelias!")
+            return
+        msg = ""
+        if horses == 1:
+            largest_factor = 1
+        else:
+            largest_factor = [x for x in range(1, horses) if horses % x == 0][-1]
+        #largest_factor = [x for x in largest_factor if x <= 15][-1]
+        if largest_factor == 1:
+            largest_factor = horses
+            rows = 1
+        else:
+            rows = horses // largest_factor
+        rows = rows if rows > 0 else 1
+        for _ in range(rows):
+            for _ in range(largest_factor):
+                msg += "{} ".format(army_emoji)
+            msg += "\n"
+            if len(msg) + len(army_emoji) + 20 > 2000:
+                await ctx.send(msg)
+                msg = ""
+
+        if msg != "":
+            await ctx.send(msg)
+
+    @commands.command()
+    @commands.guild_only()
+    async def boop(self, ctx, boop_target, *, intensity):
+        """
+        Boops a user. 10 intensity levels.
+        """
+        user = ctx.guild.get_member_named(boop_target + " " + intensity)
+        if user is not None:
+            name = italics(user.display_name)
+            msg = "/) {}" + name
+            await ctx.send(msg)
+            return
+        args = intensity.split(" ")
+        try:
+            intensity = int(args[-1])
+        except:
+            await ctx.send("Can't boop what I can't see!")
+            return
+        args.insert(0, boop_target)
+        user = ctx.guild.get_member_named(" ".join(args[:-1]))
+        if user is not None:
+            name = italics(user.display_name)
+            if intensity <= 3:
+                msg = "/) {}".format(name)
+            elif intensity <= 6:
+                msg = "**/)** {}".format(name)
+            elif intensity <= 9:
+                msg = "**__/)__** {}".format(name)
+            elif intensity >= 10:
+                msg = "**__/)__** {} **__(\\__**".format(name)
+            await ctx.send(msg)
+        else:
+            await ctx.send("Can't boop what I can't see!")
+
+    @commands.command()
+    @commands.guild_only()
+    async def bap(self, ctx, *, user: discord.Member):
+        """
+        Baps a user
+        """
+        if user.id == ctx.bot.user.id:
+            await ctx.send(":newspaper2: :newspaper2: :newspaper2: " + italics(ctx.message.author.display_name))
+        else:
+            await ctx.send(":newspaper2: " + italics(user.display_name) )
+
+    @commands.command()
+    async def flip(self, ctx, *, user: discord.Member = None):
+        """Flip a coin... or a user.
+        Defaults to a coin.
+        """
+        if user is not None:
+            msg = ""
+            if user.id == ctx.bot.user.id:
+                user = ctx.author
+                msg = _("Nice try. You think this is funny?\n How about *this* instead:\n\n")
+            char = "abcdefghijklmnopqrstuvwxyz"
+            tran = "ɐqɔpǝɟƃɥᴉɾʞlɯuodbɹsʇnʌʍxʎz"
+            table = str.maketrans(char, tran)
+            name = user.display_name.translate(table)
+            char = char.upper()
+            tran = "∀qƆpƎℲפHIſʞ˥WNOԀQᴚS┴∩ΛMX⅄Z"
+            table = str.maketrans(char, tran)
+            name = name.translate(table)
+            await ctx.send(msg + "(╯°□°）╯︵ " + name[::-1])
+        else:
+            await ctx.send(_("*flips a coin and... ") + choice([_("HEADS!*"), _("TAILS!*")]))