juni/aikaterna-cogs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Pupper] Initial commit

Browse Source
This commit is contained in:
aikaterna
2019-08-06 18:58:40 -07:00
parent d57ba2d78a
commit d4a70d5782
4 changed files with 244 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -31,6 +31,8 @@ partycrash - A port of Will's partycrash command from the v2 Admin cog. This cog
pingtime - Show all shards' pingtimes.
pupper - A cog for Ryan5374. A pet that comes around on an on_message listener and waits for someone to pet it (react with a standard wave emoji), and rewards with credits. Many attributes are configurable.
retrosign - A v3 port of Anismash's retrosign cog: https://github.com/Anismash/Ani-Cogs/tree/master/retrosign
rndstatus - A v3 port of Twentysix's rndstatus cog with a couple extra settings.

5
pupper/__init__.py Normal file
View File

@@ -0,0 +1,5 @@
from .pupper import Pupper
async def setup(bot):
bot.add_cog(Pupper(bot))

12
pupper/info.json Normal file
View File

@@ -0,0 +1,12 @@
{
"author": [
"aikaterna"
],
"description": "Pupper is a configurable pet that can hand out credits.",
"install_msg": "This cog and its countdowns are triggered by messages. If your bot doesn't see any messages posted anywhere in any server, the pet will never come around. This pet helps to keep chat active but it needs to be slightly active in the first place. If you see a message asking for pats (or whatever other greeting message you set), the first person to react with a wave emoji will get a random amount of credits. This cog also deletes the messages involved in the game, which can lead to channels showing an unread message when there really isn't anything there any more. Use `[p]help pet` to get started.",
"short": "Pet the dog.",
"tags": [
"pets"
],
"type": "COG"
}

225
pupper/pupper.py Normal file
View File

@@ -0,0 +1,225 @@
import asyncio
import datetime
import discord
import random
from redbot.core import commands, checks, Config, bank
from redbot.core.utils.chat_formatting import box, humanize_list
class Pupper(commands.Cog):
def __init__(self, bot):
self.bot = bot
self.config = Config.get_conf(self, 2767241393, force_registration=True)
self.pets = {}
default_guild = {
"borf_msg": "borf! (thank for pats h00man, have a doggocoin)",
"channel": [],
"cooldown": 3600,
"credits": [100, 500],
"hello_msg": "Hi! Can someone pet me?",
"last_pet": "2019-08-01 00:00:00.000001",
"toggle": False,
}
self.config.register_guild(**default_guild)
@commands.guild_only()
@checks.mod_or_permissions(administrator=True)
@commands.group()
async def pet(self, ctx):
"""Manage your pet."""
pass
@pet.command()
async def toggle(self, ctx):
"""Toggle pets on the server."""
toggle = await self.config.guild(ctx.guild).toggle()
msg = f"Pets active: {not toggle}.\n"
await self.config.guild(ctx.guild).toggle.set(not toggle)
await ctx.send(msg)
@pet.command()
async def cooldown(self, ctx, seconds: int = None):
"""Set the pet appearance cooldown in seconds.
300s/5 minute minimum. Default is 3600s/1 hour."""
if not seconds:
seconds = 3600
if seconds < 300:
seconds = 300
await self.config.guild(ctx.guild).cooldown.set(seconds)
await ctx.send(f"Pet appearance cooldown set to {seconds}.")
@pet.command()
async def credits(self, ctx, min_amt: int, max_amt: int):
"""Set the pet credits range on successful petting."""
if min_amt > max_amt:
return await ctx.send("Min must be less than max.")
if min_amt or max_amt < 1:
return await ctx.send("Min and max amounts must be greater than 1.")
await self.config.guild(ctx.guild).credits.set([min_amt, max_amt])
await ctx.send(f"Pet credit range set to {min_amt} - {max_amt}.")
@pet.command()
async def hello(self, ctx, *, message: str = None):
"""Set the pet greeting message."""
if not message:
hello = await self.config.guild(ctx.guild).hello_msg()
return await ctx.send(
f"Current greeting message: `{hello}`\nUse this command with the message you would like to set."
)
if len(message) > 1000:
return await ctx.send("That dog sure likes to talk a lot. Try a shorter message.")
await self.config.guild(ctx.guild).hello_msg.set(message)
await ctx.send(f"Pet hello message set to: `{message}`.")
@pet.command()
async def thanks(self, ctx, *, message: str = None):
"""Set the pet thanks message."""
if not message:
bye = await self.config.guild(ctx.guild).borf_msg()
return await ctx.send(
f"Current thanks message: `{bye}`\nUse this command with the message you would like to set."
)
if len(message) > 1000:
return await ctx.send("That dog sure likes to talk a lot. Try a shorter message.")
await self.config.guild(ctx.guild).borf_msg.set(message)
await ctx.send(f"Pet thanks message set to: `{message}`.")
@commands.guild_only()
@checks.mod_or_permissions(administrator=True)
@pet.group(invoke_without_command=True)
async def channel(self, ctx):
"""Channel management for pet appearance."""
await ctx.send_help()
channel_list = await self.config.guild(ctx.guild).channel()
channel_msg = "Petting Channels:\n"
if not channel_list:
channel_msg += "None."
for chan in channel_list:
channel_obj = self.bot.get_channel(chan)
channel_msg += f"{channel_obj.name}\n"
await ctx.send(box(channel_msg))
@channel.command()
async def add(self, ctx, channel: discord.TextChannel):
"""Add a text channel for pets."""
channel_list = await self.config.guild(ctx.guild).channel()
if channel.id not in channel_list:
channel_list.append(channel.id)
await self.config.guild(ctx.guild).channel.set(channel_list)
await ctx.send(f"{channel.mention} added to the valid petting channels.")
else:
await ctx.send(f"{channel.mention} is already in the list of petting channels.")
@channel.command()
async def addall(self, ctx):
"""Add all valid channels for the guild that the bot can speak in."""
bot_text_channels = [
c
for c in ctx.guild.text_channels
if c.permissions_for(ctx.guild.me).send_messages is True
]
channel_list = await self.config.guild(ctx.guild).channel()
channels_appended = []
channels_in_list = []
for text_channel in bot_text_channels:
if text_channel.id not in channel_list:
channel_list.append(text_channel.id)
channels_appended.append(text_channel.mention)
else:
channels_in_list.append(text_channel.mention)
pass
first_msg = ""
second_msg = ""
await self.config.guild(ctx.guild).channel.set(channel_list)
if len(channels_appended) > 0:
first_msg = (
f"{humanize_list(channels_appended)} added to the valid petting channels.\n"
)
if len(channels_in_list) > 0:
second_msg = (
f"{humanize_list(channels_in_list)}: already in the list of petting channels."
)
await ctx.send(f"{first_msg}{second_msg}")
@channel.command()
async def remove(self, ctx, channel: discord.TextChannel):
"""Remove a text channel from petting."""
channel_list = await self.config.guild(ctx.guild).channel()
if channel.id in channel_list:
channel_list.remove(channel.id)
else:
return await ctx.send(f"{channel.mention} not in the active channel list.")
await self.config.guild(ctx.guild).channel.set(channel_list)
await ctx.send(f"{channel.mention} removed from the list of petting channels.")
@channel.command()
async def removeall(self, ctx):
"""Remove all petting channels from the list."""
await self.config.guild(ctx.guild).channel.set([])
await ctx.send("All channels have been removed from the list of petting channels.")
def _pet_lock(self, guild_id, tf):
if tf:
self.pets[guild_id] = True
else:
self.pets[guild_id] = False
@commands.Cog.listener()
async def on_message(self, message):
if isinstance(message.channel, discord.abc.PrivateChannel):
return
if message.author.bot:
return
guild_data = await self.config.guild(message.guild).all()
if not guild_data["toggle"]:
return
if not guild_data["channel"]:
return
self.pets.setdefault(message.guild.id, False)
if self.pets[message.guild.id]:
return
last_time = datetime.datetime.strptime(str(guild_data["last_pet"]), "%Y-%m-%d %H:%M:%S.%f")
now = datetime.datetime.now(datetime.timezone.utc)
now = now.replace(tzinfo=None)
if (
int((now - last_time).total_seconds())
> await self.config.guild(message.guild).cooldown()
):
self._pet_lock(message.guild.id, True)
rando_channel = random.choice(guild_data["channel"])
await asyncio.sleep(random.randint(60, 480))
rando_channel_obj = self.bot.get_channel(rando_channel)
borf_msg = await rando_channel_obj.send(guild_data["hello_msg"])
pets = "👋"
pets_action = {"veryfastpats": "👋"}
def check(r, u):
return r.message.id == borf_msg.id and any(e in str(r.emoji) for e in pets)
try:
r, u = await self.bot.wait_for("reaction_add", check=check, timeout=300.0)
except asyncio.TimeoutError:
return await borf_msg.delete()
reacts = {v: k for k, v in pets_action.items()}
react = reacts[r.emoji]
if react == "veryfastpats":
await borf_msg.delete()
deposit = random.randint(guild_data["credits"][0], guild_data["credits"][1])
await bank.deposit_credits(u, deposit)
credits_name = await bank.get_currency_name(message.guild)
await rando_channel_obj.send(
content=f"{guild_data['borf_msg']} `(+{deposit} {credits_name})`",
delete_after=10,
)
else:
pass
self._pet_lock(message.guild.id, False)
await self.config.guild(message.guild).last_pet.set(str(now))