123

database/stocks/models/__init__.py

@@ -0,0 +1,2 @@
+from .base import BaseStocksModel
+from .general import *

database/stocks/models/base.py

@@ -0,0 +1,10 @@
+from sqlalchemy import MetaData
+from sqlalchemy.ext.asyncio import AsyncAttrs
+from sqlalchemy.orm import DeclarativeBase
+
+
+class BaseStocksModel(AsyncAttrs, DeclarativeBase):
+    metadata = MetaData(schema='stocks')
+    __table_args__ = {
+        'schema': 'stocks',
+    }

database/stocks/models/general.py

@@ -0,0 +1,10 @@
+from sqlalchemy.orm import mapped_column, Mapped
+
+from .base import BaseStocksModel
+
+
+class DailyStock(BaseStocksModel):
+    __tablename__ = 'daily_stocks'
+    product_id: Mapped[int] = mapped_column(primary_key=True)
+    sold_today: Mapped[int] = mapped_column()
+    test: Mapped[bool] = mapped_column()