nora/Birthday-Pool-Bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Little fixes, add exception handler

Browse Source
This commit is contained in:
Oleg Yurchik
2026-01-24 21:55:27 +03:00
parent 7758a3cf62
commit 239dd05992
14 changed files with 244 additions and 52 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
birthday_pool_bot/notifications/service.py
View File

@@ -35,20 +35,22 @@ class NotificationsService(facet.AsyncioServiceMixin):
users_repository = self._repositories_container.users
subscriptions_repository = self._repositories_container.subscriptions
users_generator = users_repository.get_users_by_birthdays(
birthday=birthday,
)
async with users_repository.transaction():
users = await users_repository.get_users_by_birthdays(
birthday=birthday,
)
users = [user async for user in users_generator]
if not users:
return
subscriptions_generator = subscriptions_repository.get_to_users_subscriptions(
to_users_ids={user.id for user in users},
with_from_user=True,
with_to_user=True,
with_pool=True,
)
async with subscriptions_repository.transaction():
subscriptions = await subscriptions_repository.get_to_users_subscriptions(
to_users_ids={user.id for user in users},
with_from_user=True,
with_to_user=True,
with_pool=True,
)
subscriptions = [subscription async for subscription in subscriptions_generator]
if not subscriptions:
return