init

model/owned_bot_entity.py

@@ -0,0 +1,44 @@
+from sqlmodel import BIGINT, Field, select, func
+from sqlmodel.ext.asyncio.session import AsyncSession
+
+
+from .bot_entity import BotEntity
+from .descriptors import EntityField
+from .user import UserBase
+from . import session_dep
+
+
+class OwnedBotEntity(BotEntity, table = False):
+    
+    user_id: int | None =  EntityField(
+        sm_descriptor = Field(sa_type = BIGINT, foreign_key = "user.id", ondelete="SET NULL"),
+        is_visible = False)
+    
+
+    @classmethod
+    @session_dep
+    async def get_multi_by_user(cls, *,
+                        session: AsyncSession | None = None,
+                        user_id: int,
+                        order_by = None,
+                        skip: int = 0,
+                        limit: int = None):
+        
+        select_statement = select(cls).where(cls.user_id == user_id).offset(skip)
+        if limit:
+            select_statement = select_statement.limit(limit)
+        if order_by:
+            select_statement = select_statement.order_by(order_by)
+        return (await session.exec(select_statement)).all()
+    
+
+    @classmethod
+    @session_dep
+    async def get_count_by_user(cls, *,
+                        session: AsyncSession | None = None,
+                        user_id: int):
+        
+        return await session.scalar(
+            select(func.count()).
+            select_from(cls).
+            where(cls.user_id == user_id))