79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
import os
|
|
import pika
|
|
import json
|
|
import requests
|
|
|
|
from components.video import process_full_video
|
|
|
|
RABBITMQ_HOST = os.environ.get('RABBITMQ_HOST', 'rabbitmq')
|
|
RABBITMQ_PORT = int(os.environ.get('RABBITMQ_PORT', 5672))
|
|
RABBITMQ_USER = os.environ.get('RABBITMQ_USER', 'admin')
|
|
RABBITMQ_PASS = os.environ.get('RABBITMQ_PASS')
|
|
RABBITMQ_QUEUE = os.environ.get('RABBITMQ_QUEUE', 'to-render')
|
|
|
|
if not RABBITMQ_PASS:
|
|
raise RuntimeError("RABBITMQ_PASS não definido no ambiente")
|
|
|
|
def callback(ch, method, properties, body):
|
|
try:
|
|
data = json.loads(body)
|
|
filename = data.get("filename")
|
|
times = data.get("times", [])
|
|
webhook_url = data.get("webhookUrl")
|
|
url = data.get("url")
|
|
video_id = data.get("videoId")
|
|
|
|
print(f"Processando vídeo: {filename}")
|
|
|
|
processed_files = process_full_video(filename, times)
|
|
|
|
payload = {
|
|
"videosProcessedQuantity": len(processed_files),
|
|
"filename": filename,
|
|
"processedFiles": processed_files,
|
|
"url": url,
|
|
"videoId": video_id,
|
|
"error": False,
|
|
}
|
|
except Exception as e:
|
|
payload = {
|
|
"videosProcessedQuantity": 0,
|
|
"filename": filename if 'filename' in locals() else None,
|
|
"processedFiles": [],
|
|
"url": url if 'url' in locals() else None,
|
|
"videoId": video_id if 'video_id' in locals() else None,
|
|
"error": str(e),
|
|
}
|
|
print(f"Erro no processamento: {e}")
|
|
|
|
try:
|
|
if webhook_url:
|
|
resp = requests.post(webhook_url, json=payload, timeout=30)
|
|
print(f"Webhook status: {resp.status_code}")
|
|
except Exception as wh_err:
|
|
print(f"Erro ao chamar webhook: {wh_err}")
|
|
|
|
ch.basic_ack(delivery_tag=method.delivery_tag)
|
|
|
|
def main():
|
|
credentials = pika.PlainCredentials(RABBITMQ_USER, RABBITMQ_PASS)
|
|
parameters = pika.ConnectionParameters(
|
|
host=RABBITMQ_HOST,
|
|
port=RABBITMQ_PORT,
|
|
credentials=credentials,
|
|
heartbeat=600,
|
|
blocked_connection_timeout=300
|
|
)
|
|
connection = pika.BlockingConnection(parameters)
|
|
channel = connection.channel()
|
|
channel.queue_declare(queue=RABBITMQ_QUEUE, durable=True)
|
|
channel.basic_qos(prefetch_count=1)
|
|
channel.basic_consume(queue=RABBITMQ_QUEUE, on_message_callback=callback)
|
|
|
|
print(' [*] Esperando mensagens. Para sair: CTRL+C')
|
|
|
|
channel.start_consuming()
|
|
|
|
if __name__ == "__main__":
|
|
main()
|