diff options
author | 2021-05-10 09:45:32 -0400 | |
---|---|---|
committer | 2021-05-10 09:45:32 -0400 | |
commit | e15d1eca771901280297bee15d5ee901c965d0d7 (patch) | |
tree | e4c05ee7ab13aa852c68e81464999b5bd054a356 /bot/utils | |
parent | chore: Use ctx instead of ctx.channel (diff) | |
parent | Merge pull request #725 from python-discord/change-to-fetch-user (diff) |
fix: Resolve Merge Conflicts
Diffstat (limited to 'bot/utils')
-rw-r--r-- | bot/utils/converters.py | 29 | ||||
-rw-r--r-- | bot/utils/messages.py | 19 | ||||
-rw-r--r-- | bot/utils/pagination.py | 6 |
3 files changed, 51 insertions, 3 deletions
diff --git a/bot/utils/converters.py b/bot/utils/converters.py index 72b64848..9e9616d8 100644 --- a/bot/utils/converters.py +++ b/bot/utils/converters.py @@ -85,3 +85,32 @@ class DateConverter(commands.Converter): f"Can't convert `{argument}` to `datetime` in format `YYYY-MM-DD` or `int` in SOL." ) return date + + +class Subreddit(commands.Converter): + """Forces a string to begin with "r/" and checks if it's a valid subreddit.""" + + @staticmethod + async def convert(ctx: commands.Context, sub: str) -> str: + """ + Force sub to begin with "r/" and check if it's a valid subreddit. + + If sub is a valid subreddit, return it prepended with "r/" + """ + sub = sub.lower() + + if not sub.startswith("r/"): + sub = f"r/{sub}" + + resp = await ctx.bot.http_session.get( + "https://www.reddit.com/subreddits/search.json", + params={"q": sub} + ) + + json = await resp.json() + if not json["data"]["children"]: + raise commands.BadArgument( + f"The subreddit `{sub}` either doesn't exist, or it has no posts." + ) + + return sub diff --git a/bot/utils/messages.py b/bot/utils/messages.py new file mode 100644 index 00000000..a6c035f9 --- /dev/null +++ b/bot/utils/messages.py @@ -0,0 +1,19 @@ +import re +from typing import Optional + + +def sub_clyde(username: Optional[str]) -> Optional[str]: + """ + Replace "e"/"E" in any "clyde" in `username` with a Cyrillic "е"/"E" and return the new string. + + Discord disallows "clyde" anywhere in the username for webhooks. It will return a 400. + Return None only if `username` is None. + """ + def replace_e(match: re.Match) -> str: + char = "е" if match[2] == "e" else "Е" + return match[1] + char + + if username: + return re.sub(r"(clyd)(e)", replace_e, username, flags=re.I) + else: + return username # Empty string or None diff --git a/bot/utils/pagination.py b/bot/utils/pagination.py index e0dc25b4..742281d7 100644 --- a/bot/utils/pagination.py +++ b/bot/utils/pagination.py @@ -4,6 +4,7 @@ from typing import Iterable, List, Optional, Tuple from discord import Embed, Member, Reaction from discord.abc import User +from discord.embeds import EmptyEmbed from discord.ext.commands import Context, Paginator from bot.constants import Emojis @@ -419,9 +420,8 @@ class ImagePaginator(Paginator): await message.edit(embed=embed) embed.description = paginator.pages[current_page] - image = paginator.images[current_page] - if image: - embed.set_image(url=image) + image = paginator.images[current_page] or EmptyEmbed + embed.set_image(url=image) embed.set_footer(text=f"Page {current_page + 1}/{len(paginator.pages)}") log.debug(f"Got {reaction_type} page reaction - changing to page {current_page + 1}/{len(paginator.pages)}") |