From 84f4c099d6b35ed917b92d6c7ad7f7ede01dfa6e Mon Sep 17 00:00:00 2001 From: Sabina Date: Mon, 3 Jul 2023 17:34:10 +0300 Subject: [PATCH] Another commit --- src/bot/handlers/tasks.py | 5 +++-- src/bot/services/task.py | 3 +-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/bot/handlers/tasks.py b/src/bot/handlers/tasks.py index a038d69a..6d4d2f74 100644 --- a/src/bot/handlers/tasks.py +++ b/src/bot/handlers/tasks.py @@ -72,8 +72,9 @@ async def back_subcategory_callback(update: Update, context: ContextTypes.DEFAUL @logger_decor async def view_task_callback(update: Update, context: CallbackContext, limit: int = 3): task_service = TaskService() - user_data = context.user_data - tasks_to_show, offset, page_number = await task_service.get_user_tasks_by_page(user_data, limit) + tasks_to_show, offset, page_number = await task_service.get_user_tasks_by_page( + context.user_data.get("page_number", 1), limit + ) for task in tasks_to_show: message = display_tasks(task) diff --git a/src/bot/services/task.py b/src/bot/services/task.py index 1df671f0..1d9f4e33 100644 --- a/src/bot/services/task.py +++ b/src/bot/services/task.py @@ -14,8 +14,7 @@ class TaskService: def __init__(self, sessionmaker: Generator[AsyncSession, None, None] = get_session): self._sessionmaker = contextlib.asynccontextmanager(sessionmaker) - async def get_user_tasks_by_page(self, user_data: dict, limit: int) -> list[Task]: - page_number = user_data.get("page_number", 1) + async def get_user_tasks_by_page(self, page_number, limit: int) -> list[Task]: offset = (page_number - 1) * limit async with self._sessionmaker() as session: repository = TaskRepository(session)