X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/2f7a76a00f2bcae2f67459b7fd86806278ef9b66..8315186b58e7fb5ff12472eff11fb6e0a789a51e:/models/nodes.py?ds=inline diff --git a/models/nodes.py b/models/nodes.py index bb1601e..1e7622a 100644 --- a/models/nodes.py +++ b/models/nodes.py @@ -4,7 +4,9 @@ from django.contrib.contenttypes import generic from django.contrib.sites.models import Site from django.http import HttpResponse, HttpResponseServerError, HttpResponseRedirect from django.core.servers.basehttp import FileWrapper -from philo.models.base import TreeEntity, Entity +from django.core.urlresolvers import resolve +from inspect import getargspec +from philo.models.base import TreeEntity, Entity, QuerySetMapper from philo.utils import ContentTypeSubclassLimiter from philo.validators import RedirectValidator @@ -21,8 +23,8 @@ class Node(TreeEntity): def accepts_subpath(self): return self.view.accepts_subpath - def render_to_response(self, request, path=None, subpath=None): - return self.view.render_to_response(self, request, path, subpath) + def render_to_response(self, request, path=None, subpath=None, extra_context=None): + return self.view.render_to_response(self, request, path, subpath, extra_context) class Meta: app_label = 'philo' @@ -37,12 +39,17 @@ class View(Entity): accepts_subpath = False - def render_to_response(self, node, request, path=None, subpath=None): + def attributes_with_node(self, node): + return QuerySetMapper(self.attribute_set, passthrough=node.attributes) + + def relationships_with_node(self, node): + return QuerySetMapper(self.relationship_set, passthrough=node.relationships) + + def render_to_response(self, node, request, path=None, subpath=None, extra_context=None): raise NotImplementedError('View subclasses must implement render_to_response.') class Meta: abstract = True - app_label = 'philo' _view_content_type_limiter.cls = View @@ -53,17 +60,22 @@ class MultiView(View): urlpatterns = [] - def render_to_response(self, node, request, path=None, subpath=None): + def render_to_response(self, node, request, path=None, subpath=None, extra_context=None): if not subpath: subpath = "" subpath = "/" + subpath - from django.core.urlresolvers import resolve view, args, kwargs = resolve(subpath, urlconf=self) + view_args = getargspec(view)[0] + if extra_context is not None and 'extra_context' in view_args: + if 'extra_context' in kwargs: + extra_context.update(kwargs['extra_context']) + kwargs['extra_context'] = extra_context + if 'node' in view_args: + kwargs['node'] = node return view(request, *args, **kwargs) class Meta: abstract = True - app_label = 'philo' class Redirect(View): @@ -74,7 +86,7 @@ class Redirect(View): target = models.CharField(max_length=200, validators=[RedirectValidator()]) status_code = models.IntegerField(choices=STATUS_CODES, default=302, verbose_name='redirect type') - def render_to_response(self, node, request, path=None, subpath=None): + def render_to_response(self, node, request, path=None, subpath=None, extra_context=None): response = HttpResponseRedirect(self.target) response.status_code = self.status_code return response @@ -89,11 +101,14 @@ class File(View): mimetype = models.CharField(max_length=255) file = models.FileField(upload_to='philo/files/%Y/%m/%d') - def render_to_response(self, node, request, path=None, subpath=None): + def render_to_response(self, node, request, path=None, subpath=None, extra_context=None): wrapper = FileWrapper(self.file) response = HttpResponse(wrapper, content_type=self.mimetype) response['Content-Length'] = self.file.size return response class Meta: - app_label = 'philo' \ No newline at end of file + app_label = 'philo' + + def __unicode__(self): + return self.file.name \ No newline at end of file