mirror of
https://github.com/GNS3/gns3-server.git
synced 2026-10-01 08:10:13 +03:00
fix(typing): resolve mypy errors in db.repositories.computes
This commit is contained in:
parent
2fe4e39715
commit
36945daa83
@ -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
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user