Implemented everything upload

src/views/uploadview.py

@@ -1,6 +1,17 @@
 #!/usr/bin/env python3
-from flask import render_template
+import tempfile
+import os
+import os.path
+import minio
+from flask import render_template, flash, redirect, url_for, request, current_app
 from flask_classful import FlaskView
+from flask_security import current_user, login_required
+from utils import storage
+import bleach
+from models import db, Item
+
+from utils import create_caff_preview
+from requests import HTTPError
 
 """
 Upload VIEW
@@ -17,6 +28,81 @@ class UploadView(FlaskView):
     def index(self):
         return render_template('upload.html')
 
-
+    @login_required
     def post(self):
-        pass
+        title = request.form.get('title')
+        title = title[:Item.name.property.columns[0].type.length]
+        title = bleach.clean(title, tags=[])
+
+        if not title:
+            flash("Title must be filled", "primary")
+            return redirect(url_for('UploadView:index'))
+
+        if 'file' not in request.files:
+            flash("No file provided", "primary")
+            return redirect(url_for('UploadView:index'))
+
+        file = request.files['file']
+        uploaded_caff_fd, uploaded_caff_path = tempfile.mkstemp(prefix=current_user.name, suffix='.caff')
+
+        with open(uploaded_caff_fd, "wb") as f:
+            file.save(f)
+
+        # let the memes begin! -----------------------------------------------------------------------------------------
+        success = False
+        try:
+            converted_png_path = create_caff_preview(uploaded_caff_path)
+            success = os.path.isfile(converted_png_path)  # check if the file really is there
+        except HTTPError as e:
+            if e.response.status_code == 400:
+                flash("Invalid CAFF file", "danger")
+            elif e.response.status_code == 413:
+                flash("CAFF file too large", "danger")
+            else:
+                raise  # Whatever... we'll just check the Sentry alert
+
+        except OverflowError:
+            flash("CAFF file too large", "danger")
+
+        except ValueError:
+            flash("Something went wrong, try again later...", "warning")
+
+        if not success:
+            os.unlink(uploaded_caff_path)
+            return redirect(url_for('UploadView:index'))
+
+        # End of the meme part -----------------------------------------------------------------------------------------
+
+        # Upload everything to minio
+        item = Item(name=title, uploader=current_user)
+        db.session.add(item)
+        db.session.flush()  # To obtain an id
+
+        try:
+            storage.connection.make_bucket(current_app.config['MINIO_PREVIEW_BUCKET_NAME'])
+        except minio.error.BucketAlreadyOwnedByYou:
+            pass
+
+        storage.connection.fput_object(
+            current_app.config['MINIO_PREVIEW_BUCKET_NAME'],
+            str(item.id),
+            converted_png_path,
+            content_type="image/png"
+        )
+
+        try:
+            storage.connection.make_bucket(current_app.config['MINIO_CAFF_BUCKET_NAME'])
+        except minio.error.BucketAlreadyOwnedByYou:
+            pass
+        storage.connection.fput_object(
+            current_app.config['MINIO_CAFF_BUCKET_NAME'],
+            str(item.id),
+            uploaded_caff_path
+        )
+
+        # Always clean up after ourselves
+        os.unlink(uploaded_caff_path)
+        os.unlink(converted_png_path)
+
+        db.session.commit()
+        return redirect(url_for('UploadView:index'))  # TODO: report item id