fix(typing): resolve mypy errors in db.repositories.computes

This commit is contained in:
Mark Paronyan 2026-09-29 22:02:22 +03:00
parent 2fe4e39715
commit 36945daa83
No known key found for this signature in database
GPG Key ID: 8DC7338A87FE22F8

View File

@ -19,6 +19,8 @@ from uuid import UUID
from typing import Optional, List, Union from typing import Optional, List, Union
from sqlalchemy import select, update, delete from sqlalchemy import select, update, delete
from sqlalchemy.ext.asyncio import AsyncSession from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy.engine import CursorResult
from typing import cast
from .base import BaseRepository from .base import BaseRepository
@ -47,7 +49,7 @@ class ComputesRepository(BaseRepository):
query = select(models.Compute) query = select(models.Compute)
result = await self._db_session.execute(query) result = await self._db_session.execute(query)
return result.scalars().all() return list(result.scalars().all())
async def create_compute(self, compute_create: schemas.ComputeCreate) -> models.Compute: async def create_compute(self, compute_create: schemas.ComputeCreate) -> models.Compute:
@ -58,7 +60,7 @@ class ComputesRepository(BaseRepository):
host=compute_create.host, host=compute_create.host,
port=compute_create.port, port=compute_create.port,
user=compute_create.user, user=compute_create.user,
password=compute_create.password.get_secret_value(), password=compute_create.password.get_secret_value() if compute_create.password else None,
) )
self._db_session.add(db_compute) self._db_session.add(db_compute)
await self._db_session.commit() await self._db_session.commit()
@ -87,4 +89,4 @@ class ComputesRepository(BaseRepository):
query = delete(models.Compute).where(models.Compute.compute_id == compute_id) query = delete(models.Compute).where(models.Compute.compute_id == compute_id)
result = await self._db_session.execute(query) result = await self._db_session.execute(query)
await self._db_session.commit() await self._db_session.commit()
return result.rowcount > 0 return cast(CursorResult, result).rowcount > 0