From a13075486deed9e14b7979542303aa9dcc255be0 Mon Sep 17 00:00:00 2001 From: Petros Moisiadis Date: Tue, 19 May 2015 17:57:27 +0300 Subject: [PATCH] Compatibility code for getting user model Compatibility code for getting user model --- rest_framework/compat.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/rest_framework/compat.py b/rest_framework/compat.py index c6a4a8698..1ba907314 100644 --- a/rest_framework/compat.py +++ b/rest_framework/compat.py @@ -119,6 +119,14 @@ def get_model_name(model_cls): return model_cls._meta.module_name +# Support custom user models in Django 1.5+ +try: + from django.contrib.auth import get_user_model +except ImportError: + from django.contrib.auth.models import User + get_user_model = lambda: User + + # View._allowed_methods only present from 1.5 onwards if django.VERSION >= (1, 5): from django.views.generic import View