X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/cc6dbd55a2af538657c288a709662b9c84949854..5adae760fc382cd1f3fdc7ee02f2b8cc0dc5cfa6:/philo/models/nodes.py

diff --git a/philo/models/nodes.py b/philo/models/nodes.py
index 87ccb40..0c89095 100644
--- a/philo/models/nodes.py
+++ b/philo/models/nodes.py
@@ -1,4 +1,5 @@
 from inspect import getargspec
+import mimetypes
 
 from django.contrib.contenttypes import generic
 from django.contrib.contenttypes.models import ContentType
@@ -11,7 +12,7 @@ from django.http import HttpResponse, HttpResponseServerError, HttpResponseRedir
 from django.utils.encoding import smart_str
 
 from philo.exceptions import MIDDLEWARE_NOT_CONFIGURED, ViewCanNotProvideSubpath, ViewDoesNotProvideSubpaths
-from philo.models.base import TreeEntity, Entity, register_value_model
+from philo.models.base import SlugTreeEntity, Entity, register_value_model
 from philo.models.fields import JSONField
 from philo.utils import ContentTypeSubclassLimiter
 from philo.utils.entities import LazyPassthroughAttributeMapper
@@ -24,7 +25,7 @@ __all__ = ('Node', 'View', 'MultiView', 'Redirect', 'File')
 _view_content_type_limiter = ContentTypeSubclassLimiter(None)
 
 
-class Node(TreeEntity):
+class Node(SlugTreeEntity):
 	"""
 	:class:`Node`\ s are the basic building blocks of a website using Philo. They define the URL hierarchy and connect each URL to a :class:`View` subclass instance which is used to generate an HttpResponse.
 	
@@ -104,7 +105,7 @@ class Node(TreeEntity):
 		
 		return '%s%s%s%s' % (domain, root_url, path, subpath)
 	
-	class Meta:
+	class Meta(SlugTreeEntity.Meta):
 		app_label = 'philo'
 
 
@@ -235,7 +236,7 @@ class MultiView(View):
 		
 		"""
 		clear_url_caches()
-		subpath = request.node.subpath
+		subpath = request.node._subpath
 		view, args, kwargs = resolve(subpath, urlconf=self)
 		view_args = getargspec(view)
 		if extra_context is not None and ('extra_context' in view_args[0] or view_args[2] is not None):
@@ -370,11 +371,19 @@ class Redirect(TargetURLModel, View):
 
 class File(View):
 	"""Stores an arbitrary file."""
-	#: Defines the mimetype of the uploaded file. This will not be validated.
-	mimetype = models.CharField(max_length=255)
+	#: The name of the uploaded file. This is meant for finding the file again later, not for display.
+	name = models.CharField(max_length=255)
+	#: Defines the mimetype of the uploaded file. This will not be validated. If no mimetype is provided, it will be automatically generated based on the filename.
+	mimetype = models.CharField(max_length=255, blank=True)
 	#: Contains the uploaded file. Files are uploaded to ``philo/files/%Y/%m/%d``.
 	file = models.FileField(upload_to='philo/files/%Y/%m/%d')
 	
+	def clean(self):
+		if not self.mimetype:
+			self.mimetype = mimetypes.guess_type(self.file.name, strict=False)[0]
+			if self.mimetype is None:
+				raise ValidationError("Unknown file type.")
+	
 	def actually_render_to_response(self, request, extra_context=None):
 		wrapper = FileWrapper(self.file)
 		response = HttpResponse(wrapper, content_type=self.mimetype)
@@ -385,8 +394,8 @@ class File(View):
 		app_label = 'philo'
 	
 	def __unicode__(self):
-		"""Returns the path of the uploaded file."""
-		return self.file.name
+		"""Returns the value of :attr:`File.name`."""
+		return self.name
 
 
 register_value_model(Node)
\ No newline at end of file