CopyBot/Red-DiscordBot
1
0
Fork 0
mirror of https://github.com/Cog-Creators/Red-DiscordBot.git synced 2025-11-06 03:08:55 -05:00
Code Issues Packages Projects Releases Wiki Activity

Optimize Trivia list loading (#6336)

Browse Source 
Co-authored-by: Kreusada <67752638+Kreusada@users.noreply.github.com>
This commit is contained in:
Jakub Kuczys 2024-10-29 18:58:28 +01:00 committed by GitHub
parent 4396323205
commit 4134881fae
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
redbot/cogs/trivia/trivia.py
View File

@ -29,6 +29,7 @@ __all__ = ("Trivia", "UNIQUE_ID", "InvalidListError", "get_core_lists", "get_lis
UNIQUE_ID = 0xB3C0E453
_ = Translator("Trivia", __file__)
YAMLSafeLoader = getattr(yaml, "CSafeLoader", yaml.SafeLoader)
class InvalidListError(Exception):
@ -759,7 +760,7 @@ class Trivia(commands.Cog):
return
buffer = io.BytesIO(await attachment.read())
trivia_dict = yaml.safe_load(buffer)
trivia_dict = yaml.load(buffer, YAMLSafeLoader)
TRIVIA_LIST_SCHEMA.validate(trivia_dict)
buffer.seek(0)
@ -803,7 +804,7 @@ def get_core_lists() -> List[pathlib.Path]:
return list(core_lists_path.glob("*.yaml"))
def get_list(path: pathlib.Path) -> Dict[str, Any]:
def get_list(path: pathlib.Path, *, validate_schema: bool = True) -> Dict[str, Any]:
"""
Returns a trivia list dictionary from the given path.
@ -814,12 +815,14 @@ def get_list(path: pathlib.Path) -> Dict[str, Any]:
"""
with path.open(encoding="utf-8") as file:
try:
trivia_dict = yaml.safe_load(file)
trivia_dict = yaml.load(file, YAMLSafeLoader)
except yaml.error.YAMLError as exc:
raise InvalidListError("YAML parsing failed.") from exc
if validate_schema:
try:
TRIVIA_LIST_SCHEMA.validate(trivia_dict)
except schema.SchemaError as exc:
raise InvalidListError("The list does not adhere to the schema.") from exc
return trivia_dict