Merge pull request #154 from FOSSEE/fix_cms

Fix CMS app

1 files changed, 29 insertions, 0 deletions

diff --git a/cms/views.py b/cms/views.py
new file mode 100644
index 0000000..fc1439a
--- /dev/null
+++ b/cms/views.py
@@ -0,0 +1,29 @@
+# Create your views here.
+from django.forms import model_to_dict
+from django.http import Http404
+from django.shortcuts import render
+
+from cms.models import Page, Nav, SubNav
+
+
+def home(request, permalink=''):
+    if permalink == '':
+        permalink = 'home'
+    page = Page.objects.filter(permalink=permalink, active=True)
+    if page.exists():
+        page = page.first()
+    else:
+        raise Http404("The requested page does not exists")
+    nav_objs = Nav.objects.filter(active=True).order_by('-position')
+
+    navs = []
+
+    for nav in nav_objs:
+        nav_obj = model_to_dict(nav)
+        nav_obj['subnavs'] = nav.subnav_set.filter(
+            active=True).order_by('position')
+        navs.insert(-1, nav_obj)
+
+    return render(
+        request, 'cms_base.html', {'page': page, 'navs': navs}
+    )