Fix getting user daily predictions

ptrvsrg · Jul 4, 2024 · d727e3f · d727e3f
1 parent 90e87bf
commit d727e3f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/app/bot/middlewares/subscription_limit_filter.py b/app/bot/middlewares/subscription_limit_filter.py
@@ -5,7 +5,7 @@
 from app.bot.utils.message_utils import (
  get_username_from_message,
 )
-from app.service import user_service
+from app.service import statistics_service
 from app.config import config
 
 
@@ -16,9 +16,9 @@ def __init__(self):
 
  async def __call__(self, message: Message) -> bool:
  username = get_username_from_message(message)
- user = user_service.get_user_by_username(username)
+ statistics = statistics_service.get_statistics(username)
 
- limit_reached = user.daily_predictions >= config.bot.daily_limit
+ limit_reached = statistics.daily_predictions >= config.bot.daily_limit
 
  if limit_reached:
  await message.answer(t("error.subscription_limit"))