didier/cogs/poke.py

from data import constants
import datetime
from decorators import help
from discord.ext import commands
from enums.help_categories import Category
from functions import checks, timeFormatters
from functions.database import poke, stats


class Poke(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.group(name="Poke", usage="[@Persoon]", case_insensitive=True, invoke_without_command=True)
    @commands.check(checks.allowedChannels)
    @help.Category(category=Category.Games)
    async def poke(self, ctx, member=None):
        if not await self.pokeChecks(ctx):
            return

        member = ctx.message.mentions[0]
        await ctx.send("**{}** heeft **{}** getikt. **{}** is hem!".format(
            ctx.author.display_name, member.display_name, member.display_name))

        # Add into the database
        poke.update(ctx.author.id, member.id)
        stats.update(member.id, "poked", int(stats.getOrAddUser(member.id)[1]) + 1)

    @poke.command(name="Blacklist", aliases=["Bl"])
    async def blacklist(self, ctx):
        if poke.blacklisted(ctx.author.id):
            await ctx.send("Je hebt jezelf al geblacklisted, {}.".format(ctx.author.display_name))
            return
        if str(poke.get()[0]) == str(ctx.author.id):
            await ctx.send("Je kan jezelf niet blacklisten als je aan de beurt bent, {}.".format(
                ctx.author.display_name))
            return
        poke.blacklist(ctx.author.id)
        await ctx.send("**{}** heeft zichzelf geblacklisted en kan niet meer getikt worden.".format(
            ctx.author.display_name))

    @poke.command(aliases=["wl"], hidden=True)
    async def whitelist(self, ctx, *user):
        user = ctx.message.mentions[0].id
        if not poke.blacklisted(user):
            await ctx.send("Deze persoon is niet geblacklisted.")
            return

        poke.blacklist(user, False)
        await ctx.send("**{}** heeft {} gewhitelisted.".format(
            ctx.author.display_name, self.utilsCog.getDisplayName(ctx, user)))

    @poke.command(name="Current")
    async def current(self, ctx):
        p = poke.get()
        pokedTimeStamp = datetime.datetime.fromtimestamp(int(p[1]))
        timeString = timeFormatters.diffDayBasisString(pokedTimeStamp)

        await ctx.send("Het is al **{}** aan **{}**.".format(timeString, self.utilsCog.getDisplayName(ctx, p[0])))

    @poke.command(hidden=True)
    async def me(self, ctx):
        await ctx.send("Liever niet.")

    @poke.command(hidden=True)
    @commands.check(checks.isMe)
    async def reset(self, ctx):
        new = poke.reset()
        await ctx.send("Poke is gereset. <@!{}> is hem!".format(str(new)))

    @poke.command(aliases=["Lb", "Leaderboards"], hidden=True)
    async def leaderboard(self, ctx):
        await self.client.get_cog("Leaderboards").callLeaderboard("poke", ctx)

    async def pokeChecks(self, ctx):
        if len(ctx.message.mentions) == 0:
            await ctx.send("Dit is geen geldige persoon.")
            return False
        if len(ctx.message.mentions) > 1:
            await ctx.send("Je kan maar 1 persoon tegelijk tikken.")
            return False
        if ctx.message.mentions[0].id == ctx.author.id:
            await ctx.send("Je kan jezelf niet tikken, {}.".format(ctx.author.display_name))
            return False
        if str(ctx.message.mentions[0].id) == constants.didierId:
            await ctx.send("Je kan me niet tikken, {}.".format(ctx.author.display_name))
            return False
        if str(ctx.message.mentions[0].id) in constants.botIDs:
            await ctx.send("Je kan geen bots tikken, {}.".format(ctx.author.display_name))
            return False

        # Check database things
        p = poke.get()
        if str(p[0]) != str(ctx.author.id):
            await ctx.send("Het is niet jouw beurt, {}.".format(ctx.author.display_name))
            return False
        if str(ctx.message.mentions[0].id) == str(p[2]):
            await ctx.send("Je mag niet terugtikken, {}.".format(ctx.author.display_name))
            return False
        if poke.blacklisted(ctx.message.mentions[0].id):
            await ctx.send("Deze persoon heeft zichzelf geblacklisted en kan niet meer getikt worden.")
            return False

        return True


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