Stop ReST from dropping the primary title if it's the first title in the document.

[sommitrealweird.git] / sommitrealweird / bpcms / models.py

diff --git a/sommitrealweird/bpcms/models.py b/sommitrealweird/bpcms/models.py
index f3d9b8ae492a7da57e29cd387a7e2ed74616e3ee..ca3eb06a380326a3cda7142f9f0b9e2300cf3359 100644 (file)
--- a/sommitrealweird/bpcms/models.py
+++ b/sommitrealweird/bpcms/models.py
@@ -2,15 +2,28 @@ from django.db import models
 from django.conf import settings
 from django.contrib import admin
 from django.core.cache import cache
 from django.conf import settings
 from django.contrib import admin
 from django.core.cache import cache

+from django.core.urlresolvers import reverse

 
 FORMAT_CHOICES = (
         ('rst', 'reStructuredText'),
         ('html', 'HTML'),
     )
 
 
 FORMAT_CHOICES = (
         ('rst', 'reStructuredText'),
         ('html', 'HTML'),
     )
 

+class CSSDocument(models.Model):
+    title = models.CharField(max_length=150)
+    slug = models.SlugField()
+    content = models.TextField()
+
+    def __str__(self):
+        return self.__unicode__()
+
+    def __unicode__(self):
+        return u'%s' %(self.title,)
+

 class Document(models.Model):
     title = models.CharField(max_length=150)
     islive = models.BooleanField()
 class Document(models.Model):
     title = models.CharField(max_length=150)
     islive = models.BooleanField()

+    extracss = models.ManyToManyField(CSSDocument, null=True, blank=True)

     folder = models.ForeignKey('Folder', null=True, blank=True)
     format = models.CharField(max_length=10, choices=FORMAT_CHOICES)
     slug = models.SlugField()
     folder = models.ForeignKey('Folder', null=True, blank=True)
     format = models.CharField(max_length=10, choices=FORMAT_CHOICES)
     slug = models.SlugField()


@@ -20,7 +33,7 @@ class Document(models.Model):
         return self.__unicode__()
 
     def __unicode__(self):
         return self.__unicode__()
 
     def __unicode__(self):

-        return u'%s - %s' %(self.get_basic_url(), self.title)
+        return u'%s (%s)' %(self.title, self.slug)

 
     def save(self):
         super(Document, self).save()
 
     def save(self):
         super(Document, self).save()


@@ -35,19 +48,30 @@ class Document(models.Model):
                 doc.save()
 
     def get_basic_url(self):
                 doc.save()
 
     def get_basic_url(self):

-        folders = []
-        curfolder = self.folder
-        while curfolder != None:
-            folders.append(curfolder.slug)
-            curfolder = curfolder.parent
-
-        folders.reverse()
-        folderstring = "/".join(folders)
+        folderstring = self.get_folder_path()

         if folderstring != u'':
             folderstring = "%s/" %(folderstring,)
 
         return "%s%s%s/" %(settings.BPCMS_ROOT, folderstring, self.slug)
 
         if folderstring != u'':
             folderstring = "%s/" %(folderstring,)
 
         return "%s%s%s/" %(settings.BPCMS_ROOT, folderstring, self.slug)
 

+    def get_folder_path(self):
+        if not self.folder:
+            return ''
+        else:
+            return self.folder.get_path()
+
+    def get_absolute_url(self):
+        if self.slug == 'index':
+            if self.folder.parent:
+                return reverse('bpcms.views.document_view', kwargs={'folders': self.folder.parent.get_path(), 'slug': self.folder.slug})
+            else:
+                return reverse('bpcms.views.document_view', kwargs={'slug': self.folder.slug})
+        else:
+            if self.folder:
+                return reverse('bpcms.views.document_view', kwargs={'folders': self.get_folder_path(), 'slug': self.slug})
+            else:
+                return reverse('bpcms.views.document_view', kwargs={'slug': self.slug})
+

 class Folder(models.Model):
     title = models.CharField(max_length=150)
     slug = models.SlugField()
 class Folder(models.Model):
     title = models.CharField(max_length=150)
     slug = models.SlugField()


@@ -65,12 +89,9 @@ class Folder(models.Model):
         while curfolder != None:
             folders.append(curfolder.slug)
             curfolder = curfolder.parent
         while curfolder != None:
             folders.append(curfolder.slug)
             curfolder = curfolder.parent

-
+        

         folders.reverse()
         folderstring = "/".join(folders)
         folders.reverse()
         folderstring = "/".join(folders)

-        if folderstring != u'':
-            folderstring = "%s/" %(folderstring,)
-

         return folderstring
     
     def save(self, force_insert=False, force_update=False):
         return folderstring
     
     def save(self, force_insert=False, force_update=False):


@@ -79,7 +100,9 @@ class Folder(models.Model):
             cache.delete('bpcms_mainmenu')
         else:
             cache.delete('bpcms_submenu_%s' %(self.get_path()))
             cache.delete('bpcms_mainmenu')
         else:
             cache.delete('bpcms_submenu_%s' %(self.get_path()))

-        

 
 

-    def get_basic_url(self):
-        return "%s%s" %(settings.BPCMS_ROOT, self.get_path())
+    def get_absolute_url(self):
+        if self.parent:
+            return reverse('bpcms.views.document_view', kwargs={'folders': self.parent.get_path(), 'slug': self.slug})
+        else:
+            return reverse('bpcms.views.document_view', kwargs={'slug': self.slug})