birbnetes/input-service
birbnetes/input-service
4
0
Fork 0
Code Pull Requests Releases Activity

remove so many transactions
All checks were successful
continuous-integration/drone/push Build is passing
Details

Browse Source
This commit is contained in:
Torma Kristóf 2020-10-20 00:15:37 +02:00
parent 5a64c72cc9
commit 5e29de0c44
Signed by: tormakris
GPG Key ID: DC83C4F2C41B1047
1 changed files with 45 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/resources.py
View File

@ -4,7 +4,6 @@ from xeger import Xeger
from flask_restful import Resource from flask_restful import Resource
from flask import request, current_app from flask import request, current_app
import requests import requests
from sentry_sdk import start_transaction
import pika import pika
from db import db from db import db
from models import SampleMetadata from models import SampleMetadata
@ -35,32 +34,27 @@ class SampleResource(Resource):
Post request send to the endpoint Post request send to the endpoint
:return: :return:
""" """
with start_transaction(op="get-file", name="get-file-from-request"):
if 'file' not in request.files: if 'file' not in request.files:
return {"err_msg": "no file found"}, 469 return {"err_msg": "no file found"}, 469
else: else:
soundfile = request.files['file'] soundfile = request.files['file']
with start_transaction(op="get-description", name="get-description-from-request"):
if 'description' not in request.form: if 'description' not in request.form:
return {"err_msg": "no description found"}, 470 return {"err_msg": "no description found"}, 470
else: else:
description = request.form.get("description") description = request.form.get("description")
with start_transaction(op="validate-wave", name="validate-wave-type"):
if soundfile.content_type != 'audio/wave': if soundfile.content_type != 'audio/wave':
current_app.logger.info( current_app.logger.info(
f"Input file was not WAV.") f"Input file was not WAV.")
return {'err_msg': 'Input file not a wave file.'}, 415 return {'err_msg': 'Input file not a wave file.'}, 415
with start_transaction(op="validate-description", name="validate-description-schema"):
try: try:
desc = self.sampleschema.loads(description) desc = self.sampleschema.loads(description)
except Exception as e: except Exception as e:
current_app.logger.exception(e) current_app.logger.exception(e)
return {'err_msg': 'Input JSON schema invalid'}, 417 return {'err_msg': 'Input JSON schema invalid'}, 417
with start_transaction(op="tag-generate", name="generate-tag"):
xeger = Xeger(limit=30) xeger = Xeger(limit=30)
while True: while True:
generated_tag = xeger.xeger(r'^[a-zA-Z]+[0-9a-zA-Z_]*$')[:32] generated_tag = xeger.xeger(r'^[a-zA-Z]+[0-9a-zA-Z_]*$')[:32]
@ -82,7 +76,6 @@ class SampleResource(Resource):
soundfile, soundfile,
soundfile.content_type, soundfile.content_type,
{'Content-Length': soundfile.content_length})}).raise_for_status() {'Content-Length': soundfile.content_length})}).raise_for_status()
with start_transaction(op="rabbitmq-send", name="rabbitmq-connect-and-publish"):
credentials = pika.PlainCredentials(current_app.config['FLASK_PIKA_PARAMS']['username'], credentials = pika.PlainCredentials(current_app.config['FLASK_PIKA_PARAMS']['username'],
current_app.config['FLASK_PIKA_PARAMS']['password']) current_app.config['FLASK_PIKA_PARAMS']['password'])
connection = pika.BlockingConnection( connection = pika.BlockingConnection(
@ -97,6 +90,7 @@ class SampleResource(Resource):
routing_key='feature', routing_key='feature',
body=json.dumps({'tag': generated_tag}).encode('UTF-8')) body=json.dumps({'tag': generated_tag}).encode('UTF-8'))
connection.close() connection.close()
except Exception as e: except Exception as e:
current_app.logger.exception(e) current_app.logger.exception(e)
db.session.rollback() db.session.rollback()
@ -106,6 +100,7 @@ class SampleResource(Resource):
db.session.commit() db.session.commit()
return {"tag": generated_tag}, 200 return {"tag": generated_tag}, 200
def get(self): def get(self):
""" """
Get all stored items Get all stored items