PaiGram/core/admin/repositories.py

34 lines
1.1 KiB
Python
Raw Normal View History

from typing import List, cast
from sqlalchemy import select
from sqlmodel.ext.asyncio.session import AsyncSession
2022-07-26 10:07:31 +00:00
from core.admin.models import Admin
from core.base.mysql import MySQL
2022-07-26 10:07:31 +00:00
class BotAdminRepository:
def __init__(self, mysql: MySQL):
self.mysql = mysql
async def delete_by_user_id(self, user_id: int):
async with self.mysql.Session() as session:
session = cast(AsyncSession, session)
statement = select(Admin).where(Admin.user_id == user_id)
results = await session.exec(statement)
admin = results.one()
await session.delete(admin)
2022-07-26 10:07:31 +00:00
async def add_by_user_id(self, user_id: int):
async with self.mysql.Session() as session:
admin = Admin(user_id=user_id)
session.add(admin)
await session.commit()
2022-07-26 10:07:31 +00:00
async def get_all_user_id(self) -> List[int]:
async with self.mysql.Session() as session:
query = select(Admin)
results = await session.exec(query)
admins = results.all()
return [admin[0].user_id for admin in admins]