applied fixes per review

1 files changed, 10 insertions, 10 deletions

diff --git a/django_openid_auth/auth.py b/django_openid_auth/auth.py
index 741fdf4..6a572b5 100644
--- a/django_openid_auth/auth.py
+++ b/django_openid_auth/auth.py
@@ -51,7 +51,7 @@ from django_openid_auth.exceptions import (
 )
 
 
-USER_MODEL = get_user_model()
+User = get_user_model()
 
 
 class OpenIDBackend:
@@ -63,8 +63,8 @@ class OpenIDBackend:
 
     def get_user(self, user_id):
         try:
-            return USER_MODEL.objects.get(pk=user_id)
-        except USER_MODEL.DoesNotExist:
+            return User.objects.get(pk=user_id)
+        except User.DoesNotExist:
             return None
 
     def authenticate(self, **kwargs):
@@ -207,8 +207,8 @@ class OpenIDBackend:
 
         # See if we already have this nickname assigned to a username
         try:
-            USER_MODEL.objects.get(username__exact=nickname)
-        except USER_MODEL.DoesNotExist:
+            User.objects.get(username__exact=nickname)
+        except User.DoesNotExist:
             # No conflict, we can use this nickname
             return nickname
 
@@ -240,7 +240,7 @@ class OpenIDBackend:
             pass
 
         if getattr(settings, 'OPENID_STRICT_USERNAMES', False):
-            if USER_MODEL.objects.filter(username__exact=nickname).count() > 0:
+            if User.objects.filter(username__exact=nickname).count() > 0:
                 raise DuplicateUsernameViolation(
                     "The username (%s) with which you tried to log in is "
                     "already in use for a different account." % nickname)
@@ -249,14 +249,14 @@ class OpenIDBackend:
         # checking for conflicts.  Start with number of existing users who's
         # username starts with this nickname to avoid having to iterate over
         # all of the existing ones.
-        i = USER_MODEL.objects.filter(username__startswith=nickname).count() + 1
+        i = User.objects.filter(username__startswith=nickname).count() + 1
         while True:
             username = nickname
             if i > 1:
                 username += str(i)
             try:
-                USER_MODEL.objects.get(username__exact=username)
-            except USER_MODEL.DoesNotExist:
+                User.objects.get(username__exact=username)
+            except User.DoesNotExist:
                 break
             i += 1
         return username
@@ -280,7 +280,7 @@ class OpenIDBackend:
         username = self._get_available_username(
             nickname, openid_response.identity_url)
 
-        user = USER_MODEL.objects.create_user(username, email, password=None)
+        user = User.objects.create_user(username, email, password=None)
         self.associate_openid(user, openid_response)
         self.update_user_details(user, details, openid_response)