diff options
Diffstat (limited to 'bot/converters.py')
| -rw-r--r-- | bot/converters.py | 42 | 
1 files changed, 20 insertions, 22 deletions
| diff --git a/bot/converters.py b/bot/converters.py index 0d9a519df..67525cd4d 100644 --- a/bot/converters.py +++ b/bot/converters.py @@ -16,6 +16,7 @@ from discord.utils import DISCORD_EPOCH, snowflake_time  from bot.api import ResponseCodeError  from bot.constants import URLs  from bot.utils.regex import INVITE_RE +from bot.utils.time import parse_duration_string  log = logging.getLogger(__name__) @@ -301,16 +302,6 @@ class TagContentConverter(Converter):  class DurationDelta(Converter):      """Convert duration strings into dateutil.relativedelta.relativedelta objects.""" -    duration_parser = re.compile( -        r"((?P<years>\d+?) ?(years|year|Y|y) ?)?" -        r"((?P<months>\d+?) ?(months|month|m) ?)?" -        r"((?P<weeks>\d+?) ?(weeks|week|W|w) ?)?" -        r"((?P<days>\d+?) ?(days|day|D|d) ?)?" -        r"((?P<hours>\d+?) ?(hours|hour|H|h) ?)?" -        r"((?P<minutes>\d+?) ?(minutes|minute|M) ?)?" -        r"((?P<seconds>\d+?) ?(seconds|second|S|s))?" -    ) -      async def convert(self, ctx: Context, duration: str) -> relativedelta:          """          Converts a `duration` string to a relativedelta object. @@ -326,13 +317,9 @@ class DurationDelta(Converter):          The units need to be provided in descending order of magnitude.          """ -        match = self.duration_parser.fullmatch(duration) -        if not match: +        if not (delta := parse_duration_string(duration)):              raise BadArgument(f"`{duration}` is not a valid duration string.") -        duration_dict = {unit: int(amount) for unit, amount in match.groupdict(default=0).items()} -        delta = relativedelta(**duration_dict) -          return delta @@ -357,27 +344,38 @@ class Duration(DurationDelta):  class OffTopicName(Converter):      """A converter that ensures an added off-topic name is valid.""" +    ALLOWED_CHARACTERS = "ABCDEFGHIJKLMNOPQRSTUVWXYZ!?'`-" + +    @classmethod +    def translate_name(cls, name: str, *, from_unicode: bool = True) -> str: +        """ +        Translates `name` into a format that is allowed in discord channel names. + +        If `from_unicode` is True, the name is translated from a discord-safe format, back to normalized text. +        """ +        if from_unicode: +            table = str.maketrans(cls.ALLOWED_CHARACTERS, '๐ ๐ก๐ข๐ฃ๐ค๐ฅ๐ฆ๐ง๐จ๐ฉ๐ช๐ซ๐ฌ๐ญ๐ฎ๐ฏ๐ฐ๐ฑ๐ฒ๐ณ๐ด๐ต๐ถ๐ท๐ธ๐นว๏ผโโ-') +        else: +            table = str.maketrans('๐ ๐ก๐ข๐ฃ๐ค๐ฅ๐ฆ๐ง๐จ๐ฉ๐ช๐ซ๐ฌ๐ญ๐ฎ๐ฏ๐ฐ๐ฑ๐ฒ๐ณ๐ด๐ต๐ถ๐ท๐ธ๐นว๏ผโโ-', cls.ALLOWED_CHARACTERS) + +        return name.translate(table) +      async def convert(self, ctx: Context, argument: str) -> str:          """Attempt to replace any invalid characters with their approximate Unicode equivalent.""" -        allowed_characters = "ABCDEFGHIJKLMNOPQRSTUVWXYZ!?'`-" -          # Chain multiple words to a single one          argument = "-".join(argument.split())          if not (2 <= len(argument) <= 96):              raise BadArgument("Channel name must be between 2 and 96 chars long") -        elif not all(c.isalnum() or c in allowed_characters for c in argument): +        elif not all(c.isalnum() or c in self.ALLOWED_CHARACTERS for c in argument):              raise BadArgument(                  "Channel name must only consist of "                  "alphanumeric characters, minus signs or apostrophes."              )          # Replace invalid characters with unicode alternatives. -        table = str.maketrans( -            allowed_characters, '๐ ๐ก๐ข๐ฃ๐ค๐ฅ๐ฆ๐ง๐จ๐ฉ๐ช๐ซ๐ฌ๐ญ๐ฎ๐ฏ๐ฐ๐ฑ๐ฒ๐ณ๐ด๐ต๐ถ๐ท๐ธ๐นว๏ผโโ-' -        ) -        return argument.translate(table) +        return self.translate_name(argument)  class ISODateTime(Converter): | 
