From 6be5ff7b2abf1aa23c77e035ca30510f505bba57 Mon Sep 17 00:00:00 2001 From: Sharad Heft Date: Sun, 5 Nov 2017 21:40:44 +0100 Subject: [PATCH] Refactor getting of user model in auth backend. --- test_oauth/backend.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/test_oauth/backend.py b/test_oauth/backend.py index e5eb1ad..5349efd 100644 --- a/test_oauth/backend.py +++ b/test_oauth/backend.py @@ -5,15 +5,12 @@ from django.conf import settings from django.db import transaction -UserModel = get_user_model() - - class TESTOAuth2Backend(ModelBackend): def authenticate(self, id=None, username=None, **profile): if id is None or username is None: return None - user, _ = UserModel.objects.get_or_create( + user, _ = get_user_model().objects.get_or_create( pk=id, username=username ) @@ -24,6 +21,8 @@ class TESTOAuth2Backend(ModelBackend): return user def get_user(self, user_id): + UserModel = get_user_model() + try: return UserModel.objects.get(pk=user_id) except UserModel.DoesNotExist: -- GitLab