From f93b77d89b155fcf43a521c83fb9474521e03376 Mon Sep 17 00:00:00 2001 From: Ross Mountjoy Date: Mon, 11 May 2020 06:18:16 -0400 Subject: [PATCH] - fixes #163 --- dashmachine/__init__.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/dashmachine/__init__.py b/dashmachine/__init__.py index 564fe2b..6be3e7e 100755 --- a/dashmachine/__init__.py +++ b/dashmachine/__init__.py @@ -1,5 +1,6 @@ #!/usr/bin/env python3 import os +import uuid from flask import Flask from flask_caching import Cache from flask_sqlalchemy import SQLAlchemy @@ -11,6 +12,16 @@ from dashmachine.paths import user_data_folder if not os.path.isdir(user_data_folder): os.mkdir(user_data_folder) +secret_file = os.path.join(user_data_folder, ".secret") +if not os.path.isfile(secret_file): + with open(secret_file, "w") as new_file: + new_file.write(uuid.uuid4().hex) + +with open(secret_file, "r") as secret_file: + secret_key = secret_file.read().encode("utf-8") + if len(secret_key) < 32: + secret_key = uuid.uuid4().hex + context_path = os.getenv("CONTEXT_PATH", "") app = Flask(__name__, static_url_path=context_path + "/static") cache = Cache(app, config={"CACHE_TYPE": "simple"})