add file upload to api

WebHostLib/check.py

@@ -28,7 +28,7 @@ def mysterycheck():
             if type(options) == str:
                 flash(options)
             else:
-                results, _ = roll_yamls(options)
+                results, _ = roll_options(options)
                 return render_template("checkresult.html", results=results)
 
     return render_template("check.html")
@@ -60,11 +60,14 @@ def get_yaml_data(file) -> Union[Dict[str, str], str]:
     return options
 
 
-def roll_yamls(options: Dict[str, Union[str, str]]) -> Tuple[Dict[str, Union[str, bool]], Dict[str, dict]]:
+def roll_options(options: Dict[str, Union[dict, str]]) -> Tuple[Dict[str, Union[str, bool]], Dict[str, dict]]:
     results = {}
     rolled_results = {}
     for filename, text in options.items():
         try:
+            if type(text) is dict:
+                yaml_data = text
+            else:
                 yaml_data = parse_yaml(text)
         except Exception as e:
             results[filename] = f"Failed to parse YAML data in {filename}: {e}"

WebHostLib/generate.py

@@ -13,7 +13,7 @@ import pickle
 
 from .models import *
 from WebHostLib import app
-from .check import get_yaml_data, roll_yamls
+from .check import get_yaml_data, roll_options
 
 
 @app.route('/generate', methods=['GET', 'POST'])
@@ -29,7 +29,7 @@ def generate(race=False):
             if type(options) == str:
                 flash(options)
             else:
-                results, gen_options = roll_yamls(options)
+                results, gen_options = roll_options(options)
                 if any(type(result) == str for result in results.values()):
                     return render_template("checkresult.html", results=results)
                 elif len(gen_options) > app.config["MAX_ROLL"]:
@@ -55,20 +55,28 @@ def generate(race=False):
 @app.route('/api/generate', methods=['POST'])
 def generate_api():
     try:
+        options = {}
+        if 'file' in request.files:
+            file = request.files['file']
+            options = get_yaml_data(file)
+
         json_data = request.get_json()
-        if json_data:
+        if 'weights' in json_data:
             # example: options = {"player1weights" : {<weightsdata>}}
             options = json_data["weights"]
+        if not options:
+            return {"text": "No options found. Expected file attachment or json weights."
+                    }, 400
+
         if len(options) > app.config["MAX_ROLL"]:
             return {"text": "Max size of multiworld exceeded",
                     "detail": app.config["MAX_ROLL"]}, 409
 
         race = bool(json_data["race"])
-            results, gen_options = roll_yamls(options)
+        results, gen_options = roll_options(options)
         if any(type(result) == str for result in results.values()):
             return {"text": str(results),
                     "detail": results}, 400
-
         else:
             gen = Generation(
                 options=pickle.dumps({name: vars(options) for name, options in gen_options.items()}),
@@ -81,8 +89,6 @@ def generate_api():
                     "encoded": app.url_map.converters["suuid"].to_url(gen.id),
                     "wait_api_url": url_for(wait_seed_api, seed=gen.id),
                     "url": url_for(wait_seed, seed=gen.id)}, 201
-        else:
-            return {"text": "POST data empty or incorrectly headered."}, 400
     except Exception as e:
         return {"text": "Uncaught Exception:" + str(e)}, 500
 
@@ -151,6 +157,7 @@ def wait_seed(seed: UUID):
         return "Generation failed, please retry."
     return render_template("wait_seed.html", seed_id=seed_id)
 
+
 @app.route('/api/status/<suuid:seed>')
 def wait_seed_api(seed: UUID):
     seed_id = seed
@@ -165,6 +172,7 @@ def wait_seed_api(seed: UUID):
         return {"text": "Generation failed"}, 500
     return {"text": "Generation running"}, 202
 
+
 def upload_to_db(folder, owner, sid):
     patches = set()
     spoiler = ""