Skip to content

Commit 0ab4fa4

Browse files
refactor to handle exceptions
1 parent 6795cef commit 0ab4fa4

1 file changed

Lines changed: 15 additions & 10 deletions

File tree

cogs/strike.py

Lines changed: 15 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -897,18 +897,23 @@ async def get_strikes( # type: ignore[misc]
897897
await self.command_send_error(ctx, message=member_id_not_integer_error.args[0])
898898
return
899899

900-
strike_obj: DiscordMemberStrikes = next(
901-
strike_object
902-
for strike_object in [
903-
all_strike_object
904-
async for all_strike_object in DiscordMemberStrikes.objects.select_related().all() # noqa: E501 # NOTE: I tried to reformat this to avoid the ruff error but the stupid fucking reformat put it all back on one line again
905-
]
906-
if str(strike_object.discord_member) # type: ignore[has-type]
907-
== (DiscordMember.hash_discord_id(strike_member.id))
908-
)
900+
strike_obj: DiscordMemberStrikes | None = None
901+
try:
902+
strike_obj = next(
903+
strike_object
904+
for strike_object in [
905+
all_strike_object
906+
async for all_strike_object in DiscordMemberStrikes.objects.select_related().all() # noqa: E501 # NOTE: I tried to reformat this to avoid the ruff error but the stupid fucking reformat put it all back on one line again
907+
]
908+
if str(strike_object.discord_member) # type: ignore[has-type]
909+
== (DiscordMember.hash_discord_id(strike_member.id))
910+
)
911+
except StopIteration:
912+
logger.debug("No strikes found for user %s", strike_member)
909913

910914
await ctx.respond(
911-
f"User {strike_member.mention} has {strike_obj.strikes or 0} strikes."
915+
f"User {strike_member.mention} has "
916+
f"{strike_obj.strikes if strike_obj else 0} strikes."
912917
)
913918

914919

0 commit comments

Comments
 (0)