App config

console

@@ -19,7 +19,7 @@
 import os
 import argparse
 
-from utils import env
+from utils import env, generate_app_key
 
 parser = argparse.ArgumentParser(description='Run the k8s.elections app')
 
@@ -34,6 +34,10 @@ parser.add_argument('--host',
                     server is available externally. Defaults to 127.0.0.1 \
                     making the it only visable on localhost')
 
+parser.add_argument('--generate',
+                    help='Generate 32 byte app secret key for session cookie encryption',
+                    action='store_true')
+
 parser.add_argument('--migrate',
                     help='Create the database tables',
                     action='store_true')
@@ -53,6 +57,11 @@ if __name__ == "__main__":
     # ###################################################################### #
     args = parser.parse_args()
 
+    if args.generate:
+        print('# ----- Generating 32 byte app secret key ----- #')
+        generate_app_key()
+        exit()
+
     if args.migrate:
         from config import DATABASE_URL
         from elekto.models.sql import migrate

utils.py

@@ -15,13 +15,18 @@
 # Author(s):         Manish Sahani <[email protected]>
 
 import os
-from dotenv import load_dotenv
+from dotenv import load_dotenv, set_key
 
 # Load the custom environment file into the program
 targetenv = '.env.testing' if os.getenv('TESTING') else '.env'
 load_dotenv(os.path.join(os.path.dirname(__file__), targetenv))
 
 
+def generate_app_key():
+    key = os.urandom(32).hex()
+    set_key(targetenv, "APP_KEY", key)
+
+
 def env(key, default=None):
     v = os.getenv(key)
     return default if v is None or v == '' else v