harmacist/DashMachine
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

- fixes #163

Browse Source
This commit is contained in:
Ross Mountjoy 2020-05-11 06:18:16 -04:00
parent 3bd607a33d
commit f93b77d89b
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
dashmachine/__init__.py
View File

@ -1,5 +1,6 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
import os import os
import uuid
from flask import Flask from flask import Flask
from flask_caching import Cache from flask_caching import Cache
from flask_sqlalchemy import SQLAlchemy from flask_sqlalchemy import SQLAlchemy
@ -11,6 +12,16 @@ from dashmachine.paths import user_data_folder
if not os.path.isdir(user_data_folder): if not os.path.isdir(user_data_folder):
os.mkdir(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", "") context_path = os.getenv("CONTEXT_PATH", "")
app = Flask(__name__, static_url_path=context_path + "/static") app = Flask(__name__, static_url_path=context_path + "/static")
cache = Cache(app, config={"CACHE_TYPE": "simple"}) cache = Cache(app, config={"CACHE_TYPE": "simple"})