models/nodes.py

   1 from django.db import models
   2 from django.contrib.contenttypes.models import ContentType
   3 from django.contrib.contenttypes import generic
   4 from django.contrib.sites.models import Site
   5 from django.http import HttpResponse, HttpResponseServerError, HttpResponseRedirect
   6 from django.core.servers.basehttp import FileWrapper
   7 from django.core.urlresolvers import resolve
   8 from inspect import getargspec
   9 from philo.models.base import TreeEntity, Entity, QuerySetMapper
  10 from philo.utils import ContentTypeSubclassLimiter
  11 from philo.validators import RedirectValidator
  12
  13
  14 _view_content_type_limiter = ContentTypeSubclassLimiter(None)
  15
  16
  17 class Node(TreeEntity):
  18         view_content_type = models.ForeignKey(ContentType, related_name='node_view_set', limit_choices_to=_view_content_type_limiter)
  19         view_object_id = models.PositiveIntegerField()
  20         view = generic.GenericForeignKey('view_content_type', 'view_object_id')
  21
  22         @property
  23         def accepts_subpath(self):
  24                 return self.view.accepts_subpath
  25
  26         def render_to_response(self, request, path=None, subpath=None, extra_context=None):
  27                 return self.view.render_to_response(self, request, path, subpath, extra_context)
  28
  29         class Meta:
  30                 app_label = 'philo'
  31
  32
  33 # the following line enables the selection of a node as the root for a given django.contrib.sites Site object
  34 models.ForeignKey(Node, related_name='sites', null=True, blank=True).contribute_to_class(Site, 'root_node')
  35
  36
  37 class View(Entity):
  38         nodes = generic.GenericRelation(Node, content_type_field='view_content_type', object_id_field='view_object_id')
  39
  40         accepts_subpath = False
  41
  42         def attributes_with_node(self, node):
  43                 return QuerySetMapper(self.attribute_set, passthrough=node.attributes)
  44
  45         def relationships_with_node(self, node):
  46                 return QuerySetMapper(self.relationship_set, passthrough=node.relationships)
  47
  48         def render_to_response(self, node, request, path=None, subpath=None, extra_context=None):
  49                 raise NotImplementedError('View subclasses must implement render_to_response.')
  50
  51         class Meta:
  52                 abstract = True
  53
  54
  55 _view_content_type_limiter.cls = View
  56
  57
  58 class MultiView(View):
  59         accepts_subpath = True
  60
  61         urlpatterns = []
  62
  63         def render_to_response(self, node, request, path=None, subpath=None, extra_context=None):
  64                 if not subpath:
  65                         subpath = ""
  66                 subpath = "/" + subpath
  67                 view, args, kwargs = resolve(subpath, urlconf=self)
  68                 view_args = getargspec(view)[0]
  69                 if extra_context is not None and 'extra_context' in view_args:
  70                         if 'extra_context' in kwargs:
  71                                 extra_context.update(kwargs['extra_context'])
  72                         kwargs['extra_context'] = extra_context
  73                 if 'node' in view_args:
  74                         kwargs['node'] = node
  75                 return view(request, *args, **kwargs)
  76
  77         class Meta:
  78                 abstract = True
  79
  80
  81 class Redirect(View):
  82         STATUS_CODES = (
  83                 (302, 'Temporary'),
  84                 (301, 'Permanent'),
  85         )
  86         target = models.CharField(max_length=200, validators=[RedirectValidator()])
  87         status_code = models.IntegerField(choices=STATUS_CODES, default=302, verbose_name='redirect type')
  88
  89         def render_to_response(self, node, request, path=None, subpath=None, extra_context=None):
  90                 response = HttpResponseRedirect(self.target)
  91                 response.status_code = self.status_code
  92                 return response
  93
  94         class Meta:
  95                 app_label = 'philo'
  96
  97
  98 class File(View):
  99         """ For storing arbitrary files """
 100
 101         mimetype = models.CharField(max_length=255)
 102         file = models.FileField(upload_to='philo/files/%Y/%m/%d')
 103
 104         def render_to_response(self, node, request, path=None, subpath=None, extra_context=None):
 105                 wrapper = FileWrapper(self.file)
 106                 response = HttpResponse(wrapper, content_type=self.mimetype)
 107                 response['Content-Length'] = self.file.size
 108                 return response
 109
 110         class Meta:
 111                 app_label = 'philo'