storage-service/storage_service/views/object_view.py

45 lines
1.6 KiB
Python
Raw Normal View History

2020-03-25 00:19:12 +01:00
#!/usr/bin/env python3
2020-03-25 04:02:26 +01:00
from flask import jsonify, request, abort, current_app
2020-03-25 00:19:12 +01:00
from flask_classful import FlaskView
2020-03-25 04:02:26 +01:00
from minio.error import BucketAlreadyExists, BucketAlreadyOwnedByYou, ResponseError
from marshmallow import ValidationError
2020-03-25 01:57:54 +01:00
from utils import storage
2020-03-25 04:02:26 +01:00
from schemas import DescriptionSchema
2020-03-25 00:19:12 +01:00
class ObjectView(FlaskView):
2020-03-25 04:02:26 +01:00
description_schema = DescriptionSchema(many=False)
2020-03-25 00:19:12 +01:00
def post(self):
2020-03-25 04:02:26 +01:00
# get important data from the request
try:
description = self.description_schema.loads(request.form.get('description'))
except ValidationError as e:
abort(400, str(e))
file = request.files['soundFile']
if file.content_type != 'audio/wave':
abort(400, f"{file.content_type} is not audio/wave")
if file.content_length <= 0:
abort(411, f"Content length for soundFile is not a positive integer or missing.")
# create bucket if necessary
try:
storage.connection.make_bucket(current_app.config['MINIO_BUCKET_NAME'])
except BucketAlreadyOwnedByYou as err:
pass
except BucketAlreadyExists as err:
pass
# Everything else should be raised
# poot file into bucket
try:
storage.connection.put_object(current_app.config['MINIO_BUCKET_NAME'], description['tag'], file, file.content_length, content_type=file.content_type)
except ResponseError: # TODO: Check if object already exists... somehow
raise
return jsonify({"status": "200"}), 200 # TODO: 200 should be OK but that would be inconsistent with the errors