Merge pull request #179 from mhindery/code-cleanup

Code update and cleanup

Author: knaperek

.gitignore

@@ -4,3 +4,11 @@
 *.sqp
 build/
 dist/
+_build/
+.pytest_cache
+.env
+env/
+venv
+tags
+.idea/
+.vscode/

.hgignore

@@ -5,52 +5,31 @@ sudo: false
 
 matrix:
   include:
-    - python: 2.7
-      env: TOX_ENV=py27-django18
-    - python: 2.7
-      env: TOX_ENV=py27-django19
-    - python: 3.5
-      env: TOX_ENV=py35-django19
-    - python: 2.7
-      env: TOX_ENV=py27-django110
-    - python: 3.5
-      env: TOX_ENV=py35-django110
-    - python: 2.7
-      env: TOX_ENV=py27-django111
-    - python: 3.5
-      env: TOX_ENV=py35-django111
-    - python: 3.6
-      env: TOX_ENV=py36-django111
-    - python: 3.5
-      env: TOX_ENV=py35-django20
-    - python: 3.6
-      env: TOX_ENV=py36-django20
-    - python: 3.7
-      env: TOX_ENV=py37-django20
-    - python: 3.5
-      env: TOX_ENV=py35-django21
-    - python: 3.6
-      env: TOX_ENV=py36-django21
-    - python: 3.7
-      env: TOX_ENV=py37-django21
     - python: 3.5
       env: TOX_ENV=py35-django22
     - python: 3.6
       env: TOX_ENV=py36-django22
     - python: 3.7
       env: TOX_ENV=py37-django22
+    - python: 3.8
+      env: TOX_ENV=py38-django22
     - python: 3.6
       env: TOX_ENV=py36-django30
     - python: 3.7
       env: TOX_ENV=py37-django30
+    - python: 3.8
+      env: TOX_ENV=py38-django30
     - python: 3.6
       env: TOX_ENV=py36-djangomaster
     - python: 3.7
       env: TOX_ENV=py37-djangomaster
+    - python: 3.8
+      env: TOX_ENV=py38-djangomaster
   fast_finish: true
   allow_failures:
     - env: TOX_ENV=py36-djangomaster
     - env: TOX_ENV=py37-djangomaster
+    - env: TOX_ENV=py38-djangomaster
 
 addons:
   apt:

.travis.yml

@@ -0,0 +1 @@
+default_app_config = 'djangosaml2.apps.DjangoSaml2Config'

djangosaml2/__init__.py

@@ -3,7 +3,6 @@
 # This module defines a set of useful ACS failure functions that are used to
 # produce an output suitable for end user in case of SAML failure.
 #
-from __future__ import unicode_literals
 
 from django.core.exceptions import PermissionDenied
 from django.shortcuts import render

djangosaml2/acs_failures.py

@@ -0,0 +1,9 @@
+from django.apps import AppConfig
+
+
+class DjangoSaml2Config(AppConfig):
+    name = 'djangosaml2'
+    verbose_name = "DjangoSAML2"
+
+    def ready(self):
+        from . import signals  # noqa

djangosaml2/apps.py

@@ -18,45 +18,28 @@
 from django.conf import settings
 from django.contrib import auth
 from django.contrib.auth.backends import ModelBackend
-from django.core.exceptions import (
-    MultipleObjectsReturned, ImproperlyConfigured,
-)
-
-from djangosaml2.signals import pre_user_save
+from django.core.exceptions import (ImproperlyConfigured,
+                                    MultipleObjectsReturned)
 
+from .signals import pre_user_save
 
 logger = logging.getLogger('djangosaml2')
 
 
 def get_model(model_path):
+    from django.apps import apps
     try:
-        from django.apps import apps
         return apps.get_model(model_path)
-    except ImportError:
-        # Django < 1.7 (cannot use the new app loader)
-        from django.db.models import get_model as django_get_model
-        try:
-            app_label, model_name = model_path.split('.')
-        except ValueError:
-            raise ImproperlyConfigured("SAML_USER_MODEL must be of the form "
-                "'app_label.model_name'")
-        user_model = django_get_model(app_label, model_name)
-        if user_model is None:
-            raise ImproperlyConfigured("SAML_USER_MODEL refers to model '%s' "
-                "that has not been installed" % model_path)
-        return user_model
+    except LookupError:
+        raise ImproperlyConfigured("SAML_USER_MODEL refers to model '%s' that has not been installed" % model_path)
+    except ValueError:
+        raise ImproperlyConfigured("SAML_USER_MODEL must be of the form 'app_label.model_name'")
 
 
 def get_saml_user_model():
-    try:
-        # djangosaml2 custom user model
+    if hasattr(settings, 'SAML_USER_MODEL'):
         return get_model(settings.SAML_USER_MODEL)
-    except AttributeError:
-        try:
-            # Django 1.5 Custom user model
-            return auth.get_user_model()
-        except AttributeError:
-            return auth.models.User
+    return auth.get_user_model()
 
 
 class Saml2Backend(ModelBackend):

djangosaml2/backends.py

@@ -25,7 +25,7 @@ def __init__(self, django_session, key_suffix):
         self.session = django_session
         self.key = self.key_prefix + key_suffix
 
-        super(DjangoSessionCacheAdapter, self).__init__(self._get_objects())
+        super().__init__(self._get_objects())
 
     def _get_objects(self):
         return self.session.get(self.key, {})
@@ -37,9 +37,9 @@ def sync(self):
         # Changes in inner objects do not cause session invalidation
         # https://docs.djangoproject.com/en/1.9/topics/http/sessions/#when-sessions-are-saved
 
-        #add objects to session
+        # add objects to session
         self._set_objects(dict(self))
-        #invalidate session
+        # invalidate session
         self.session.modified = True
 
 
@@ -49,8 +49,7 @@ class OutstandingQueriesCache(object):
     """
 
     def __init__(self, django_session):
-        self._db = DjangoSessionCacheAdapter(django_session,
-                                             '_outstanding_queries')
+        self._db = DjangoSessionCacheAdapter(django_session, '_outstanding_queries')
 
     def outstanding_queries(self):
         return self._db._get_objects()
@@ -86,4 +85,4 @@ class StateCache(DjangoSessionCacheAdapter):
     """
 
     def __init__(self, django_session):
-        super(StateCache, self).__init__(django_session, '_state')
+        super().__init__(django_session, '_state')

djangosaml2/cache.py

@@ -18,10 +18,9 @@
 
 from django.conf import settings
 from django.core.exceptions import ImproperlyConfigured
-
 from saml2.config import SPConfig
 
-from djangosaml2.utils import get_custom_setting
+from .utils import get_custom_setting
 
 
 def get_config_loader(path, request=None):