move to postgresql

core/db/assets.py

@@ -1,48 +1,58 @@
-"""Database operations for MediaAsset."""
+"""Database operations for MediaAsset — SQLModel."""
+
+from __future__ import annotations
 
 from typing import Optional
 from uuid import UUID
 
+from sqlmodel import select
 
-def list_assets(status: Optional[str] = None, search: Optional[str] = None):
-    from admin.mpr.media_assets.models import MediaAsset
-
-    qs = MediaAsset.objects.all()
-    if status:
-        qs = qs.filter(status=status)
-    if search:
-        qs = qs.filter(filename__icontains=search)
-    return list(qs)
+from .connection import get_session
+from .models import MediaAsset
 
 
-def get_asset(id: UUID):
-    from admin.mpr.media_assets.models import MediaAsset
+def list_assets(status: Optional[str] = None, search: Optional[str] = None) -> list[MediaAsset]:
+    with get_session() as session:
+        stmt = select(MediaAsset)
+        if status:
+            stmt = stmt.where(MediaAsset.status == status)
+        if search:
+            stmt = stmt.where(MediaAsset.filename.ilike(f"%{search}%"))
+        return list(session.exec(stmt).all())
 
-    return MediaAsset.objects.get(id=id)
+
+def get_asset(id: UUID) -> MediaAsset | None:
+    with get_session() as session:
+        return session.get(MediaAsset, id)
 
 
 def get_asset_filenames() -> set[str]:
-    from admin.mpr.media_assets.models import MediaAsset
-
-    return set(MediaAsset.objects.values_list("filename", flat=True))
+    with get_session() as session:
+        return set(session.exec(select(MediaAsset.filename)).all())
 
 
-def create_asset(*, filename: str, file_path: str, file_size: int):
-    from admin.mpr.media_assets.models import MediaAsset
-
-    return MediaAsset.objects.create(
-        filename=filename,
-        file_path=file_path,
-        file_size=file_size,
-    )
+def create_asset(*, filename: str, file_path: str, file_size: int) -> MediaAsset:
+    asset = MediaAsset(filename=filename, file_path=file_path, file_size=file_size)
+    with get_session() as session:
+        session.add(asset)
+        session.commit()
+        session.refresh(asset)
+        return asset
 
 
-def update_asset(asset, **fields):
-    for key, value in fields.items():
-        setattr(asset, key, value)
-    asset.save(update_fields=list(fields.keys()))
-    return asset
+def update_asset(id: UUID, **fields) -> None:
+    with get_session() as session:
+        asset = session.get(MediaAsset, id)
+        if not asset:
+            return
+        for k, v in fields.items():
+            setattr(asset, k, v)
+        session.commit()
 
 
-def delete_asset(asset):
-    asset.delete()
+def delete_asset(id: UUID) -> None:
+    with get_session() as session:
+        asset = session.get(MediaAsset, id)
+        if asset:
+            session.delete(asset)
+            session.commit()