feat: make all attrs configurable via flask config

giuppep · May 20, 2023 · e6ecaab · e6ecaab
1 parent a6d8286
commit e6ecaab
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/flask_pg_session/session.py b/flask_pg_session/session.py
@@ -24,6 +24,7 @@
 DEFAULT_KEY_PREFIX = ""
 DEFAULT_USE_SIGNER = False
 DELETE_EXPIRED_SESSIONS_EVERY_REQUESTS = 1000
+DEFAULT_PG_MAX_DB_CONN = 100
 
 
 # This is copied verbatim from flask-session
@@ -60,6 +61,12 @@ def init_app(cls, app: Flask) -> "FlaskPgSession":
             key_prefix=app.config.get("SESSION_KEY_PREFIX", DEFAULT_KEY_PREFIX),
             use_signer=app.config.get("SESSION_USE_SIGNER", DEFAULT_USE_SIGNER),
             permanent=app.config.get("SESSION_PERMANENT", True),
+            autodelete_expired_sessions=app.config.get(
+                "SESSION_AUTODELETE_EXPIRED", True
+            ),
+            max_db_conn=app.config.get(
+                "SESSION_PG_MAX_DB_CONN", DEFAULT_PG_MAX_DB_CONN
+            ),
         )
         app.session_interface = session_interface
         return session_interface