mirror of
https://github.com/PaiGramTeam/GramCore.git
synced 2024-11-26 15:59:33 +00:00
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
from typing import Optional
|
|
|
|
from sqlmodel import select
|
|
|
|
from gram_core.base_service import BaseService
|
|
from gram_core.dependence.database import Database
|
|
from gram_core.services.devices.models import DevicesDataBase as Devices
|
|
from gram_core.sqlmodel.session import AsyncSession
|
|
|
|
__all__ = ("DevicesRepository",)
|
|
|
|
|
|
class DevicesRepository(BaseService.Component):
|
|
def __init__(self, database: Database):
|
|
self.engine = database.engine
|
|
|
|
async def get(
|
|
self,
|
|
account_id: int,
|
|
) -> Optional[Devices]:
|
|
async with AsyncSession(self.engine) as session:
|
|
statement = select(Devices).where(Devices.account_id == account_id)
|
|
results = await session.exec(statement)
|
|
return results.first()
|
|
|
|
async def add(self, devices: Devices) -> None:
|
|
async with AsyncSession(self.engine) as session:
|
|
session.add(devices)
|
|
await session.commit()
|
|
|
|
async def update(self, devices: Devices) -> Devices:
|
|
async with AsyncSession(self.engine) as session:
|
|
session.add(devices)
|
|
await session.commit()
|
|
await session.refresh(devices)
|
|
return devices
|
|
|
|
async def delete(self, devices: Devices) -> None:
|
|
async with AsyncSession(self.engine) as session:
|
|
await session.delete(devices)
|
|
await session.commit()
|