X-Git-Url: https://git.sommitrealweird.co.uk/sommitrealweird.git/blobdiff_plain/56b2cdf3569443cf4504fe085de9db0466dcd09f..5d2c22628bbba4eb7ea12a1f5e11bdcc6c866e03:/sommitrealweird/bpcms/context_processors.py

diff --git a/sommitrealweird/bpcms/context_processors.py b/sommitrealweird/bpcms/context_processors.py
index 90b98cc..2df38ae 100644
--- a/sommitrealweird/bpcms/context_processors.py
+++ b/sommitrealweird/bpcms/context_processors.py
@@ -1,33 +1,151 @@
 from bpcms.models import Document, Folder
-import operator
+from django.conf import settings
+from django.core.cache import cache
+import re
+import sys
 
 def content_menu(request):
-    items = Document.objects.filter(folder__isnull=True, islive__exact=True)
-    menu_items = []
-    for item in items:
-        if item.slug != 'index':
-            menu_items.append(
-                {
-                    'title': item.title,
-                    'url': item.get_basic_url(),
-                }
-            )
-    folders = Folder.objects.filter(parent__isnull=True)
-    for folder in folders:
-        try:
-            items = Document.objects.filter(folder__exact=folder, islive__exact=True, slug__exact='index')
-            item = items[0]
-            menu_items.append(
-                {
-                    'title': item.title,
-                    'url': folder.get_basic_url(),
-                }
-            )
-        except:
-            pass
 
-    menu_items.sort(cmp=lambda x,y: cmp(x['title'].lower(), y['title'].lower()))
+    print >>sys.stderr, "Trying to get menu items from cache"
+
+    menu_items = cache.get('bpcms_mainmenu')
+
+    if not menu_items:
+        print >>sys.stderr, "Generating menu items"
+        items = Document.objects.filter(folder__isnull=True, islive__exact=True)
+        menu_items = []
+        for item in items:
+            print >>sys.stderr, "Found: %s" %(item.slug)
+            if item.slug != 'index':
+                menu_items.append(
+                    {
+                        'title': item.title,
+                        'url': item.get_absolute_url(),
+                    }
+                )
+        print >>sys.stderr, "Now looking for folders"
+        folders = Folder.objects.filter(parent__isnull=True)
+        for folder in folders:
+            print >>sys.stderr, "Found: %s" %(folder)
+            print >>sys.stderr, "folder path: %s" %(folder.get_absolute_url())
+            try:
+                items = Document.objects.filter(folder__exact=folder, islive__exact=True, slug__exact='index')
+                item = items[0]
+                menu_items.append(
+                    {
+                        'title': item.title,
+                        'url': folder.get_absolute_url(),
+                    }
+                )
+            except:
+                print >>sys.stderr, "Failed to add: %s" %(folder)
+                pass
+
+        menu_items.sort( \
+            cmp=lambda x,y: cmp(x['title'].lower(), y['title'].lower()))
+        cache.set('bpcms_mainmenu', menu_items, 3600)
 
     return {
         'content_menu': menu_items,
     }
+
+def content_submenu(request):
+    # find where we are in the tree first
+    folderpath = request.path
+    folderpath = re.sub(u'^%s' %(settings.BPCMS_ROOT), '', folderpath)
+    # right - that should be any prefix got rid of, now traverse the tree till
+    # we've got something
+    menu_items = cache.get('bpcms_contentmenu_%s' %(folderpath))
+
+    if not menu_items:
+        folder = None
+        folderpath = folderpath.split('/')
+        for part in folderpath:
+            try:
+                if folder:
+                    f = Folder.objects.get(parent__exact=folder, slug__exact=part)
+                else:
+                    f = Folder.objects.get(parent__isnull=True, slug__exact=part)
+                folder = f
+            except:
+                break
+
+        # We have the folder (in theory)
+        menu_items = []
+        if folder:
+            items = Document.objects.filter(folder__exact=folder, \
+                islive__exact=True)
+            for item in items:
+                if item.slug != 'index':
+                    menu_items.append(
+                        {
+                            'title': item.title,
+                            'url': item.get_absolute_url(),
+                        }
+                    )
+            folders = Folder.objects.filter(parent__exact=folder)
+            for folder in folders:
+                items = Document.objects.filter(folder__exact=folder, \
+                    islive__exact=True, slug__exact='index')
+                for item in items:
+                    menu_items.append(
+                        {
+                            'title': item.title,
+                            'url': folder.get_absolute_url(),
+                        }
+                    )
+
+        menu_items.sort( \
+            cmp=lambda x,y: cmp(x['title'].lower(), y['title'].lower()))
+
+        cache.set('bpcms_contentmenu_%s' %("/".join(folderpath)), menu_items)
+
+    return {
+        'content_submenu': menu_items,
+    }
+
+def content_breadcrumb(request):
+    if settings.BPCMS_ROOT == "/":
+        breadcrumb = [{'url': settings.BPCMS_ROOT, 'title': "Home"}]
+    else:
+        breadcrumb = [{'url': settings.BPCMS_ROOT, 'title': settings.BPCMS_ROOT},]
+    # find where we are in the tree first
+    folderpath = request.path
+    folderpath = re.sub(u'^%s' %(settings.BPCMS_ROOT), '', folderpath)
+    # right - that should be any prefix got rid of, now traverse the tree till
+    # we've got something
+    folder = None
+    folderpath = folderpath.split('/')
+    for part in folderpath:
+        try:
+            if folder:
+                f = Folder.objects.get(parent__exact=folder, slug__exact=part)
+            else:
+                f = Folder.objects.get(parent__isnull=True, slug__exact=part)
+            folder = f
+            try:
+                item = Document.objects.get(folder__exact=folder,
+                    slug__exact=u'index', islive__exact=True)
+                breadcrumb.append({'url': folder.get_absolute_url(),
+                    'title': item.title})
+            except:
+                breadcrumb.append({'url': folder.get_absolute_url(),
+                    'title': folder.title})
+        except:
+            # must have hit the end of the tree, lets see if we can get an
+            # item...
+            try:
+                if folder:
+                    item = Document.objects.get(folder__exact=folder, \
+                        slug__exact=part, islive__exact=True)
+                else:
+                    item = Document.objects.get(folder__isnull=True, \
+                        slug__exact=part, islive__exact=True)
+                breadcrumb.append({'url': item.get_absolute_url(), 'title': item.title})
+            except:
+                pass
+            break
+
+    return {
+        'content_breadcrumb': breadcrumb,
+    }