35 lines
1.0 KiB
Python
35 lines
1.0 KiB
Python
from flask import Blueprint, request, jsonify
|
|
from flask_apispec import use_kwargs
|
|
from marshmallow import fields
|
|
|
|
from api.exceptions import NotFoundException
|
|
from api.registration.models import Registration
|
|
from api.utils import docwrap
|
|
from api.extensions import rabbit
|
|
|
|
blueprint = Blueprint('webhook', __name__)
|
|
|
|
|
|
@docwrap('webhook', 'api_key')
|
|
@blueprint.route('/webhook', methods=['GET', 'POST'])
|
|
@use_kwargs({'apikey': fields.String(required=True)}, location='query')
|
|
def webhook(apikey):
|
|
reg: Registration = Registration.query.filter_by(token=apikey).first()
|
|
|
|
if reg is None:
|
|
raise NotFoundException(Registration.__name__)
|
|
|
|
if request.method == 'GET':
|
|
message = request.args.to_dict()
|
|
message.pop('apikey', None)
|
|
else:
|
|
message = request.get_json()
|
|
|
|
for target in reg.targets:
|
|
if target.exchange.name == '':
|
|
rabbit.queue_publish(target.name, message)
|
|
else:
|
|
rabbit.exchange_publish(target.exchange.name, target.name, message)
|
|
|
|
return None, 204
|