From ce4db6751af7292a27192b2694cb2f4da218f04b Mon Sep 17 00:00:00 2001 From: wlinator Date: Wed, 14 Aug 2024 02:58:49 -0400 Subject: [PATCH] chore: Refactor birthday_service.py to simplify code --- services/birthday_service.py | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/services/birthday_service.py b/services/birthday_service.py index 151f0e9..5ec2c66 100644 --- a/services/birthday_service.py +++ b/services/birthday_service.py @@ -38,9 +38,7 @@ class Birthday: tz = pytz.timezone("US/Eastern") today = datetime.datetime.now(tz).strftime("%m-%d") - birthdays = database.select_query(query, (today,)) - - return birthdays + return database.select_query(query, (today,)) @staticmethod def get_upcoming_birthdays(guild_id): @@ -53,10 +51,4 @@ class Birthday: data = database.select_query(query, (guild_id,)) - upcoming = [] - for row in data: - user_id = row[0] - birthday = row[1] - upcoming.append((user_id, birthday)) - - return upcoming + return [(row[0], row[1]) for row in data]