Move get_user method to specific models backends

diff --git a/social_auth/backends/__init__.py b/social_auth/backends/__init__.py
index 59c47517efa4712307390481adff4af4045db85e..bb6b3e6c8516d64beafe3660c862b795bf1b08f8 100644 (file)
--- a/social_auth/backends/__init__.py
+++ b/social_auth/backends/__init__.py
@@ -19,13 +19,13 @@ from openid.extensions import sreg, ax
 
 from oauth2 import Consumer as OAuthConsumer, Token, Request as OAuthRequest
 
-from django.db import models
 from django.contrib.auth import authenticate
 from django.contrib.auth.backends import ModelBackend
 from django.utils import simplejson
 from django.utils.importlib import import_module
 
 from social_auth.models import get_social_auth_for_user
+from social_auth.models import get_user
 from social_auth.utils import setting, log, model_to_ctype, ctype_to_model, \
                               clean_partial_pipeline
 from social_auth.store import DjangoOpenIDStore
@@ -36,11 +36,6 @@ from social_auth.backends.exceptions import StopPipeline, AuthException, \
 from social_auth.backends.utils import build_consumer_oauth_request
 
 
-def get_user_model():
-    from social_auth.models import User
-    return User
-
-
 # OpenID configuration
 OLD_AX_ATTRS = [
     ('http://schema.openid.net/contact/email', 'old_email'),
@@ -195,11 +190,7 @@ class SocialAuthBackend(ModelBackend):
         """
         Return user with given ID from the User model used by this backend
         """
-        user_cls = get_user_model()
-        try:
-            return user_cls.objects.get(id=user_id)
-        except user_cls.DoesNotExist:
-            return None
+        return get_user(user_id)
 
 
 class OAuthBackend(SocialAuthBackend):

diff --git a/social_auth/django_models.py b/social_auth/django_models.py
index 1fce960a7dbe90957f557b038df1f19031cbcd78..34f04f93943823dc12f27407cca26f9d41cb8997 100644 (file)
--- a/social_auth/django_models.py
+++ b/social_auth/django_models.py
@@ -45,6 +45,13 @@ def create_user(*args, **kwargs):
     return User.objects.create_user(*args, **kwargs)
 
 
+def get_user(user_id):
+    try:
+        return User.objects.get(id=user_id)
+    except User.DoesNotExist:
+        return None
+
+
 def get_social_auth_for_user(user):
     return user.social_auth.all()
 

diff --git a/social_auth/mongoengine_models.py b/social_auth/mongoengine_models.py
index d6b746d00b65da3b3ad99f65d81974fb62f12365..1f12b868c81ad07e9d550e2381e2f2d290959634 100644 (file)
--- a/social_auth/mongoengine_models.py
+++ b/social_auth/mongoengine_models.py
@@ -53,6 +53,13 @@ def create_user(*args, **kwargs):
     return User.objects.create(*args, **kwargs)
 
 
+def get_user(user_id):
+    try:
+        return User.objects.get(id=user_id)
+    except User.DoesNotExist:
+        return None
+
+
 def get_social_auth_for_user(user):
     return UserSocialAuth.objects(user=user)