refactor code

OloruntobiOlurombi · Sep 12, 2024 · e3b5674 · e3b5674
1 parent b7a08d6
commit e3b5674
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/app.py b/app.py
@@ -1,13 +1,9 @@
-import os
 from flask import Flask
 from flask_wtf.csrf import CSRFProtect
+from config import Config
 
 app = Flask(__name__)
-
-secret_key = os.getenv('SECRET_KEY')
-if secret_key is None:
-    raise ValueError("No SECRET_KEY set for Flask application. Please set the SECRET_KEY environment variable.")
-app.config['SECRET_KEY'] = secret_key
+app.config.from_object(Config)
 
 csrf = CSRFProtect(app)
 

diff --git a/config.py b/config.py
@@ -0,0 +1,6 @@
+# config.py
+import os
+import secrets
+
+class Config:
+    SECRET_KEY = os.getenv('SECRET_KEY', secrets.token_urlsafe(32))