From 8db2b7087a22faab3ffdb352df4080d21a2b1b2e Mon Sep 17 00:00:00 2001 From: Caio Ariede Date: Mon, 22 Nov 2010 23:26:51 -0200 Subject: [PATCH] get_user_model() more Django compliant --- social_auth/models.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) 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() -- 2.39.5