2022-08-04 13:18:23 +00:00
|
|
|
from typing import List, cast
|
|
|
|
|
|
|
|
from sqlalchemy import select
|
|
|
|
from sqlmodel.ext.asyncio.session import AsyncSession
|
2022-07-26 10:07:31 +00:00
|
|
|
|
2022-09-08 01:08:37 +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):
|
2022-08-04 13:18:23 +00:00
|
|
|
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):
|
2022-08-04 13:18:23 +00:00
|
|
|
async with self.mysql.Session() as session:
|
|
|
|
admin = Admin(user_id=user_id)
|
2022-08-06 06:38:09 +00:00
|
|
|
session.add(admin)
|
|
|
|
await session.commit()
|
2022-07-26 10:07:31 +00:00
|
|
|
|
2022-08-04 13:18:23 +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]
|