19 lines
392 B
Python
19 lines
392 B
Python
|
import uuid
|
||
|
|
|
||
|
|
import redis
|
||
|
|
from rq import Queue
|
||
|
|
|
||
|
|
from .config import settings
|
||
|
|
|
||
|
|
redis_conn = redis.from_url(settings.redis_url)
|
||
|
|
generation_queue = Queue("generation", connection=redis_conn)
|
||
|
|
|
||
|
|
|
||
|
|
def enqueue_generation(song_id: uuid.UUID) -> None:
|
||
|
|
generation_queue.enqueue(
|
||
|
|
"worker.tasks.generate_song",
|
||
|
|
str(song_id),
|
||
|
|
job_timeout=60 * 20,
|
||
|
|
result_ttl=3600,
|
||
|
|
)
|