X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/47e6d542650e61e998e8b8d2195755c3eeec2037..da63b5b1fdcd03f375ff6b40a1d022d4e8778801:/models.py diff --git a/models.py b/models.py index 39f166c..6be9951 100644 --- a/models.py +++ b/models.py @@ -19,6 +19,7 @@ from django.template.loader import get_template from django.http import Http404, HttpResponse, HttpResponseServerError, HttpResponseRedirect from django.core.servers.basehttp import FileWrapper from django.conf import settings +from philo.validators import URLRedirectValidator def register_value_model(model): @@ -213,7 +214,10 @@ class InheritableTreeEntity(TreeEntity): @property def instance(self): - return self.instance_type.get_object_for_this_type(id=self.id) + try: + return self.instance_type.get_object_for_this_type(id=self.id) + except: + return None def get_path(self, pathsep='/', field='slug'): path = getattr(self.instance, field, '?') @@ -272,13 +276,16 @@ class Redirect(Node): (302, 'Temporary'), (301, 'Permanent'), ) - target = models.URLField(help_text='Must be a valid, absolute URL (i.e. http://)') + target = models.CharField(max_length=200,validators=[URLRedirectValidator()]) status_code = models.IntegerField(choices=STATUS_CODES, default=302, verbose_name='redirect type') def render_to_response(self, request, path=None, subpath=None): response = HttpResponseRedirect(self.target) response.status_code = self.status_code return response + + def __unicode__(self): + return self.slug class File(Node): @@ -291,9 +298,6 @@ class File(Node): response = HttpResponse(wrapper, content_type=self.mimetype) response['Content-Length'] = self.file.size return response - - def __unicode__(self): - return self.file class Template(TreeModel): @@ -399,7 +403,7 @@ class ContentReference(models.Model): page = models.ForeignKey(Page, related_name='contentreferences') name = models.CharField(max_length=255) content_type = models.ForeignKey(ContentType, verbose_name='Content type') - content_id = models.PositiveIntegerField(verbose_name='Content ID') + content_id = models.PositiveIntegerField(verbose_name='Content ID', blank=True, null=True) content = generic.GenericForeignKey('content_type', 'content_id') def __unicode__(self):