X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/21abe382ad825e1383dfb1299efc1201ddd1e1c0..3a3123ecda265f9fa1becea878427d4d20a54f19:/philo/models/base.py

diff --git a/philo/models/base.py b/philo/models/base.py
index 0218261..86569b7 100644
--- a/philo/models/base.py
+++ b/philo/models/base.py
@@ -1,7 +1,7 @@
 from django import forms
 from django.contrib.contenttypes.models import ContentType
 from django.contrib.contenttypes import generic
-from django.core.exceptions import ObjectDoesNotExist
+from django.core.exceptions import ValidationError
 from django.core.validators import RegexValidator
 from django.db import models
 from django.utils import simplejson as json
@@ -16,7 +16,7 @@ from philo.utils.entities import AttributeMapper, TreeAttributeMapper
 from philo.validators import json_validator
 
 
-__all__ = ('Tag', 'value_content_type_limiter', 'register_value_model', 'unregister_value_model', 'JSONValue', 'ForeignKeyValue', 'ManyToManyValue', 'Attribute', 'Entity', 'TreeEntity')
+__all__ = ('Tag', 'value_content_type_limiter', 'register_value_model', 'unregister_value_model', 'JSONValue', 'ForeignKeyValue', 'ManyToManyValue', 'Attribute', 'Entity', 'TreeEntity', 'SlugTreeEntity')
 
 
 class Tag(models.Model):
@@ -522,6 +522,15 @@ class SlugTreeEntity(TreeEntity):
 		return super(SlugTreeEntity, self).get_path(root, pathsep, field)
 	path = property(get_path)
 	
+	def clean(self):
+		if self.parent is None:
+			try:
+				self._default_manager.exclude(pk=self.pk).get(slug=self.slug, parent__isnull=True)
+			except self.DoesNotExist:
+				pass
+			else:
+				raise ValidationError(self.unique_error_message(self.__class__, ('parent', 'slug')))
+	
 	class Meta:
 		unique_together = ('parent', 'slug')
 		abstract = True
\ No newline at end of file