From 38cd770c9b4a234116b6134f844c6a262c6d9a01 Mon Sep 17 00:00:00 2001 From: Andrey Antukh Date: Thu, 5 Jun 2014 12:52:47 +0200 Subject: [PATCH] Cosmetic fixes on references services and user form. --- taiga/projects/references/services.py | 8 +++++--- taiga/users/forms.py | 8 ++++---- 2 files changed, 9 insertions(+), 7 deletions(-) diff --git a/taiga/projects/references/services.py b/taiga/projects/references/services.py index 197c0111..519116c4 100644 --- a/taiga/projects/references/services.py +++ b/taiga/projects/references/services.py @@ -1,9 +1,11 @@ -from .models import Reference +from django.db.models.loading import get_model + def get_instance_by_ref(project_id, obj_ref): + model_cls = get_model("references", "Reference") try: - instance = Reference.objects.get(project_id=project_id, ref=obj_ref) - except Reference.DoesNotExist: + instance = model_cls.objects.get(project_id=project_id, ref=obj_ref) + except model_cls.DoesNotExist: instance = None return instance diff --git a/taiga/users/forms.py b/taiga/users/forms.py index 3f987f7d..338eb410 100644 --- a/taiga/users/forms.py +++ b/taiga/users/forms.py @@ -15,10 +15,10 @@ # along with this program. If not, see . from django import forms -from django.contrib.auth.forms import ( - UserCreationForm as DjangoUserCreationForm, - UserChangeForm as DjangoUserChangeForm -) +from django.contrib.auth.forms import UserCreationForm as DjangoUserCreationForm +from django.contrib.auth.forms import UserChangeForm as DjangoUserChangeForm + + from .models import User