init: new structure + fix lint errors

cpv3/db/base.py

@@ -0,0 +1,25 @@
+from __future__ import annotations
+
+import uuid
+from datetime import datetime, timezone
+
+from sqlalchemy import Boolean, DateTime
+from sqlalchemy.dialects.postgresql import UUID
+from sqlalchemy.orm import DeclarativeBase, Mapped, mapped_column
+
+
+def utcnow() -> datetime:
+    return datetime.now(timezone.utc)
+
+
+class Base(DeclarativeBase):
+    pass
+
+
+class BaseModelMixin:
+    id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
+    created_at: Mapped[datetime] = mapped_column(DateTime(timezone=True), default=utcnow)
+    updated_at: Mapped[datetime] = mapped_column(
+        DateTime(timezone=True), default=utcnow, onupdate=utcnow
+    )
+    is_active: Mapped[bool] = mapped_column(Boolean, default=True)

cpv3/db/models.py

@@ -0,0 +1,21 @@
+from cpv3.db.base import Base
+from cpv3.modules.jobs.models import Job, JobEvent
+from cpv3.modules.media.models import ArtifactMediaFile, MediaFile
+from cpv3.modules.projects.models import Project
+from cpv3.modules.files.models import File
+from cpv3.modules.transcription.models import Transcription
+from cpv3.modules.users.models import User
+from cpv3.modules.webhooks.models import Webhook
+
+__all__ = [
+    "Base",
+    "User",
+    "Project",
+    "File",
+    "MediaFile",
+    "ArtifactMediaFile",
+    "Transcription",
+    "Job",
+    "JobEvent",
+    "Webhook",
+]

cpv3/db/session.py

@@ -0,0 +1,22 @@
+from __future__ import annotations
+
+from collections.abc import AsyncGenerator
+
+from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
+
+from cpv3.infrastructure.settings import get_settings
+
+
+_settings = get_settings()
+_engine = create_async_engine(
+    _settings.get_database_url(),
+    echo=_settings.debug,
+    pool_pre_ping=True,
+)
+
+SessionLocal = async_sessionmaker(bind=_engine, class_=AsyncSession, expire_on_commit=False)
+
+
+async def get_db() -> AsyncGenerator[AsyncSession, None]:
+    async with SessionLocal() as session:
+        yield session