diff options
Diffstat (limited to 'bot/exts')
| -rw-r--r-- | bot/exts/events/advent_of_code/_cog.py | 20 | ||||
| -rw-r--r-- | bot/exts/fun/anagram.py | 1 | ||||
| -rw-r--r-- | bot/exts/fun/battleship.py | 1 | ||||
| -rw-r--r-- | bot/exts/fun/connect_four.py | 3 | ||||
| -rw-r--r-- | bot/exts/fun/game.py | 32 | ||||
| -rw-r--r-- | bot/exts/fun/tic_tac_toe.py | 3 | 
6 files changed, 45 insertions, 15 deletions
diff --git a/bot/exts/events/advent_of_code/_cog.py b/bot/exts/events/advent_of_code/_cog.py index c597fd0e..a9625153 100644 --- a/bot/exts/events/advent_of_code/_cog.py +++ b/bot/exts/events/advent_of_code/_cog.py @@ -213,9 +213,13 @@ class AdventOfCode(commands.Cog):      @whitelist_override(channels=AOC_WHITELIST)      async def join_leaderboard(self, ctx: commands.Context) -> None:          """DM the user the information for joining the Python Discord leaderboard.""" -        current_year = datetime.now().year -        if current_year != AocConfig.year: -            await ctx.send(f"The Python Discord leaderboard for {current_year} is not yet available!") +        current_date = datetime.now() +        if ( +            current_date.month not in (Month.NOVEMBER, Month.DECEMBER) and current_date.year != AocConfig.year or +            current_date.month != Month.JANUARY and current_date.year != AocConfig.year + 1 +        ): +            # Only allow joining the leaderboard in the run up to AOC and the January following. +            await ctx.send(f"The Python Discord leaderboard for {current_date.year} is not yet available!")              return          author = ctx.author @@ -254,7 +258,7 @@ class AdventOfCode(commands.Cog):          else:              await ctx.message.add_reaction(Emojis.envelope) -    @in_month(Month.NOVEMBER, Month.DECEMBER) +    @in_month(Month.NOVEMBER, Month.DECEMBER, Month.JANUARY)      @adventofcode_group.command(          name="link",          aliases=("connect",), @@ -306,7 +310,7 @@ class AdventOfCode(commands.Cog):                      " Please re-run the command with one specified."                  ) -    @in_month(Month.NOVEMBER, Month.DECEMBER) +    @in_month(Month.NOVEMBER, Month.DECEMBER, Month.JANUARY)      @adventofcode_group.command(          name="unlink",          aliases=("disconnect",), @@ -327,7 +331,7 @@ class AdventOfCode(commands.Cog):              log.info(f"Attempted to unlink {ctx.author} ({ctx.author.id}), but no link was found.")              await ctx.reply("You don't have an Advent of Code account linked.") -    @in_month(Month.DECEMBER) +    @in_month(Month.DECEMBER, Month.JANUARY)      @adventofcode_group.command(          name="dayandstar",          aliases=("daynstar", "daystar"), @@ -365,7 +369,7 @@ class AdventOfCode(commands.Cog):          await view.wait()          await message.edit(view=None) -    @in_month(Month.DECEMBER) +    @in_month(Month.DECEMBER, Month.JANUARY)      @adventofcode_group.command(          name="leaderboard",          aliases=("board", "lb"), @@ -410,7 +414,7 @@ class AdventOfCode(commands.Cog):          await ctx.send(content=f"{header}\n\n{table}", embed=info_embed)          return -    @in_month(Month.DECEMBER) +    @in_month(Month.DECEMBER, Month.JANUARY)      @adventofcode_group.command(          name="global",          aliases=("globalboard", "gb"), diff --git a/bot/exts/fun/anagram.py b/bot/exts/fun/anagram.py index 9aee5f18..79280fa9 100644 --- a/bot/exts/fun/anagram.py +++ b/bot/exts/fun/anagram.py @@ -49,7 +49,6 @@ class Anagram(commands.Cog):          self.games: dict[int, AnagramGame] = {}      @commands.command(name="anagram", aliases=("anag", "gram", "ag")) -    @commands.guild_only()      async def anagram_command(self, ctx: commands.Context) -> None:          """          Given shuffled letters, rearrange them into anagrams. diff --git a/bot/exts/fun/battleship.py b/bot/exts/fun/battleship.py index f4351954..beff196f 100644 --- a/bot/exts/fun/battleship.py +++ b/bot/exts/fun/battleship.py @@ -369,7 +369,6 @@ class Battleship(commands.Cog):          return any(player in (game.p1.user, game.p2.user) for game in self.games)      @commands.group(invoke_without_command=True) -    @commands.guild_only()      async def battleship(self, ctx: commands.Context) -> None:          """          Play a game of Battleship with someone else! diff --git a/bot/exts/fun/connect_four.py b/bot/exts/fun/connect_four.py index 647bb2b7..f53695d5 100644 --- a/bot/exts/fun/connect_four.py +++ b/bot/exts/fun/connect_four.py @@ -6,7 +6,6 @@ from typing import Optional, Union  import discord  import emojis  from discord.ext import commands -from discord.ext.commands import guild_only  from bot.bot import Bot  from bot.constants import Emojis @@ -361,7 +360,6 @@ class ConnectFour(commands.Cog):                  self.games.remove(game)              raise -    @guild_only()      @commands.group(          invoke_without_command=True,          aliases=("4inarow", "connect4", "connectfour", "c4"), @@ -426,7 +424,6 @@ class ConnectFour(commands.Cog):          await self._play_game(ctx, user, board_size, str(emoji1), str(emoji2)) -    @guild_only()      @connect_four.command(aliases=("bot", "computer", "cpu"))      async def ai(          self, diff --git a/bot/exts/fun/game.py b/bot/exts/fun/game.py index f9c150e6..5f56bef7 100644 --- a/bot/exts/fun/game.py +++ b/bot/exts/fun/game.py @@ -118,6 +118,7 @@ class GameStatus(IntEnum):      Offline = 5      Cancelled = 6      Rumored = 7 +    Delisted = 8  class AgeRatingCategories(IntEnum): @@ -125,6 +126,11 @@ class AgeRatingCategories(IntEnum):      ESRB = 1      PEGI = 2 +    CERO = 3 +    USK = 4 +    GRAC = 5 +    CLASS_IND = 6 +    ACB = 7  class AgeRatings(IntEnum): @@ -142,6 +148,32 @@ class AgeRatings(IntEnum):      T = 10      M = 11      AO = 12 +    CERO_A = 13 +    CERO_B = 14 +    CERO_C = 15 +    CERO_D = 16 +    CERO_Z = 17 +    USK_0 = 18 +    USK_6 = 19 +    USK_12 = 20 +    USK_18 = 21 +    GRAC_ALL = 22 +    GRAC_Twelve = 23 +    GRAC_Fifteen = 24 +    GRAC_Eighteen = 25 +    GRAC_TESTING = 26 +    CLASS_IND_L = 27 +    CLASS_IND_Ten = 28 +    CLASS_IND_Twelve = 29 +    CLASS_IND_Fourteen = 30 +    CLASS_IND_Sixteen = 31 +    CLASS_IND_Eighteen = 32 +    ACB_G = 33 +    ACB_PG = 34 +    ACB_M = 35 +    ACB_MA15 = 36 +    ACB_R18 = 37 +    ACB_RC = 38  class Games(Cog): diff --git a/bot/exts/fun/tic_tac_toe.py b/bot/exts/fun/tic_tac_toe.py index 946b6f7b..5dd38a81 100644 --- a/bot/exts/fun/tic_tac_toe.py +++ b/bot/exts/fun/tic_tac_toe.py @@ -3,7 +3,7 @@ import random  from typing import Callable, Optional, Union  import discord -from discord.ext.commands import Cog, Context, check, group, guild_only +from discord.ext.commands import Cog, Context, check, group  from bot.bot import Bot  from bot.constants import Emojis @@ -253,7 +253,6 @@ class TicTacToe(Cog):      def __init__(self):          self.games: list[Game] = [] -    @guild_only()      @is_channel_free()      @is_requester_free()      @group(name="tictactoe", aliases=("ttt", "tic"), invoke_without_command=True)  |