Handle case where SignedCookieSessionFactory is not available.

Preston-Landers · Jul 24, 2017 · 46889a0 · 46889a0
1 parent e630a34
commit 46889a0
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/velruse/app/__init__.py b/velruse/app/__init__.py
@@ -80,8 +80,10 @@ def default_setup(config):
     specified then an in-memory storage backend will be used.
 
     """
-    # from pyramid.session import UnencryptedCookieSessionFactoryConfig
-    from pyramid.session import SignedCookieSessionFactory
+    try:
+        from pyramid.session import SignedCookieSessionFactory as SessionFactory
+    except ImportError:
+        from pyramid.session import UnencryptedCookieSessionFactoryConfig as SessionFactory
 
     log.info('Using an unencrypted cookie-based session. This can be '
              'changed by pointing the "velruse.setup" setting at a different '
@@ -102,8 +104,7 @@ def default_setup(config):
         else:
             secret = ''.join('%02x' % x for x in os.urandom(16))
         log.info('autogenerated session secret: %s', secret)
-    factory = SignedCookieSessionFactory(
-        secret, cookie_name=cookie_name)
+    factory = SessionFactory(secret, cookie_name=cookie_name)
     config.set_session_factory(factory)
 
     # setup backing storage