Some cleaning.

[wolnelektury.git] / apps / chunks / models.py

diff --git a/apps/chunks/models.py b/apps/chunks/models.py
index 396d221..5cdf8fe 100644 (file)
--- a/apps/chunks/models.py
+++ b/apps/chunks/models.py
@@ -1,5 +1,6 @@
+from django.core.cache import cache
 from django.db import models
-from django.utils.translation import ugettext_lazy as _
+from django.utils.translation import ugettext_lazy as _, get_language
 
 
 class Chunk(models.Model):
@@ -8,14 +9,35 @@ class Chunk(models.Model):
     any template with the use of a special template tag.
     """
     key = models.CharField(_('key'), help_text=_('A unique name for this chunk of content'), primary_key=True, max_length=255)
-    description = models.CharField(_('description'), blank=True, max_length=255)
-    content = models.TextField(_('content'), blank=True)
+    description = models.CharField(_('description'), blank=True, null=True, max_length=255)
+    content = models.TextField(_('content'), blank=True, null=True)
 
     class Meta:
         ordering = ('key',)
         verbose_name = _('chunk')
         verbose_name_plural = _('chunks')
-    
+
+    def __unicode__(self):
+        return self.key
+
+    @staticmethod
+    def cache_key(key):
+        return 'chunk/%s/%s' % (key, get_language())
+
+    def save(self, *args, **kwargs):
+        ret = super(Chunk, self).save(*args, **kwargs)
+        cache.delete(self.cache_key(self.key))
+        return ret
+
+
+class Attachment(models.Model):
+    key = models.CharField(_('key'), help_text=_('A unique name for this attachment'), primary_key=True, max_length=255)
+    attachment = models.FileField(upload_to='chunks/attachment')
+
+    class Meta:
+        ordering = ('key',)
+        verbose_name, verbose_name_plural = _('attachment'), _('attachments')
+
     def __unicode__(self):
-        return u'%s' % (self.key,)
+        return self.key