major refactor

core/db/__init__.py

@@ -0,0 +1,19 @@
+from .assets import (
+    create_asset,
+    delete_asset,
+    get_asset,
+    get_asset_filenames,
+    list_assets,
+    update_asset,
+)
+from .jobs import (
+    create_job,
+    get_job,
+    list_jobs,
+    update_job,
+    update_job_fields,
+)
+from .presets import (
+    get_preset,
+    list_presets,
+)

core/db/assets.py

@@ -0,0 +1,48 @@
+"""Database operations for MediaAsset."""
+
+from typing import Optional
+from uuid import UUID
+
+
+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)
+
+
+def get_asset(id: UUID):
+    from admin.mpr.media_assets.models import MediaAsset
+
+    return MediaAsset.objects.get(id=id)
+
+
+def get_asset_filenames() -> set[str]:
+    from admin.mpr.media_assets.models import MediaAsset
+
+    return set(MediaAsset.objects.values_list("filename", flat=True))
+
+
+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 update_asset(asset, **fields):
+    for key, value in fields.items():
+        setattr(asset, key, value)
+    asset.save(update_fields=list(fields.keys()))
+    return asset
+
+
+def delete_asset(asset):
+    asset.delete()

core/db/jobs.py

@@ -0,0 +1,40 @@
+"""Database operations for TranscodeJob."""
+
+from typing import Optional
+from uuid import UUID
+
+
+def list_jobs(status: Optional[str] = None, source_asset_id: Optional[UUID] = None):
+    from admin.mpr.media_assets.models import TranscodeJob
+
+    qs = TranscodeJob.objects.all()
+    if status:
+        qs = qs.filter(status=status)
+    if source_asset_id:
+        qs = qs.filter(source_asset_id=source_asset_id)
+    return list(qs)
+
+
+def get_job(id: UUID):
+    from admin.mpr.media_assets.models import TranscodeJob
+
+    return TranscodeJob.objects.get(id=id)
+
+
+def create_job(**fields):
+    from admin.mpr.media_assets.models import TranscodeJob
+
+    return TranscodeJob.objects.create(**fields)
+
+
+def update_job(job, **fields):
+    for key, value in fields.items():
+        setattr(job, key, value)
+    job.save(update_fields=list(fields.keys()))
+    return job
+
+
+def update_job_fields(job_id, **fields):
+    from admin.mpr.media_assets.models import TranscodeJob
+
+    TranscodeJob.objects.filter(id=job_id).update(**fields)

core/db/presets.py

@@ -0,0 +1,15 @@
+"""Database operations for TranscodePreset."""
+
+from uuid import UUID
+
+
+def list_presets():
+    from admin.mpr.media_assets.models import TranscodePreset
+
+    return list(TranscodePreset.objects.all())
+
+
+def get_preset(id: UUID):
+    from admin.mpr.media_assets.models import TranscodePreset
+
+    return TranscodePreset.objects.get(id=id)