models/nodes.py

   1 from django.db import models
   2 from django.contrib.sites.models import Site
   3 from django.http import HttpResponse, HttpResponseServerError, HttpResponseRedirect
   4 from django.core.servers.basehttp import FileWrapper
   5 from philo.models.base import InheritableTreeEntity
   6 from philo.validators import RedirectValidator
   7
   8
   9 class Node(InheritableTreeEntity):
  10         accepts_subpath = False
  11
  12         def render_to_response(self, request, path=None, subpath=None):
  13                 return HttpResponseServerError()
  14
  15         class Meta:
  16                 unique_together = (('parent', 'slug'),)
  17                 app_label = 'philo'
  18
  19
  20 # the following line enables the selection of a node as the root for a given django.contrib.sites Site object
  21 models.ForeignKey(Node, related_name='sites', null=True, blank=True).contribute_to_class(Site, 'root_node')
  22
  23
  24 class MultiNode(Node):
  25         accepts_subpath = True
  26
  27         urlpatterns = []
  28
  29         def render_to_response(self, request, path=None, subpath=None):
  30                 if not subpath:
  31                         subpath = ""
  32                 subpath = "/" + subpath
  33                 from django.core.urlresolvers import resolve
  34                 view, args, kwargs = resolve(subpath, urlconf=self)
  35                 return view(request, *args, **kwargs)
  36
  37         class Meta:
  38                 abstract = True
  39                 app_label = 'philo'
  40
  41
  42 class Redirect(Node):
  43         STATUS_CODES = (
  44                 (302, 'Temporary'),
  45                 (301, 'Permanent'),
  46         )
  47         target = models.CharField(max_length=200,validators=[RedirectValidator()])
  48         status_code = models.IntegerField(choices=STATUS_CODES, default=302, verbose_name='redirect type')
  49
  50         def render_to_response(self, request, path=None, subpath=None):
  51                 response = HttpResponseRedirect(self.target)
  52                 response.status_code = self.status_code
  53                 return response
  54
  55         class Meta:
  56                 app_label = 'philo'
  57
  58
  59 class File(Node):
  60         """ For storing arbitrary files """
  61         mimetype = models.CharField(max_length=255)
  62         file = models.FileField(upload_to='philo/files/%Y/%m/%d')
  63
  64         def render_to_response(self, request, path=None, subpath=None):
  65                 wrapper = FileWrapper(self.file)
  66                 response = HttpResponse(wrapper, content_type=self.mimetype)
  67                 response['Content-Length'] = self.file.size
  68                 return response
  69
  70         class Meta:
  71                 app_label = 'philo'