feat(E): add MediaAsset catalog — model, CRUD API, MediaBrowser UI
Migration 040: media_assets table with RLS (tenant_isolation + admin_bypass). domains/media/: MediaAsset model, schemas, service, router with ZIP-download. publish_asset Celery task in rendering/tasks.py. core/storage.py: download_bytes() method for MinIO + LocalStorage. frontend: MediaBrowser.tsx (grid/list, multi-select, zip-download, pagination) + api/media.ts. Route /media (AdminRoute) + sidebar link with Image icon for admin+pm. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,52 @@
|
||||
import uuid
|
||||
import enum
|
||||
from datetime import datetime
|
||||
from sqlalchemy import String, DateTime, Boolean, Text, BigInteger, Float, Integer, ForeignKey
|
||||
from sqlalchemy import Enum as SAEnum
|
||||
from sqlalchemy.orm import Mapped, mapped_column
|
||||
from sqlalchemy.dialects.postgresql import UUID, JSONB
|
||||
from app.database import Base
|
||||
|
||||
|
||||
class MediaAssetType(str, enum.Enum):
|
||||
thumbnail = "thumbnail"
|
||||
still = "still"
|
||||
turntable = "turntable"
|
||||
stl_low = "stl_low"
|
||||
stl_high = "stl_high"
|
||||
gltf_geometry = "gltf_geometry"
|
||||
gltf_production = "gltf_production"
|
||||
blend_production = "blend_production"
|
||||
|
||||
|
||||
class MediaAsset(Base):
|
||||
__tablename__ = "media_assets"
|
||||
|
||||
id: Mapped[uuid.UUID] = mapped_column(UUID(as_uuid=True), primary_key=True, default=uuid.uuid4)
|
||||
tenant_id: Mapped[uuid.UUID | None] = mapped_column(
|
||||
UUID(as_uuid=True), ForeignKey("tenants.id", ondelete="CASCADE"), nullable=True, index=True
|
||||
)
|
||||
product_id: Mapped[uuid.UUID | None] = mapped_column(
|
||||
UUID(as_uuid=True), ForeignKey("products.id", ondelete="CASCADE"), nullable=True, index=True
|
||||
)
|
||||
cad_file_id: Mapped[uuid.UUID | None] = mapped_column(
|
||||
UUID(as_uuid=True), ForeignKey("cad_files.id", ondelete="SET NULL"), nullable=True
|
||||
)
|
||||
order_line_id: Mapped[uuid.UUID | None] = mapped_column(
|
||||
UUID(as_uuid=True), ForeignKey("order_lines.id", ondelete="SET NULL"), nullable=True, index=True
|
||||
)
|
||||
workflow_run_id: Mapped[uuid.UUID | None] = mapped_column(
|
||||
UUID(as_uuid=True), ForeignKey("workflow_runs.id", ondelete="SET NULL"), nullable=True
|
||||
)
|
||||
asset_type: Mapped[MediaAssetType] = mapped_column(
|
||||
SAEnum(MediaAssetType, name="media_asset_type"), nullable=False
|
||||
)
|
||||
storage_key: Mapped[str] = mapped_column(Text, nullable=False)
|
||||
file_size_bytes: Mapped[int | None] = mapped_column(BigInteger, nullable=True)
|
||||
mime_type: Mapped[str | None] = mapped_column(String(100), nullable=True)
|
||||
width: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
||||
height: Mapped[int | None] = mapped_column(Integer, nullable=True)
|
||||
duration_s: Mapped[float | None] = mapped_column(Float, nullable=True)
|
||||
render_config: Mapped[dict | None] = mapped_column(JSONB, nullable=True)
|
||||
is_archived: Mapped[bool] = mapped_column(Boolean, nullable=False, default=False)
|
||||
created_at: Mapped[datetime] = mapped_column(DateTime, default=datetime.utcnow, nullable=False)
|
||||
@@ -0,0 +1,101 @@
|
||||
"""MediaAsset router — /api/media."""
|
||||
import io
|
||||
import uuid
|
||||
import zipfile
|
||||
|
||||
from fastapi import APIRouter, Depends, HTTPException, Query
|
||||
from fastapi.responses import StreamingResponse
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from app.database import get_db
|
||||
from app.domains.media.models import MediaAssetType
|
||||
from app.domains.media.schemas import MediaAssetOut
|
||||
from app.domains.media import service
|
||||
|
||||
router = APIRouter(prefix="/api/media", tags=["media"])
|
||||
|
||||
|
||||
@router.get("/", response_model=list[MediaAssetOut])
|
||||
async def list_assets(
|
||||
product_id: uuid.UUID | None = None,
|
||||
order_line_id: uuid.UUID | None = None,
|
||||
asset_type: MediaAssetType | None = None,
|
||||
skip: int = Query(0, ge=0),
|
||||
limit: int = Query(50, ge=1, le=500),
|
||||
db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
assets = await service.list_media_assets(
|
||||
db,
|
||||
product_id=product_id,
|
||||
order_line_id=order_line_id,
|
||||
asset_type=asset_type,
|
||||
skip=skip,
|
||||
limit=limit,
|
||||
)
|
||||
for a in assets:
|
||||
a.download_url = service.get_download_url(a)
|
||||
return assets
|
||||
|
||||
|
||||
@router.get("/{asset_id}", response_model=MediaAssetOut)
|
||||
async def get_asset(asset_id: uuid.UUID, db: AsyncSession = Depends(get_db)):
|
||||
asset = await service.get_media_asset(db, asset_id)
|
||||
if not asset:
|
||||
raise HTTPException(404, "Asset not found")
|
||||
asset.download_url = service.get_download_url(asset)
|
||||
return asset
|
||||
|
||||
|
||||
@router.get("/{asset_id}/download")
|
||||
async def download_asset(asset_id: uuid.UUID, db: AsyncSession = Depends(get_db)):
|
||||
from fastapi.responses import RedirectResponse
|
||||
asset = await service.get_media_asset(db, asset_id)
|
||||
if not asset:
|
||||
raise HTTPException(404, "Asset not found")
|
||||
url = service.get_download_url(asset)
|
||||
if url:
|
||||
return RedirectResponse(url)
|
||||
raise HTTPException(404, "File not available")
|
||||
|
||||
|
||||
@router.post("/zip")
|
||||
async def zip_download(
|
||||
asset_ids: list[uuid.UUID],
|
||||
db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
assets = []
|
||||
for aid in asset_ids:
|
||||
a = await service.get_media_asset(db, aid)
|
||||
if a:
|
||||
assets.append(a)
|
||||
if not assets:
|
||||
raise HTTPException(404, "No assets found")
|
||||
|
||||
def generate():
|
||||
buf = io.BytesIO()
|
||||
with zipfile.ZipFile(buf, "w", zipfile.ZIP_DEFLATED) as zf:
|
||||
from app.core.storage import get_storage
|
||||
storage = get_storage()
|
||||
for a in assets:
|
||||
ext = (a.mime_type or "").split("/")[-1] or "bin"
|
||||
fname = f"{a.asset_type.value}_{a.id}.{ext}"
|
||||
try:
|
||||
data = storage.download_bytes(a.storage_key)
|
||||
zf.writestr(fname, data)
|
||||
except Exception:
|
||||
pass
|
||||
yield buf.getvalue()
|
||||
|
||||
return StreamingResponse(
|
||||
generate(),
|
||||
media_type="application/zip",
|
||||
headers={"Content-Disposition": "attachment; filename=media-export.zip"},
|
||||
)
|
||||
|
||||
|
||||
@router.delete("/{asset_id}")
|
||||
async def archive_asset(asset_id: uuid.UUID, db: AsyncSession = Depends(get_db)):
|
||||
asset = await service.archive_media_asset(db, asset_id)
|
||||
if not asset:
|
||||
raise HTTPException(404, "Asset not found")
|
||||
return {"ok": True}
|
||||
@@ -0,0 +1,26 @@
|
||||
import uuid
|
||||
from datetime import datetime
|
||||
from pydantic import BaseModel
|
||||
from app.domains.media.models import MediaAssetType
|
||||
|
||||
|
||||
class MediaAssetOut(BaseModel):
|
||||
id: uuid.UUID
|
||||
tenant_id: uuid.UUID | None
|
||||
product_id: uuid.UUID | None
|
||||
cad_file_id: uuid.UUID | None
|
||||
order_line_id: uuid.UUID | None
|
||||
workflow_run_id: uuid.UUID | None
|
||||
asset_type: MediaAssetType
|
||||
storage_key: str
|
||||
file_size_bytes: int | None
|
||||
mime_type: str | None
|
||||
width: int | None
|
||||
height: int | None
|
||||
duration_s: float | None
|
||||
render_config: dict | None
|
||||
is_archived: bool
|
||||
created_at: datetime
|
||||
download_url: str | None = None
|
||||
|
||||
model_config = {"from_attributes": True}
|
||||
@@ -0,0 +1,68 @@
|
||||
"""MediaAsset service."""
|
||||
import uuid
|
||||
from sqlalchemy import select, update as sql_update
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
from app.domains.media.models import MediaAsset, MediaAssetType
|
||||
|
||||
|
||||
async def list_media_assets(
|
||||
db: AsyncSession,
|
||||
product_id: uuid.UUID | None = None,
|
||||
order_line_id: uuid.UUID | None = None,
|
||||
asset_type: MediaAssetType | None = None,
|
||||
is_archived: bool | None = False,
|
||||
skip: int = 0,
|
||||
limit: int = 50,
|
||||
) -> list[MediaAsset]:
|
||||
q = select(MediaAsset).order_by(MediaAsset.created_at.desc())
|
||||
if product_id:
|
||||
q = q.where(MediaAsset.product_id == product_id)
|
||||
if order_line_id:
|
||||
q = q.where(MediaAsset.order_line_id == order_line_id)
|
||||
if asset_type:
|
||||
q = q.where(MediaAsset.asset_type == asset_type)
|
||||
if is_archived is not None:
|
||||
q = q.where(MediaAsset.is_archived == is_archived)
|
||||
q = q.offset(skip).limit(limit)
|
||||
result = await db.execute(q)
|
||||
return list(result.scalars().all())
|
||||
|
||||
|
||||
async def get_media_asset(db: AsyncSession, asset_id: uuid.UUID) -> MediaAsset | None:
|
||||
result = await db.execute(select(MediaAsset).where(MediaAsset.id == asset_id))
|
||||
return result.scalar_one_or_none()
|
||||
|
||||
|
||||
async def create_media_asset(db: AsyncSession, **kwargs) -> MediaAsset:
|
||||
asset = MediaAsset(**kwargs)
|
||||
db.add(asset)
|
||||
await db.commit()
|
||||
await db.refresh(asset)
|
||||
return asset
|
||||
|
||||
|
||||
async def archive_media_asset(db: AsyncSession, asset_id: uuid.UUID) -> MediaAsset | None:
|
||||
await db.execute(
|
||||
sql_update(MediaAsset).where(MediaAsset.id == asset_id).values(is_archived=True)
|
||||
)
|
||||
await db.commit()
|
||||
return await get_media_asset(db, asset_id)
|
||||
|
||||
|
||||
async def delete_media_asset(db: AsyncSession, asset_id: uuid.UUID) -> bool:
|
||||
asset = await get_media_asset(db, asset_id)
|
||||
if not asset:
|
||||
return False
|
||||
await db.delete(asset)
|
||||
await db.commit()
|
||||
return True
|
||||
|
||||
|
||||
def get_download_url(asset: MediaAsset) -> str | None:
|
||||
"""Get presigned URL from MinIO or local path."""
|
||||
try:
|
||||
from app.core.storage import get_storage
|
||||
storage = get_storage()
|
||||
return storage.get_url(asset.storage_key)
|
||||
except Exception:
|
||||
return f"/uploads/{asset.storage_key}"
|
||||
@@ -220,6 +220,44 @@ def render_turntable_task(
|
||||
}
|
||||
|
||||
|
||||
@celery_app.task(
|
||||
name="rendering.publish_asset",
|
||||
queue="step_processing",
|
||||
)
|
||||
def publish_asset(
|
||||
order_line_id: str,
|
||||
asset_type: str,
|
||||
storage_key: str,
|
||||
render_config: dict | None = None,
|
||||
) -> str | None:
|
||||
"""Create a MediaAsset record after a successful render."""
|
||||
import asyncio
|
||||
|
||||
async def _run() -> str | None:
|
||||
from app.database import AsyncSessionLocal
|
||||
from app.domains.media.models import MediaAsset, MediaAssetType
|
||||
from app.domains.orders.models import OrderLine
|
||||
from sqlalchemy import select
|
||||
|
||||
async with AsyncSessionLocal() as db:
|
||||
res = await db.execute(select(OrderLine).where(OrderLine.id == order_line_id))
|
||||
line = res.scalar_one_or_none()
|
||||
if not line:
|
||||
return None
|
||||
asset = MediaAsset(
|
||||
tenant_id=getattr(line, "tenant_id", None),
|
||||
order_line_id=line.id,
|
||||
asset_type=MediaAssetType(asset_type),
|
||||
storage_key=storage_key,
|
||||
render_config=render_config,
|
||||
)
|
||||
db.add(asset)
|
||||
await db.commit()
|
||||
return str(asset.id)
|
||||
|
||||
return asyncio.get_event_loop().run_until_complete(_run())
|
||||
|
||||
|
||||
def _build_ffmpeg_cmd(
|
||||
frames_dir: Path, output_mp4: Path, fps: int = 30, bg_color: str = ""
|
||||
) -> list:
|
||||
|
||||
Reference in New Issue
Block a user