feat: upgrade tooling, fix lints

This commit is contained in:
Felix Spöttel
2023-01-25 16:56:59 +01:00
parent b4e57451e8
commit 8669a18110
17 changed files with 159 additions and 68 deletions

View File

@@ -1,7 +1,37 @@
from celery import Celery
from time import sleep
from uuid import UUID
from celery import Celery
from sqlalchemy.orm import Session
import app.shared.db.dtos as dtos
import app.shared.db.models as models
from app.shared.config import settings
from app.shared.db.base import SessionLocal
celery = Celery(__name__)
celery.conf.broker_url = settings.REDIS_URI
celery.conf.broker_url = settings.BROKER_URI
def update_job_status(db: Session, job_id: UUID, status: dtos.JobStatus) -> None:
db.begin()
job = db.query(models.Job).filter(models.Job.id == job_id).one()
job.status = status
db.commit()
@celery.task()
def transcribe(job_id: UUID) -> int:
try:
db: Session = SessionLocal()
update_job_status(db, job_id, dtos.JobStatus.processing)
sleep(60)
update_job_status(db, job_id, dtos.JobStatus.success)
db.commit()
except Exception:
update_job_status(db, job_id, dtos.JobStatus.error)
finally:
db.close()
return 0