From 65f4452744cf16d1cac4c7127cbb5dae56e3b8a0 Mon Sep 17 00:00:00 2001 From: David Gelvin Date: Tue, 7 Sep 2010 11:23:25 +0000 Subject: [PATCH] Implemented remove_label --- models.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/models.py b/models.py index 2668490..5f3d927 100644 --- a/models.py +++ b/models.py @@ -65,7 +65,6 @@ class EavAttribute(models.Model): labels = models.ManyToManyField(EavAttributeLabel, verbose_name=_(u"labels")) - def save(self, *args, **kwargs): self.full_clean() super(EavAttribute, self).save(*args, **kwargs) @@ -74,9 +73,11 @@ class EavAttribute(models.Model): self.labels.get_or_create(name=label) def remove_label(self, label): - #TODO - self.labels.get_or_create(name=label) - + try: + label_obj = EavAttributeLabel.objects.get(name=label) + except EavAttributeLabel.DoesNotExist: + return + self.labels.remove(label_obj) def get_value_for_entity(self, entity): '''