diff options
Diffstat (limited to 'bot/exts/utilities/conversationstarters.py')
-rw-r--r-- | bot/exts/utilities/conversationstarters.py | 9 |
1 files changed, 4 insertions, 5 deletions
diff --git a/bot/exts/utilities/conversationstarters.py b/bot/exts/utilities/conversationstarters.py index 410ea884..a019c789 100644 --- a/bot/exts/utilities/conversationstarters.py +++ b/bot/exts/utilities/conversationstarters.py @@ -2,7 +2,6 @@ import asyncio from contextlib import suppress from functools import partial from pathlib import Path -from typing import Union import discord import yaml @@ -16,11 +15,11 @@ from bot.utils.randomization import RandomCycle SUGGESTION_FORM = "https://forms.gle/zw6kkJqv8U43Nfjg9" with Path("bot/resources/utilities/starter.yaml").open("r", encoding="utf8") as f: - STARTERS = yaml.load(f, Loader=yaml.FullLoader) + STARTERS = yaml.safe_load(f) with Path("bot/resources/utilities/py_topics.yaml").open("r", encoding="utf8") as f: # First ID is #python-general and the rest are top to bottom categories of Topical Chat/Help. - PY_TOPICS = yaml.load(f, Loader=yaml.FullLoader) + PY_TOPICS = yaml.safe_load(f) # Removing `None` from lists of topics, if not a list, it is changed to an empty one. PY_TOPICS = {k: [i for i in v if i] if isinstance(v, list) else [] for k, v in PY_TOPICS.items()} @@ -67,7 +66,7 @@ class ConvoStarters(commands.Cog): @staticmethod def _predicate( - command_invoker: Union[discord.User, discord.Member], + command_invoker: discord.User | discord.Member, message: discord.Message, reaction: discord.Reaction, user: discord.User @@ -84,7 +83,7 @@ class ConvoStarters(commands.Cog): async def _listen_for_refresh( self, - command_invoker: Union[discord.User, discord.Member], + command_invoker: discord.User | discord.Member, message: discord.Message ) -> None: await message.add_reaction("🔄") |