feat: first commit, auth

db/base.py

@@ -0,0 +1,4 @@
+from sqlalchemy.orm import DeclarativeBase
+
+
+class Base(DeclarativeBase): ...

db/models/__init__.py

@@ -0,0 +1,4 @@
+from .sessions import Session
+from .users import User
+
+__all__ = ["Session", "User"]

db/models/sessions.py

@@ -0,0 +1,29 @@
+from datetime import datetime
+from typing import TYPE_CHECKING
+
+from sqlalchemy import BOOLEAN, INTEGER, TEXT, ForeignKey, func
+from sqlalchemy.dialects.postgresql import TIMESTAMP
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from db.base import Base
+
+if TYPE_CHECKING:
+    from db.models.users import User
+
+
+class Session(Base):
+    __tablename__ = "sessions"
+
+    id: Mapped[int] = mapped_column(
+        INTEGER, unique=True, autoincrement=True, nullable=False, primary_key=True
+    )
+    user_id: Mapped[int] = mapped_column(ForeignKey("users.id"), nullable=False)
+    refresh_token_hash: Mapped[str] = mapped_column(TEXT, nullable=False)
+
+    source: Mapped[str] = mapped_column(TEXT, nullable=False)
+    created_at: Mapped[datetime] = mapped_column(TIMESTAMP, default=func.now())
+
+    is_revoked: Mapped[bool] = mapped_column(BOOLEAN, default=False)
+    revoked_at: Mapped[datetime] = mapped_column(TIMESTAMP, nullable=True)
+
+    user: Mapped["User"] = relationship(back_populates="sessions", lazy="selectin")

db/models/users.py

@@ -0,0 +1,22 @@
+from typing import TYPE_CHECKING
+
+from sqlalchemy import BIGINT, TEXT, VARCHAR
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from db.base import Base
+
+if TYPE_CHECKING:
+    from db.models.sessions import Session
+
+
+class User(Base):
+    __tablename__ = "users"
+
+    id: Mapped[int] = mapped_column(
+        BIGINT, unique=True, autoincrement=True, nullable=False, primary_key=True
+    )
+    username: Mapped[str] = mapped_column(TEXT, unique=True, nullable=True)
+    hashed_password: Mapped[str] = mapped_column(VARCHAR(255), nullable=True)
+    telegram_id: Mapped[int] = mapped_column(BIGINT, unique=True, nullable=True)
+
+    sessions: Mapped[list["Session"]] = relationship(back_populates="user", lazy="selectin")

db/session.py

@@ -0,0 +1,11 @@
+from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine
+
+from config import cfg
+
+engine = create_async_engine(cfg.db_url, echo=True)
+async_session = async_sessionmaker(bind=engine, expire_on_commit=False)
+
+
+async def get_db():
+    async with async_session() as session:
+        yield session