morgan/Grinch-AP
1
0
Fork 0
mirror of https://github.com/MarioSpore/Grinch-AP.git synced 2025-10-21 12:11:33 -06:00
Code Issues Packages Projects Releases Wiki Activity

WebHost: On-Server rolling

Browse Source
This commit is contained in:
Fabian Dill
2020-08-02 22:11:52 +02:00
parent cfb8e2ce71
commit 22abd09087
13 changed files with 330 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

92
WebHostLib/check.py
View File

@@ -1,5 +1,5 @@
import zipfile
from typing import *
from flask import request, flash, redirect, url_for, session, render_template
@@ -11,9 +11,11 @@ banned_zip_contents = (".sfc",)
def allowed_file(filename):
return filename.endswith(('.txt', ".yaml", ".zip"))
from Mystery import roll_settings
from Utils import parse_yaml
@app.route('/mysterycheck', methods=['GET', 'POST'])
def mysterycheck():
if request.method == 'POST':
@@ -21,46 +23,56 @@ def mysterycheck():
if 'file' not in request.files:
flash('No file part')
else:
options = {}
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"):
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(".yaml"):
options[file.filename] = zfile.open(file, "r").read()
elif file.filename.endswith(".txt"):
options[file.filename] = zfile.open(file, "r").read()
else:
options = {file.filename: file.read()}
if not options:
flash("Did not find a .yaml file to process.")
else:
results = {}
for filename, text in options.items():
try:
yaml_data = parse_yaml(text)
except Exception as e:
results[filename] = f"Failed to parse YAML data in {filename}: {e}"
else:
try:
roll_settings(yaml_data)
except Exception as e:
results[filename] = f"Failed to generate mystery in {filename}: {e}"
else:
results[filename] = "Looks fine"
return render_template("checkresult.html", results=results)
options = get_yaml_data(file)
if type(options) == str:
flash(options)
else:
flash("Not recognized file format. Awaiting a .yaml file.")
results, _ = roll_yamls(options)
return render_template("checkresult.html", results=results)
return render_template("check.html")
def get_yaml_data(file) -> Union[Dict[str, str], str]:
options = {}
# if user does not select file, browser also
# submit an empty part without filename
if file.filename == '':
return 'No selected file'
elif file and allowed_file(file.filename):
if file.filename.endswith(".zip"):
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(".yaml"):
options[file.filename] = zfile.open(file, "r").read()
elif file.filename.endswith(".txt"):
options[file.filename] = zfile.open(file, "r").read()
else:
options = {file.filename: file.read()}
if not options:
return "Did not find a .yaml file to process."
return options
def roll_yamls(options: Dict[str, Union[str, str]]) -> Tuple[Dict[str, Union[str, bool]], Dict[str, dict]]:
results = {}
rolled_results = {}
for filename, text in options.items():
try:
yaml_data = parse_yaml(text)
except Exception as e:
results[filename] = f"Failed to parse YAML data in {filename}: {e}"
else:
try:
rolled_results[filename] = roll_settings(yaml_data)
except Exception as e:
results[filename] = f"Failed to generate mystery in {filename}: {e}"
else:
results[filename] = True
return results, rolled_results