From 13739302af0647e29f34e253543ff22a5dcde26a Mon Sep 17 00:00:00 2001 From: ivankirpichnikov Date: Sat, 22 Nov 2025 13:56:18 +0300 Subject: [PATCH] fix mypy --- src/template_project/adapters/data_gateways/resume.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/template_project/adapters/data_gateways/resume.py b/src/template_project/adapters/data_gateways/resume.py index fb750cc..50256e6 100644 --- a/src/template_project/adapters/data_gateways/resume.py +++ b/src/template_project/adapters/data_gateways/resume.py @@ -4,7 +4,7 @@ from typing import override from sqlalchemy import select from sqlalchemy.ext.asyncio import AsyncSession -from template_project.adapters.data_gateways.tables import resume_prediction_table +from template_project.adapters.data_gateways.tables import resume_prediction_table, resume_table from template_project.application.resume.data_gateway import ResumeDataGateway, ResumePredictionDataGateway from template_project.application.resume.entity import Resume, ResumeEmbeddingId, ResumeId, ResumePrediction from template_project.application.resume.errors import ResumeNotFoundError @@ -31,7 +31,7 @@ class DefaultResumeDataGateway(ResumeDataGateway): async def list_by_user_id(self, user_id: UserId, limit: int, offset: int) -> Sequence[Resume]: statement = ( select(Resume) - .where(Resume.user_id == user_id) + .where(resume_table.c.user_id == user_id) .limit(limit) .offset(offset) ) @@ -42,8 +42,8 @@ class DefaultResumeDataGateway(ResumeDataGateway): async def list_latest_by_user_id(self, user_id: UserId, limit: int, offset: int) -> Sequence[Resume]: statement = ( select(Resume) - .where(Resume.user_id == user_id) - .where(Resume.up_resume_id.is_(None)) + .where(resume_table.c.user_id == user_id) + .where(resume_table.c.up_resume_id.is_(None)) .limit(limit) .offset(offset) ) @@ -52,6 +52,7 @@ class DefaultResumeDataGateway(ResumeDataGateway): @override async def get_history(self, resume_id: ResumeId) -> Sequence[Resume]: + # TODO: N+1 history: list[Resume] = [] current_resume = await self.load(resume_id) history.append(current_resume)