Added config and SECRET_KEY

Now each user will generate their own SECRET_KEY

Author: m-haisham

qrshare/app.py

@@ -7,11 +7,17 @@
 from .auth import Authentication
 from .models import Route, QRContainer, ZipContent
 from .tools import NetworkTools
+from .config import UserConfig
 
 
 class App:
     def __init__(self, paths: List[Path], code=None, port=5000):
         self.app = self.init()
+        self.user = UserConfig()
+
+        # set config
+        self.app.config.from_mapping(self.user.config.data)
+
         self.auth = Authentication(self.app, code)
 
         self.paths = paths

qrshare/config/__init__.py

@@ -0,0 +1 @@
+from .user import UserConfig

qrshare/config/base.py

@@ -0,0 +1,40 @@
+import json
+from pathlib import Path
+from typing import Optional, Any
+
+
+class ConfigBase:
+    def __init__(self, path, name, load=False):
+        self.path = path / Path(name)
+        self.name = name
+        self.data = {}
+
+        if load:
+            self.load()
+
+    def put(self, key, value):
+        self.data[key] = value
+        self.save()
+
+    def get(self, key) -> Optional[Any]:
+        try:
+            return self.data[key]
+        except KeyError:
+            return None
+
+    def save(self):
+        with self.path.open('w') as f:
+            json.dump(self.data, f)
+
+    def load(self):
+        try:
+            with self.path.open('r') as f:
+                self.data = json.load(f)
+
+        # edit error message
+        except json.JSONDecodeError as e:
+            raise json.JSONDecodeError(f'{self.path} - {e.msg}', e.doc, e.pos)
+
+        # treat as empty
+        except FileNotFoundError:
+            pass

qrshare/config/user.py

@@ -0,0 +1,22 @@
+from pathlib import Path
+from uuid import uuid4
+
+from appdirs import user_config_dir
+from .base import ConfigBase
+
+appname = 'Novelsave'
+appauthor = 'mHaisham'
+
+class UserConfig:
+    def __init__(self):
+        # get cross os configuration path
+        self.path = Path(user_config_dir(appname, appauthor))
+        self.path.mkdir(parents=True, exist_ok=True)
+
+        self.config = ConfigBase(self.path, 'config.json')
+        self.init_secret_key()
+
+    def init_secret_key(self):
+        key = self.config.get("SECRET_KEY")
+        if key is None:
+            self.config.put("SECRET_KEY", uuid4().get_hex())

requirements.txt

@@ -2,3 +2,4 @@ click==7.1.2
 Flask==1.1.2
 qrcode==6.1
 waitress==1.4.4
+appdirs>=1.4.4