add proper rabbitMQ extension

This commit is contained in:
Gardient
2021-09-30 22:57:13 +03:00
parent b50c67ae1e
commit 71c1e322aa
4 changed files with 61 additions and 6 deletions

View File

@@ -5,7 +5,7 @@ from flask import current_app
from flask.cli import with_appcontext
from werkzeug.exceptions import MethodNotAllowed, NotFound
import rmq_helper
from .extensions import rabbit
from .target.models import Target
from .target_exchange.models import TargetExchange
@@ -100,9 +100,9 @@ def seed():
def setup_rabmq():
"""Set up rabbitMQ"""
for exchange in TargetExchange.query.filter(TargetExchange.name != "").all():
rmq_helper.ensure_exchange_exists(exchange.name)
rabbit.ensure_exchange_exists(exchange.name)
for target in Target.query.join(Target.exchange).filter(TargetExchange.name == "").all():
rmq_helper.ensure_queue_exists(target.routing_key)
rabbit.ensure_queue_exists(target.routing_key)
pass