Beispiel #1
0
 async def pushboard_info(self, ctx):
     """Provides info on guild's pushboard"""
     guild_config = await self.bot.get_guild_config(ctx.guild.id)
     table = CLYTable()
     table.title = guild_config.pushboard_title or "PushBoard"
     if guild_config.pushboard_render == 2:
         table.add_rows([[0, 4721, 56, "Rowcoy"],
                         [1, 4709, 42, "Stitch"],
                         [2, 4658, 37, "t3pps"]])
         render = table.render_option_2()
     else:
         table.add_rows([[0, 4721, "Rowcoy (Awesome Clan)"],
                         [1, 4709, "Stitch (Lilo's Clan)"],
                         [2, 4658, "t3pps (Other Clan)"]])
         render = table.render_option_1()
     fmt = f"**PushBoard Example Format:**\n\n{render}\n" \
           f"**Icon:** Please see the icon displayed above.\n"
     channel = guild_config.pushboard
     data = []
     if channel is None:
         data.append("**Channel:** #deleted-channel")
     else:
         data.append(f"**Channel:** {channel.mention}")
     sql = "SELECT clan_name, clan_tag FROM clans WHERE guild_id = $1"
     fetch = await ctx.db.fetch(sql, ctx.guild.id)
     data.append(f"**Clans:** {', '.join(f'{n[0]} ({n[1]})' for n in fetch)}")
     fmt += "\n".join(data)
     embed = discord.Embed(color=self.bot.color,
                           description=fmt)
     embed.set_author(name="PushBoard Info",
                      icon_url=guild_config.icon_url or "https://cdn.discordapp.com/emojis/"
                                                        "592028799768592405.png?v=1")
     await ctx.send(embed=embed)
Beispiel #2
0
 def __init__(self, ctx, data, title=None, page_count=1, rows_per_table=20, description=''):
     super().__init__(ctx, entries=[i for i in range(page_count)], per_page=1)
     self.table = CLYTable()
     self.data = [(i, v) for (i, v) in enumerate(data)]
     self.entries = [None for _ in range(page_count)]
     self.rows_per_table = rows_per_table
     self.title = title
     self.message = None
     self.ctx = ctx
     self.description = description
     if getattr(ctx, 'config', None):
         try:
             self.icon_url = ctx.config.icon_url or ctx.guild.icon_url
             self.title = ctx.config.title or title
         except AttributeError:
             self.icon_url = ctx.guild.icon_url
     else:
         self.icon_url = ctx.guild.icon_url
Beispiel #3
0
    async def info_trophyboard(self, ctx):
        """Gives you info about guild's trophyboard.
        """
        table = CLYTable()
        if ctx.config.render == 1:
            table.add_rows([[0, 4320, 955, 'Member Name'], [1, 4500, 870, 'Another Member'],
                            [2, 3900, -600, 'Yet Another'], [3, 1500, -1000, 'Worst Pusher']
                            ])

            table.title = ctx.config.title or 'TrophyBoard'
            render = table.trophyboard_1()
        else:
            table.add_rows([[0, 2000, 'Member'], [1, 1500, 'Nearly #1'],
                            [2, 1490, 'Another Member'], [3, -600, 'Winner']
                            ])

            table.title = ctx.config.title or 'TrophyBoard'
            render = table.trophyboard_2()

        fmt = f'**Trophyboard Example Format:**\n\n{render}\n**Icon:** ' \
            f'Please see the icon displayed above.\n'

        channel = ctx.config.channel
        data = []

        if channel is None:
            data.append('**Channel:** #deleted-channel')
        else:
            data.append(f'**Channel:** {channel.mention}')

        query = "SELECT clan_name, clan_tag FROM clans WHERE guild_id = $1;"
        fetch = await ctx.db.fetch(query, ctx.guild.id)

        data.append(f"**Clans:** {', '.join(f'{n[0]} ({n[1]})' for n in fetch)}")

        fmt += '\n'.join(data)

        e = discord.Embed(colour=self.bot.colour,
                          description=fmt if len(fmt) < 2048 else f'{fmt[:2040]}...')
        e.set_author(name='TrophyBoard Info',
                     icon_url=ctx.config.icon_url or 'https://cdn.discordapp.com/emojis/592028799768592405.png?v=1')

        await ctx.send(embed=e)
Beispiel #4
0
    async def update_global_board(self):
        query = """SELECT player_tag, donations
                   FROM players 
                   WHERE season_id=$1
                   ORDER BY donations DESC NULLS LAST
                   LIMIT 100;
                """
        fetch_top_players = await self.bot.pool.fetch(
            query, await self.bot.seasonconfig.get_season_id())

        players = await self.bot.coc.get_players(
            (n[0] for n in fetch_top_players)).flatten()

        top_players = {
            n.tag: n
            for n in players
            if n.tag in set(x['player_tag'] for x in fetch_top_players)
        }

        messages = await self.get_board_messages(663683345108172830,
                                                 number_of_msg=5)
        if not messages:
            return

        for i, v in enumerate(messages):
            player_data = fetch_top_players[i * 20:(i + 1) * 20]
            table = CLYTable()

            for x, y in enumerate(player_data):
                index = i * 20 + x
                table.add_row(
                    [index, y[1],
                     top_players.get(y['player_tag'], mock).name])

            fmt = table.donationboard_2()

            e = discord.Embed(colour=self.bot.colour,
                              description=fmt,
                              timestamp=datetime.utcnow())
            e.set_author(name="Global Donationboard",
                         icon_url=self.bot.user.avatar_url)
            e.set_footer(text='Last Updated')
            await v.edit(embed=e, content=None)
Beispiel #5
0
    async def info_donationboard(self, ctx):
        """Gives you info about guild's donationboard.
        """
        table = CLYTable()
        if ctx.config.render == 2:
            table.add_rows([[0, 6532, 'Member (Awesome Clan)'], [1, 4453, 'Nearly #1 (Bad Clan)'],
                            [2, 5589, 'Another Member (Awesome Clan)'], [3, 0, 'Winner (Bad Clan)']
                            ])
            table.title = ctx.config.title or 'DonationBoard'
            render = table.donationboard_2()
        else:
            table.add_rows([[0, 9913, 12354, 'Member Name'], [1, 524, 123, 'Another Member'],
                            [2, 321, 444, 'Yet Another'], [3, 0, 2, 'The Worst Donator']
                            ])
            table.title = ctx.config.title or 'DonationBoard'
            render = table.donationboard_1()

        fmt = f'**DonationBoard Example Format:**\n\n{render}\n**Icon:** ' \
            f'Please see the icon displayed above.\n'

        channel = ctx.config.channel
        data = []

        if channel is None:
            data.append('**Channel:** #deleted-channel')
        else:
            data.append(f'**Channel:** {channel.mention}')

        query = "SELECT clan_name, clan_tag FROM clans WHERE guild_id = $1;"
        fetch = await ctx.db.fetch(query, ctx.guild.id)

        data.append(f"**Clans:** {', '.join(f'{n[0]} ({n[1]})' for n in fetch)}")

        fmt += '\n'.join(data)

        e = discord.Embed(colour=self.bot.colour,
                          description=fmt if len(fmt) < 2048 else f'{fmt[:2040]}...')

        e.set_author(name='DonationBoard Info',
                     icon_url=ctx.config.icon_url or 'https://cdn.discordapp.com/emojis/592028799768592405.png?v=1')

        await ctx.send(embed=e)
Beispiel #6
0
 async def update_pushboard(self, guild_id):
     guild_config = await self.get_guild_config(guild_id)
     if not guild_config.updates_toggle:
         return
     if not guild_config.pushboard:
         return
     sql = "SELECT DISTINCT clan_tag FROM clans WHERE guild_id = $1"
     fetch = await self.bot.pool.fetch(sql, guild_id)
     clans = await self.bot.coc.get_clans((n[0] for n in fetch)).flatten()
     players = []
     for n in clans:
         players.extend(p for p in n.itermembers)
     sql = ("SELECT player_tag, current_trophies, current_attack_wins - starting_attack_wins AS attacks "
            "FROM players "
            "WHERE player_tag = ANY($1::TEXT[]) "
            "ORDER BY current_trophies "
            "LIMIT 100")
     fetch = await self.bot.pool.fetch(sql, [n.tag for n in players])
     db_players = [DatabasePlayer(bot=self.bot, record=n) for n in fetch]
     players = {n.tag: n for n in players if n.tag in set(x.player_tag for x in db_players)}
     message_count = math.ceil(len(db_players) / 20)
     messages = await self.get_updates_messages(guild_id, number_of_msg=message_count)
     if not messages:
         return
     for i, v in enumerate(messages):
         player_data = db_players[i * 20: (i + 1) * 20]
         table = CLYTable()
         for x, y in enumerate(player_data):
             index = i * 20 + x
             if guild_config.pushboard_render == 2:
                 table.add_row([index,
                                y.current_trophies,
                                players.get(y.player_tag, MockPlayer()).name])
             else:
                 table.add_row([index,
                                y.current_trophies,
                                y.attacks,
                                players.get(y.player_tag, MockPlayer()).name])
         fmt = table.render_option_2() if \
             guild_config.pushboard_render == 2 else table.render_option_1()
         e = discord.Embed(color=self.bot.color,
                           description=fmt,
                           timestamp=datetime.utcnow())
         e.set_author(name=guild_config.pushboard_title or "Trophy Push Leaderboard",
                      icon_url=guild_config.icon_url or "https://cdn.discordapp.com/emojis/"
                                                        "592028799768592405.png?v=1")
         e.set_footer(text="Last Updated")
         await v.edit(embed=e, content=None)
Beispiel #7
0
    async def pushboard_edit(self, ctx):
        """Edit the format of the guild's pushboard"""
        table = CLYTable()
        table.add_rows([[0, 4721, 56, "Rowcoy"],
                       [1, 4709, 42, "Stitch"],
                       [2, 4658, 37, "t3pps"]])
        table.title = "**Option 1 Example**"
        option_1_render = f"**Option 1 Example**\n{table.render_option_1()}"
        table.clear_rows()
        table.add_rows([[0, 4721, "Rowcoy (Awesome Clan)"],
                        [1, 4709, "Stitch (Lilo's Clan)"],
                        [2, 4658, "t3pps (Other Clan)"]])
        option_2_render = f"**Option 2 Example**\n{table.render_option_2()}"
        embed = discord.Embed(color=self.bot.color)
        fmt = (f"{option_1_render}\n\n{option_2_render}\n\n\n"
               f"These are the 2 available options.\n"
               f"Please click the reaction of the format you \n"
               f"wish to display on your pushboard.")
        embed.description = fmt
        msg = await ctx.send(embed=embed)
        sql = "UPDATE guilds SET pushboard_render = $1 WHERE guild_id = $2"
        reactions = ["1\N{combining enclosing keycap}", "2\N{combining enclosing keycap}"]
        for r in reactions:
            await msg.add_reaction(r)

        def check(r, u):
            return str(r) in reactions and u.id == ctx.author.id and r.message.id == msg.id

        try:
            r,u = await self.bot.wait_for("reaction_add", check=check, timeout=60.0)
        except asyncio.TimeoutError:
            await ctx.db.execute(sql, 1, ctx.guild.id)
            return await ctx.send("I got bored waiting and selected Option 1 for you.")
        await ctx.db.execute(sql, reactions.index(str(r)) + 1, ctx.guild.id)
        await ctx.confirm()
        await ctx.send("All done. Thank you!")
        self.get_guild_config.invalidate(self, ctx.guild.id)
Beispiel #8
0
    async def remove_event(self, ctx, *, event_name: str = None):
        """Removes a currently running event.

        **Parameters**
        :key: The event name to remove.

        **Format**
        :information_source: `+remove event EVENT_NAME`

        **Example**
        :white_check_mark: `+remove event my special event`

        **Required Permissions**
        :warning: Manage Server
        """
        if event_name:
            # Event name provided
            query = """DELETE FROM events
                       WHERE guild_id = $1 
                       AND event_name = $2
                       RETURNING id;
                    """
            fetch = await self.bot.pool.fetchrow(query, ctx.guild.id,
                                                 event_name)
            if fetch:
                return await ctx.send(f"{event_name} has been removed.")

        # No event name provided or I didn't understand the name I was given
        query = """SELECT id, event_name, start 
                   FROM events
                   WHERE guild_id = $1 
                   ORDER BY start"""
        fetch = await self.bot.pool.fetch(query, ctx.guild.id)
        if len(fetch) == 0 or not fetch:
            return await ctx.send(
                "I have no events to remove. You should create one... then remove it."
            )
        elif len(fetch) == 1:
            query = "DELETE FROM events WHERE id = $1"
            await ctx.db.execute(query, fetch[0]['id'])
            return await ctx.send(f"{fetch[0]['event_name']} has been removed."
                                  )

        table = CLYTable()
        fmt = f"Events on {ctx.guild}:\n\n"
        reactions = []
        counter = 0
        for event in fetch:
            days_until = event['start'].date() - datetime.datetime.utcnow(
            ).date()
            table.add_row([counter, days_until.days, event['event_name']])
            counter += 1
            reactions.append(f"{counter}\N{combining enclosing keycap}")
        render = table.events_list()
        fmt += f'{render}\n\nPlease select the reaction that corresponds with the event you would ' \
               f'like to remove.'
        e = discord.Embed(colour=self.bot.colour, description=fmt)
        msg = await ctx.send(embed=e)
        for r in reactions:
            await msg.add_reaction(r)

        def check(r, u):
            return str(
                r
            ) in reactions and u.id == ctx.author.id and r.message.id == msg.id

        try:
            r, u = await self.bot.wait_for('reaction_add',
                                           check=check,
                                           timeout=60.0)
        except asyncio.TimeoutError:
            await msg.clear_reactions()
            return await ctx.send(
                "We'll just hang on to all the events we have for now.")

        index = reactions.index(str(r))
        query = "DELETE FROM events WHERE id = $1"
        await ctx.db.execute(query, fetch[index]['id'])
        await msg.delete()
        ctx.bot.utils.event_config.invalidate(ctx.bot.utils, ctx.guild.id)
        self.bot.dispatch('event_register')
        return await ctx.send(f"{fetch[index]['event_name']} has been removed."
                              )
Beispiel #9
0
    async def get_board_fmt(self, guild_id, season_id, board_type):
        board_config = await self.bot.utils.get_board_configs(
            guild_id, board_type)
        if not board_config:
            board_config = SlimDummyBoardConfig(
                board_type, 2, f"{board_type.capitalize()}Board", None,
                'donations' if board_type == "donation" else "trophies")
        else:
            board_config = board_config[0]

        clans = await self.bot.get_clans(guild_id)

        players = []
        for n in clans:
            players.extend(p for p in n.itermembers)

        top_players = await self.bot.donationboard.get_top_players(
            players,
            board_type,
            board_config.sort_by,
            False,
            season_id=season_id)

        if not top_players:
            e = discord.Embed(colour=self.bot.colour, title='No Data Found.')
            return [e]

        players = {
            n.tag: n
            for n in players
            if n.tag in set(x['player_tag'] for x in top_players)
        }

        message_count = math.ceil(len(top_players) / 20)

        embeds = []
        for i in range(message_count):
            player_data = top_players[i * 20:(i + 1) * 20]
            table = CLYTable()

            for x, y in enumerate(player_data):
                index = i * 20 + x
                if board_config.render == 2:
                    table.add_row(
                        [index, y[1],
                         players.get(y['player_tag'], mock).name])
                else:
                    table.add_row([
                        index, y[1], y[2],
                        players.get(y['player_tag'], mock).name
                    ])

            render = get_render_type(board_config, table)
            fmt = render()

            e = discord.Embed(colour=self.bot.donationboard.get_colour(
                board_type, False),
                              description=fmt,
                              timestamp=datetime.utcnow())
            e.set_author(name=board_config.title,
                         icon_url=board_config.icon_url
                         or 'https://cdn.discordapp.com/'
                         'emojis/592028799768592405.png?v=1')
            e.set_footer(
                text=
                f'Historical {board_type.capitalize()}Board; Season {season_id} - Page {i+1}/{message_count}'
            )
            embeds.append(e)

        return embeds
Beispiel #10
0
class TablePaginator(Pages):
    def __init__(self, ctx, data, title=None, page_count=1, rows_per_table=20, description=''):
        super().__init__(ctx, entries=[i for i in range(page_count)], per_page=1)
        self.table = CLYTable()
        self.data = [(i, v) for (i, v) in enumerate(data)]
        self.entries = [None for _ in range(page_count)]
        self.rows_per_table = rows_per_table
        self.title = title
        self.message = None
        self.ctx = ctx
        self.description = description
        if getattr(ctx, 'config', None):
            try:
                self.icon_url = ctx.config.icon_url or ctx.guild.icon_url
                self.title = ctx.config.title or title
            except AttributeError:
                self.icon_url = ctx.guild.icon_url
        else:
            self.icon_url = ctx.guild.icon_url

    async def get_page(self, page):
        try:
            entry = self.entries[page - 1]
            if entry:
                return entry
        except IndexError:
            pass

        if not self.message:
            self.message = await self.channel.send('Loading...')
        else:
            await self.message.edit(content='Loading...', embed=None)

        entry = await self.prepare_entry(page)
        self.entries[page - 1] = entry
        return self.entries[page - 1]

    async def prepare_entry(self, page):
        self.table.clear_rows()
        base = (page - 1) * self.rows_per_table
        data = self.data[base:base + self.rows_per_table]
        for n in data:
            self.table.add_row(n)

        render = get_render_type(self.ctx.config, self.table)
        return render()

    async def get_embed(self, entries, page, *, first=False):
        if self.maximum_pages > 1:
            if self.show_entry_count:
                text = f'Page {page}/{self.maximum_pages} ({len(self.entries)} entries)'
            else:
                text = f'Page {page}/{self.maximum_pages}'

            self.embed.set_footer(text=text)

        self.embed.description = self.description + entries

        self.embed.set_author(
            name=textwrap.shorten(self.title, width=240, placeholder='...'),
            icon_url=self.icon_url
        )

        return self.embed

    async def show_page(self, page, *, first=False):
        self.current_page = page
        entries = await self.get_page(page)
        embed = await self.get_embed(entries, page, first=first)

        if not self.message:
            self.message = await self.ctx.send("Loading...")

        if not self.paginating:
            return await self.message.edit(content=None, embed=embed)

        await self.message.edit(content=None, embed=embed)

        if not first:
            return

        for (reaction, _) in self.reaction_emojis:
            if self.maximum_pages == 2 and reaction in ('\u23ed', '\u23ee'):
                # no |<< or >>| buttons if we only have two pages
                # we can't forbid it if someone ends up using it but remove
                # it from the default set
                continue

            await self.message.add_reaction(reaction)
Beispiel #11
0
    async def update_board(self, channel_id):
        config = await self.bot.utils.board_config(channel_id)

        if not config:
            return
        if not config.toggle:
            return
        if not config.channel:
            return

        if config.in_event:
            query = """SELECT DISTINCT clan_tag FROM clans WHERE channel_id=$1 AND in_event=$2"""
            fetch = await self.bot.pool.fetch(query, channel_id,
                                              config.in_event)
        else:
            query = "SELECT DISTINCT clan_tag FROM clans WHERE channel_id=$1"
            fetch = await self.bot.pool.fetch(query, channel_id)

        clans = await self.bot.coc.get_clans((n[0] for n in fetch)).flatten()

        players = []
        for n in clans:
            players.extend(p for p in n.itermembers)

        try:
            top_players = await self.get_top_players(players, config.type,
                                                     config.sort_by,
                                                     config.in_event)
        except:
            log.error(
                f"{clans} channelid: {channel_id}, guildid: {config.guild_id},"
                f" sort: {config.sort_by}, event: {config.in_event}, type: {config.type}"
            )
            return
        players = {
            n.tag: n
            for n in players
            if n.tag in set(x['player_tag'] for x in top_players)
        }

        message_count = math.ceil(len(top_players) / 20)

        messages = await self.get_board_messages(channel_id,
                                                 number_of_msg=message_count)
        if not messages:
            return

        for i, v in enumerate(messages):
            player_data = top_players[i * 20:(i + 1) * 20]
            table = CLYTable()

            for x, y in enumerate(player_data):
                index = i * 20 + x
                if config.render == 2:
                    table.add_row(
                        [index, y[1],
                         players.get(y['player_tag'], mock).name])
                else:
                    table.add_row([
                        index, y[1], y[2],
                        players.get(y['player_tag'], mock).name
                    ])

            render = get_render_type(config, table)
            fmt = render()

            e = discord.Embed(colour=self.get_colour(config.type,
                                                     config.in_event),
                              description=fmt,
                              timestamp=datetime.utcnow())
            e.set_author(name=f'Event in Progress!'
                         if config.in_event else config.title,
                         icon_url=config.icon_url
                         or 'https://cdn.discordapp.com/'
                         'emojis/592028799768592405.png?v=1')
            e.set_footer(text='Last Updated')
            await v.edit(embed=e, content=None)
Beispiel #12
0
    async def edit_event(self, ctx, *, event_name: str = None):
        """Edit a variety of settings for the current event.

        **Parameters**
        :key: Event name

        **Format**
        :information_source: `+edit event EVENT_NAME`

        **Example**
        :white_check_mark: `+edit event Donation Bot Event`

        **Required Permissions**
        :warning: Manage Server
        """
        if event_name:
            query = """SELECT id FROM events 
                       WHERE guild_id = $1 
                       AND event_name = $2"""
            fetch = await self.bot.pool.fetchrow(query, ctx.guild.id, event_name)
            if fetch:
                event_id = fetch['id']
            else:
                # ideally this would just display a list of events and let the user pick, but I
                # couldn't figure out the proper sequence of if event_name/if event_id
                return await ctx.send("There is no event on this server with that name. Try `+edit event` "
                                      "to pick from a list of events on this server.")
        else:
            # No event name provided or I didn't understand the name I was given
            query = """SELECT id, event_name, start 
                               FROM events
                               WHERE guild_id = $1 
                               ORDER BY start"""
            fetch = await self.bot.pool.fetch(query, ctx.guild.id)
            if len(fetch) == 0 or not fetch:
                return await ctx.send("There are no events currently set up on this server. "
                                      "Try `+add event`")
            elif len(fetch) == 1:
                event_id = fetch[0]['id']
            else:
                table = CLYTable()
                fmt = f"Events on {ctx.guild}:\n\n"
                reactions = []
                counter = 0
                for event in fetch:
                    days_until = event['start'].date() - datetime.datetime.utcnow().date()
                    table.add_row([counter, days_until.days, event['event_name']])
                    counter += 1
                    reactions.append(f"{counter}\N{combining enclosing keycap}")
                render = table.events_list()
                fmt += f'{render}\n\nPlease select the reaction that corresponds with the event you would ' \
                       f'like to remove.'
                e = discord.Embed(colour=self.bot.colour,
                                  description=fmt)
                msg = await ctx.send(embed=e)
                for r in reactions:
                    await msg.add_reaction(r)

                def check(r, u):
                    return str(r) in reactions and u.id == ctx.author.id and r.message.id == msg.id

                try:
                    r, u = await self.bot.wait_for('reaction_add', check=check, timeout=60.0)
                except asyncio.TimeoutError:
                    await msg.clear_reactions()
                    return await ctx.send("I feel like I'm being ignored. MAybe try again later?")

                index = reactions.index(str(r))
                event_id = fetch[index]['id']

            # Now that we have the event_id, let's edit things
            query = """SELECT event_name, start, finish 
                       FROM events
                       WHERE id = $1"""
            event = await self.bot.pool.fetchrow(query, event_id)

            def check_author(m):
                return m.author == ctx.author

            answer = await ctx.prompt(f"Event Name: **{event['event_name']}**\n"
                                      f"Would you like to edit the event name?")
            if answer:
                try:
                    await ctx.send('Please enter the new name for this event.')
                    response = await ctx.bot.wait_for('message', check=check_author, timeout=60.0)
                    new_event_name = response.content
                except asyncio.TimeoutError:
                    new_event_name = event['event_name']
            else:
                new_event_name = event['event_name']
            answer = await ctx.prompt(f"Start Date: **{event['start'].date()}\n"
                                      f"Would you like to edit the date?")
            if answer:
                try:
                    await ctx.send('Please enter the new start date.  (YYYY-MM-DD)')
                    response = await ctx.bot.wait_for('message', check=check_author, timeout=60.0)
                    new_start_date = await DateConverter().convert(ctx, response.clean_content)
                except (ValueError, commands.BadArgument):
                    await ctx.send('Date must be in the YYYY-MM-DD format. I\'m going to keep '
                                   'the current start date and you can change it later if you like.')
                    new_start_date = event['start'].date()
                except asyncio.TimeoutError:
                    await ctx.send('Seems as though you don\'t really know the answer. I\'m just going '
                                   'to keep the date I have for now.')
                    new_start_date = event['start'].date()
            else:
                new_start_date = event['start'].date()
            answer = await ctx.prompt(f"Start Time: **{event['start'].time()}\n"
                                      f"Would you like to edit the time?")
            if answer:
                try:
                    await ctx.send('Please enter the new start time. (Please provide HH:MM in UTC)')
                    response = await ctx.bot.wait_for('message', check=check_author, timeout=60.0)
                    hour, minute = map(int, response.content.split(':'))
                    if hour < 13:
                        try:
                            await ctx.send('And is that AM or PM?')
                            response = await ctx.bot.wait_for('message', check=check_author, timeout=60.0)
                            if response.content.lower() == 'pm':
                                hour += 12
                        except asyncio.TimeoutError:
                            if hour < 6:
                                await ctx.send('Well I\'ll just go with PM then.')
                                hour += 12
                            else:
                                await ctx.send('I\'m going to assume you want AM.')
                    new_start_time = datetime.time(hour, minute)
                except asyncio.TimeoutError:
                    await ctx.send('Time\'s up my friend. Start time will remain the same!')
                    new_start_time = event['start'].time()
            else:
                new_start_time = event['start'].time()
            answer = await ctx.prompt(f"End Date: **{event['finish'].date()}\n"
                                      f"Would you like to edit the date?")
            if answer:
                try:
                    await ctx.send('Please enter the new end date.  (YYYY-MM-DD)')
                    response = await ctx.bot.wait_for('message', check=check_author, timeout=60.0)
                    new_end_date = await DateConverter().convert(ctx, response.clean_content)
                except (ValueError, commands.BadArgument):
                    await ctx.send('Date must be in the YYYY-MM-DD format. I\'m going to keep '
                                   'the current end date and you can change it later if you like.')
                    new_end_date = event['finish'].date()
                except asyncio.TimeoutError:
                    await ctx.send('Seems as though you don\'t really know the answer. I\'m just going '
                                   'to keep the date I have for now.')
                    new_end_date = event['finish'].date()
            else:
                new_end_date = event['finish'].date()
            answer = await ctx.prompt(f"End Time: **{event['finish'].time()}\n"
                                      f"Would you like to edit the time?")
            if answer:
                try:
                    await ctx.send('Please enter the new end time. (Please provide HH:MM in UTC)')
                    response = await ctx.bot.wait_for('message', check=check_author, timeout=60.0)
                    hour, minute = map(int, response.content.split(':'))
                    if hour < 13:
                        try:
                            await ctx.send('And is that AM or PM?')
                            response = await ctx.bot.wait_for('message', check=check_author, timeout=60.0)
                            if response.content.lower() == 'pm':
                                hour += 12
                        except asyncio.TimeoutError:
                            if hour < 6:
                                await ctx.send('Well I\'ll just go with PM then.')
                                hour += 12
                            else:
                                await ctx.send('I\'m going to assume you want AM.')
                    new_end_time = datetime.time(hour, minute)
                except asyncio.TimeoutError:
                    await ctx.send('Time\'s up my friend. Start time will remain the same!')
                    new_end_time = event['finish'].time()
            else:
                new_end_time = event['finish'].time()

            # Assemble answers and update db
            new_start = datetime.datetime.combine(new_start_date, new_start_time)
            new_finish = datetime.datetime.combine(new_end_date, new_end_time)
            query = """UPDATE events 
                       SET event_name = $1, start = $2, finish = $3 
                       WHERE id = $4"""
            await ctx.db.execute(query, new_event_name, new_start, new_finish, event_id)

            fmt = (f'**Event Info:**\n\n{new_event_name}\n{new_start.strftime("%d %b %Y %H:%M")}\n'
                   f'{new_finish.strftime("%d %b %Y %H:%M")}')
            e = discord.Embed(colour=discord.Colour.green(),
                              description=fmt)
            await ctx.send(embed=e)
            self.bot.dispatch('event_register')
Beispiel #13
0
    async def edit_trophyboard_format(self, ctx):
        """Edit the format of the guild's trophyboard.

        The bot will provide 2 options and you must select 1 via reactions.

        **Format**
        :information_source: `+edit trophyboard format`

        **Example**
        :white_check_mark: `+edit trophyboard format`

        **Required Permissions**
        :warning: Manage Server
        """
        table = CLYTable()
        table.add_rows([[0, 4320, 955, 'Member Name'],
                        [1, 4500, 870, 'Another Member'],
                        [2, 3900, -600, 'Yet Another'],
                        [3, 1500, -1000, 'Worst Pusher']])
        table.title = '**Option 1 Example**'
        option_1_render = f'**Option 1 Example**\n{table.trophyboard_1()}'

        table.clear_rows()
        table.add_rows([[0, 2000, 'Member'], [1, 1500, 'Nearly #1'],
                        [2, 1490, 'Another Member'], [3, -600, 'Winner']])

        option_2_render = f'**Option 2 Example**\n{table.trophyboard_2()}'

        embed = discord.Embed(colour=self.bot.colour)
        fmt = f'{option_1_render}\n\n\n{option_2_render}\n\n\n' \
            f'These are the 2 available default options.\n' \
            f'Please hit the reaction of the format you \nwish to display on the trophyboard.'
        embed.description = fmt
        msg = await ctx.send(embed=embed)

        query = "UPDATE boards SET render=$1 WHERE channel_id=$2"

        reactions = [
            '1\N{combining enclosing keycap}',
            '2\N{combining enclosing keycap}'
        ]
        for r in reactions:
            await msg.add_reaction(r)

        def check(r, u):
            return str(
                r
            ) in reactions and u.id == ctx.author.id and r.message.id == msg.id

        try:
            r, u = await self.bot.wait_for('reaction_add',
                                           check=check,
                                           timeout=60.0)
        except asyncio.TimeoutError:
            await ctx.db.execute(query, 1, ctx.config.channel_id)
            return await ctx.send('You took too long. Option 1 was chosen.')

        await ctx.db.execute(query,
                             reactions.index(str(r)) + 1,
                             ctx.config.channel_id)
        await ctx.confirm()