From: Caio Ariede Date: Tue, 23 Nov 2010 01:26:51 +0000 (-0200) Subject: get_user_model() more Django compliant X-Git-Url: https://git.parisson.com/?a=commitdiff_plain;h=8db2b7087a22faab3ffdb352df4080d21a2b1b2e;p=django-social-auth.git get_user_model() more Django compliant --- diff --git a/social_auth/models.py b/social_auth/models.py index 1bac301..78b797e 100644 --- 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()