mirror of
https://github.com/PaiGramTeam/GramCore.git
synced 2024-11-22 22:37:23 +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()
|