fixed admin user generation

Author: fumblehool

configs.py

@@ -37,6 +37,16 @@ def get_sys_exec_root_or_drive():
 if not all([algolia_app_id, algolia_api_key]):
     print("Application requires 'ALGOLIA_APP_ID' and 'ALGOLIA_API_KEY' for search")
 
+
+secret_key = os.environ.get('SECRET_KEY')
+
+security_password_hash = 'pbkdf2_sha512'
+# Replace this with your own salt.
+security_password_salt = os.environ.get('SECURITY_PASSWORD_SALT')
+if not all([secret_key, security_password_salt]):
+    raise KeyError("Keys'SECRET_KEY' and 'SECURITY_PASSWORD_SALT' missing")
+
+
 index_name = os.environ.get("INDEX_NAME")
 
 
@@ -45,12 +55,9 @@ class Config:
 
     SQLALCHEMY_DATABASE_URI = f"postgresql://{pg_user}:{pg_pw}@{pg_host}:5432/{pg_db}"
 
-    SECRET_KEY = os.urandom(24)
-    # Set config values for Flask-Security.
-    # We're using PBKDF2 with salt.
-    SECURITY_PASSWORD_HASH = 'pbkdf2_sha512'
-    # Replace this with your own salt.
-    SECURITY_PASSWORD_SALT = os.urandom(140)
+    SECRET_KEY = secret_key
+    SECURITY_PASSWORD_HASH = security_password_hash
+    SECURITY_PASSWORD_SALT = security_password_salt
 
     ALGOLIA_APP_ID = algolia_app_id
     ALGOLIA_API_KEY = algolia_api_key

run.py

@@ -27,6 +27,7 @@ def make_shell_context():
     return {'db': db, 'Resource': Resource, 'Category': Category,
             'Language': Language, 'User': User, 'Role': Role}
 
+
 # Create Admin user and role.
 @app.before_first_request
 def before_first_request():
@@ -44,7 +45,8 @@ def before_first_request():
     # In each case, use Flask-Security utility function to encrypt the password.
     encrypted_password = utils.encrypt_password(admin_password)
     if not user_datastore.get_user(admin_email):
-        user_datastore.create_user(admin_email, password=encrypted_password)
+        user_datastore.create_user(email=admin_email,
+                                   password=encrypted_password)
     # Add more users.
 
     # Commit any database changes; the User and Roles must exist before we