get_user_model() more Django compliant

diff --git a/social_auth/models.py b/social_auth/models.py
index 1bac301fb1e55176c4a4f67cd80b6d913b3e4539..78b797ea1d25773f64ad8216b98e05bd940dd367 100644 (file)
--- a/social_auth/models.py
+++ b/social_auth/models.py
@@ -3,9 +3,8 @@ from django.db import models
 from django.conf import settings
 
 def get_user_model():
-   """Allow setting a custom (extended) user model"""
-   _from, _model = getattr(settings, 'SOCIAL_AUTH_MODEL_USER', 'django.contrib.auth.models.User').rsplit('.', 1)
-   return getattr(__import__(_from, globals(), locals(), [_model], -1), _model)
+    """Allow setting a custom (extended) user model"""
+    return models.get_model(*getattr(settings, 'SOCIAL_AUTH_USER_MODEL', 'auth.User').split('.'))
 
 User = get_user_model()