"…wouldn’t a more graceful error make sense?" - matthew

This commit is contained in:
nex 2022-10-29 21:22:11 +01:00
parent a7fbe6ed95
commit ece91c417e

View file

@ -375,7 +375,7 @@ class AssignmentsCog(commands.Cog):
): ):
"""Views an assignment's details""" """Views an assignment's details"""
try: try:
entry_id, *_ = title.split(":") entry_id = int(title.split(":", 1)[0])
except ValueError: except ValueError:
return await ctx.respond("\N{cross mark} Invalid Input.") return await ctx.respond("\N{cross mark} Invalid Input.")
assignment: Assignments = await get_or_none(Assignments, entry_id=int(entry_id)) assignment: Assignments = await get_or_none(Assignments, entry_id=int(entry_id))
@ -395,7 +395,7 @@ class AssignmentsCog(commands.Cog):
): ):
"""Edits an assignment""" """Edits an assignment"""
try: try:
entry_id, *_ = title.split(":") entry_id = int(title.split(":", 1)[0])
except ValueError: except ValueError:
return await ctx.respond("\N{cross mark} Invalid Input.") return await ctx.respond("\N{cross mark} Invalid Input.")
assignment: Assignments = await get_or_none(Assignments, entry_id=int(entry_id)) assignment: Assignments = await get_or_none(Assignments, entry_id=int(entry_id))
@ -636,7 +636,7 @@ class AssignmentsCog(commands.Cog):
): ):
"""Edits an assignment""" """Edits an assignment"""
try: try:
entry_id, *_ = title.split(":") entry_id = int(title.split(":", 1)[0])
except ValueError: except ValueError:
return await ctx.respond("\N{cross mark} Invalid Input.") return await ctx.respond("\N{cross mark} Invalid Input.")
assignment: Assignments = await get_or_none(Assignments, entry_id=int(entry_id)) assignment: Assignments = await get_or_none(Assignments, entry_id=int(entry_id))