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

diff --git a/sommitrealweird/bpcms/context_processors.py b/sommitrealweird/bpcms/context_processors.py
index 90b98cc..b464307 100644
--- a/sommitrealweird/bpcms/context_processors.py
+++ b/sommitrealweird/bpcms/context_processors.py
@@ -1,5 +1,6 @@
 from bpcms.models import Document, Folder
-import operator
+from django.conf import settings
+import re
 
 def content_menu(request):
     items = Document.objects.filter(folder__isnull=True, islive__exact=True)
@@ -26,8 +27,65 @@ def content_menu(request):
         except:
             pass
 
-    menu_items.sort(cmp=lambda x,y: cmp(x['title'].lower(), y['title'].lower()))
+    menu_items.sort( \
+        cmp=lambda x,y: cmp(x['title'].lower(), y['title'].lower()))
 
     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
+    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_basic_url(),
+                    }
+                )
+        try:
+            folders = Folder.objects.filter(parent__exact=folder)
+        except:
+            pass
+        for folder in folders:
+            try:
+                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_basic_url(),
+                        }
+                    )
+            except:
+                pass
+
+    menu_items.sort( \
+        cmp=lambda x,y: cmp(x['title'].lower(), y['title'].lower()))
+
+    return {
+        'content_submenu': menu_items,
+    }