Removed extraneous evaluations of TreeEntity.parent.
authorStephen Burrows <stephen.r.burrows@gmail.com>
Thu, 23 Jun 2011 19:22:48 +0000 (15:22 -0400)
committerStephen Burrows <stephen.r.burrows@gmail.com>
Thu, 23 Jun 2011 19:22:48 +0000 (15:22 -0400)
philo/models/base.py

index d1dc38a..2138381 100644 (file)
@@ -515,7 +515,7 @@ class TreeEntity(Entity, MPTTModel):
                
                """
                if mapper is None:
                
                """
                if mapper is None:
-                       if self.parent:
+                       if getattr(self, "%s_id" % self._mptt_meta.parent_attr):
                                mapper = TreeAttributeMapper
                        else:
                                mapper = AttributeMapper
                                mapper = TreeAttributeMapper
                        else:
                                mapper = AttributeMapper
@@ -542,7 +542,7 @@ class SlugTreeEntity(TreeEntity):
        path = property(get_path)
        
        def clean(self):
        path = property(get_path)
        
        def clean(self):
-               if self.parent is None:
+               if getattr(self, "%s_id" % self._mptt_meta.parent_attr) is None:
                        try:
                                self._default_manager.exclude(pk=self.pk).get(slug=self.slug, parent__isnull=True)
                        except self.DoesNotExist:
                        try:
                                self._default_manager.exclude(pk=self.pk).get(slug=self.slug, parent__isnull=True)
                        except self.DoesNotExist: