From 07b27a4b7edf92b7da200531b08d396fef5442c0 Mon Sep 17 00:00:00 2001 From: Nicholas Serra Date: Fri, 1 Apr 2016 21:01:53 -0400 Subject: [PATCH] Be more verbose --- tos/compat.py | 18 ++++++++++++------ 1 file changed, 12 insertions(+), 6 deletions(-) diff --git a/tos/compat.py b/tos/compat.py index 9816c72..72f42f1 100644 --- a/tos/compat.py +++ b/tos/compat.py @@ -5,26 +5,32 @@ from django.conf import settings def get_fk_user_model(): if django.VERSION >= (1, 5): return settings.AUTH_USER_MODEL - return django.contrib.auth.models.User + from django.contrib.auth.models import User + return User def get_runtime_user_model(): if django.VERSION >= (1, 5): from django.contrib.auth import get_user_model return get_user_model() - return django.contrib.auth.models.User + from django.contrib.auth.models import User + return User def get_request_site(): if django.VERSION >= (1, 9): - return django.contrib.sites.requests.RequestSite - return django.contrib.sites.models.RequestSite + from django.contrib.sites.requests import RequestSite + else: + from django.contrib.sites.models import RequestSite + return RequestSite def get_library(): if django.VERSION >= (1, 9): - return django.template.library.Library - return django.template.base.Library + from django.template.library import Library + else: + from django.template.base import Library + return Library if django.VERSION < (1, 5):