WebHost: Move module into WebHostLib to prevent shadowing WebHost.py

WebHostLib/upload.py

@@ -0,0 +1,73 @@
+import json
+import zlib
+import zipfile
+import logging
+
+from flask import request, flash, redirect, url_for, session, render_template
+from pony.orm import commit, select
+
+from WebHostLib import app, allowed_file, Seed, Room, Patch
+
+accepted_zip_contents = {"patches": ".bmbp",
+                         "spoiler": ".txt",
+                         "multidata": "multidata"}
+
+banned_zip_contents = (".sfc",)
+
+
+@app.route('/uploads', methods=['GET', 'POST'])
+def uploads():
+    if request.method == 'POST':
+        # check if the post request has the file part
+        if 'file' not in request.files:
+            flash('No file part')
+        else:
+            file = request.files['file']
+            # if user does not select file, browser also
+            # submit an empty part without filename
+            if file.filename == '':
+                flash('No selected file')
+            elif file and allowed_file(file.filename):
+                if file.filename.endswith(".zip"):
+                    patches = set()
+                    spoiler = ""
+                    multidata = None
+                    with zipfile.ZipFile(file, 'r') as zfile:
+                        infolist = zfile.infolist()
+
+                        for file in infolist:
+                            if file.filename.endswith(banned_zip_contents):
+                                return "Uploaded data contained a rom file, which is likely to contain copyrighted material. Your file was deleted."
+                            elif file.filename.endswith(".bmbp"):
+                                player = int(file.filename.split("P")[-1].split(".")[0].split("_")[0])
+                                patches.add(Patch(data=zfile.open(file, "r").read(), player=player))
+                            elif file.filename.endswith(".txt"):
+                                spoiler = zfile.open(file, "rt").read().decode("utf-8-sig")
+                            elif file.filename.endswith("multidata"):
+                                try:
+                                    multidata = json.loads(zlib.decompress(zfile.open(file).read()).decode("utf-8-sig"))
+                                except:
+                                    flash("Could not load multidata. File may be corrupted or incompatible.")
+                        if multidata:
+                            commit()  # commit patches
+                            seed = Seed(multidata=multidata, spoiler=spoiler, patches=patches, owner=session["_id"])
+                            commit()  # create seed
+                            for patch in patches:
+                                patch.seed = seed
+
+                            return redirect(url_for("view_seed", seed=seed.id))
+                        else:
+                            flash("No multidata was found in the zip file, which is required.")
+                else:
+                    try:
+                        multidata = json.loads(zlib.decompress(file.read()).decode("utf-8-sig"))
+                    except:
+                        flash("Could not load multidata. File may be corrupted or incompatible.")
+                    else:
+                        seed = Seed(multidata=multidata, owner=session["_id"])
+                        commit()  # place into DB and generate ids
+                        return redirect(url_for("view_seed", seed=seed.id))
+            else:
+                flash("Not recognized file format. Awaiting a .multidata file.")
+    rooms = select(room for room in Room if room.owner == session["_id"])
+    return render_template("uploads.html", rooms=rooms)