Merge pull request #152 from oasiswork/fix-custom-model-appregistrynotready

jpadilla · jpadilla · commit 8e08e1a4bd7d · 2015-08-22T09:13:55.000-04:00
Only get_user_model at runtime, avoiding AppRegistryNotReady error w. custom models
diff --git a/rest_framework_jwt/authentication.py b/rest_framework_jwt/authentication.py
@@ -10,7 +10,6 @@
 from rest_framework_jwt.settings import api_settings
 
 
-User = get_user_model()
 jwt_decode_handler = api_settings.JWT_DECODE_HANDLER
 jwt_get_username_from_payload = api_settings.JWT_PAYLOAD_GET_USERNAME_HANDLER
 
@@ -48,6 +47,7 @@ def authenticate_credentials(self, payload):
         """
         Returns an active user that matches the payload's user id and email.
         """
+        User = get_user_model()
         username = jwt_get_username_from_payload(payload)
 
         if not username:
