Codestyle adjust in sessions.py

Author: insspb

flask_mongoengine/sessions.py

@@ -1,4 +1,4 @@
-import datetime
+from datetime import timedelta, datetime
 import uuid
 
 from bson.tz_util import utc
@@ -49,12 +49,11 @@ class DBSession(db.Document):
 
         self.cls = DBSession
 
-    def get_expiration_time(self, app, session):
+    def get_expiration_time(self, app, session) -> timedelta:
         if session.permanent:
             return app.permanent_session_lifetime
-        if "SESSION_TTL" in app.config:
-            return datetime.timedelta(**app.config["SESSION_TTL"])
-        return datetime.timedelta(days=1)
+        # Fallback to 1 day session ttl, if SESSION_TTL not set.
+        return timedelta(**app.config.get("SESSION_TTL", {"days": 1}))
 
     def open_session(self, app, request):
         sid = request.cookies.get(app.session_cookie_name)
@@ -67,7 +66,7 @@ def open_session(self, app, request):
                 if not expiration.tzinfo:
                     expiration = expiration.replace(tzinfo=utc)
 
-                if expiration > datetime.datetime.utcnow().replace(tzinfo=utc):
+                if expiration > datetime.utcnow().replace(tzinfo=utc):
                     return MongoEngineSession(
                         initial=stored_session.data, sid=stored_session.sid
                     )
@@ -85,9 +84,9 @@ def save_session(self, app, session, response):
                 response.delete_cookie(app.session_cookie_name, domain=domain)
             return
 
-        expiration = datetime.datetime.utcnow().replace(
-            tzinfo=utc
-        ) + self.get_expiration_time(app, session)
+        expiration = datetime.utcnow().replace(tzinfo=utc) + self.get_expiration_time(
+            app, session
+        )
 
         if session.modified:
             self.cls(sid=session.sid, data=session, expiration=expiration).save()