Xiaoxia AI
88 lines
2.5 KiB
Python
88 lines
2.5 KiB
Python
from datetime import datetime, timezone
|
|
|
|
from worker_app.celery_app import celery_app
|
|
from worker_app.db import SessionLocal
|
|
|
|
from packages.adapters.sqlalchemy_impl import (
|
|
SQLAlchemyAssetRepository,
|
|
SQLAlchemyIngestJobRepository,
|
|
)
|
|
from packages.domain import Asset, IngestJobStatus
|
|
|
|
|
|
@celery_app.task(name="worker.ingest_asset")
|
|
def ingest_asset(job_id: str) -> dict:
|
|
"""
|
|
Ingest asset task.
|
|
|
|
Steps:
|
|
1. Fetch IngestJob from repository
|
|
2. Extract metadata from storage_key (placeholder: mock metadata)
|
|
3. Create Asset entity
|
|
4. Update IngestJob status to COMPLETED
|
|
5. Return result
|
|
"""
|
|
db = SessionLocal()
|
|
job_repo = SQLAlchemyIngestJobRepository(db)
|
|
asset_repo = SQLAlchemyAssetRepository(db)
|
|
|
|
job = job_repo.get(job_id)
|
|
if job is None:
|
|
return {"status": "failed", "error": "job not found"}
|
|
|
|
try:
|
|
# Update job status to PROCESSING
|
|
job.status = IngestJobStatus.PROCESSING
|
|
job.updated_at = datetime.now(timezone.utc)
|
|
job_repo.update(job)
|
|
|
|
# Mock metadata extraction (in real implementation: use ffprobe, Pillow, etc.)
|
|
mime_type = "video/mp4" if job.storage_key.endswith(".mp4") else "image/jpeg"
|
|
metadata = {
|
|
"duration": 10.5,
|
|
"width": 1920,
|
|
"height": 1080,
|
|
"size_bytes": 1024000,
|
|
}
|
|
|
|
# Extract filename from storage_key
|
|
filename = job.storage_key.split("/")[-1]
|
|
|
|
# Create Asset
|
|
asset = Asset.create(
|
|
workspace_id=job.workspace_id,
|
|
project_id=job.project_id,
|
|
library_id=job.library_id,
|
|
name=filename,
|
|
storage_key=job.storage_key,
|
|
mime_type=mime_type,
|
|
metadata=metadata,
|
|
)
|
|
asset_repo.create(asset)
|
|
|
|
# Update job status to COMPLETED
|
|
job.status = IngestJobStatus.COMPLETED
|
|
job.result_asset_id = asset.id
|
|
job.updated_at = datetime.now(timezone.utc)
|
|
job_repo.update(job)
|
|
|
|
return {
|
|
"status": "completed",
|
|
"job_id": job.id,
|
|
"asset_id": asset.id,
|
|
}
|
|
except Exception as e:
|
|
# Update job status to FAILED
|
|
job.status = IngestJobStatus.FAILED
|
|
job.error_message = str(e)
|
|
job.updated_at = datetime.now(timezone.utc)
|
|
job_repo.update(job)
|
|
|
|
return {
|
|
"status": "failed",
|
|
"job_id": job.id,
|
|
"error": str(e),
|
|
}
|
|
finally:
|
|
db.close()
|