Allow selecting version

master
kaiyou 7 years ago
parent 987cfde91f
commit dd8b0dba54

@ -21,6 +21,7 @@ def render_flavor(flavor, template, data):
) )
@app.add_template_global
def secret(length=16): def secret(length=16):
charset = string.ascii_uppercase + string.digits charset = string.ascii_uppercase + string.digits
return ''.join( return ''.join(
@ -29,40 +30,45 @@ def secret(length=16):
) )
def build_app(setup_path): def build_app(path):
versions = [
version for version in os.listdir(path)
if os.path.isdir(os.path.join(path, version))
]
app.jinja_env.trim_blocks = True app.jinja_env.trim_blocks = True
app.jinja_env.lstrip_blocks = True app.jinja_env.lstrip_blocks = True
app.jinja_env.globals.update(dict(
secret=secret
))
versions = [ @app.context_processor
path for path in os.listdir(setup_path) def app_context():
if os.path.isdir(os.path.join(setup_path, path)) return dict(versions=versions)
]
@app.route("/")
def index():
return flask.redirect(flask.url_for('{}.wizard'.format(versions[-1])))
for version in versions: for version in versions:
bp = flask.Blueprint(version, __name__) bp = flask.Blueprint(version, __name__)
template_dir = os.path.join(setup_path, version, "templates")
flavor_dir = os.path.join(setup_path, version, "flavors")
bp.jinja_loader = jinja2.ChoiceLoader([ bp.jinja_loader = jinja2.ChoiceLoader([
jinja2.FileSystemLoader(template_dir), jinja2.FileSystemLoader(os.path.join(path, version, "templates")),
jinja2.FileSystemLoader(flavor_dir) jinja2.FileSystemLoader(os.path.join(path, version, "flavors"))
]) ])
@bp.context_processor
def bp_context(version=version):
return dict(version=version)
@bp.route("/") @bp.route("/")
def wizard(): def wizard():
return flask.render_template('wizard.html') return flask.render_template('wizard.html')
@bp.route("/submit", methods=["POST"]) @bp.route("/submit", methods=["POST"])
def submit(): def submit():
uid = str(uuid.uuid4())
data = flask.request.form.copy() data = flask.request.form.copy()
data.update(dict(uid=uid, version=version)) data['uid'] = str(uuid.uuid4())
db.set(uid, json.dumps(data)) db.set(data['uid'], json.dumps(data))
return flask.redirect(flask.url_for('.setup', uid=uid)) return flask.redirect(flask.url_for('.setup', uid=data['uid']))
@bp.route("/setup/<uid>", methods=["GET"]) @bp.route("/setup/<uid>", methods=["GET"])
def setup(uid): def setup(uid):
@ -81,7 +87,6 @@ def build_app(setup_path):
) )
app.register_blueprint(bp, url_prefix="/{}".format(version)) app.register_blueprint(bp, url_prefix="/{}".format(version))
print("Serving version {}".format(version))
if __name__ == "__main__": if __name__ == "__main__":

Loading…
Cancel
Save