big refactor 1

flatpages.py

@@ -0,0 +1,94 @@
+from flask_flatpages import FlatPages, pygments_style_defs
+import flask_flatpages
+import re
+from utils import path_depth,pjoin2
+from collections import namedtuple
+
+FileLists=namedtuple("FileLists", ["list_files", "list_images", "sub_pages", "sub_index_pages", "breadcrumbs"])
+
+class FlatPagesIndex(FlatPages):
+    @staticmethod
+    def get_breadcrumb_paths(path):
+        """
+        breadcrumbs decompose the path
+        a/b/index -> [a/index a/b/index]
+        """
+        elements = path.split('/')
+        elements2 = ['index']
+        for i in range(1,len(elements)):
+            elements2.append(pjoin2(elements[0:i])+u'/index')
+        return elements2
+
+
+    @staticmethod
+    def get_paths(pages):
+        return (p.path for p in pages)
+    
+    @staticmethod
+    def breadcrumbs(page):
+        "Parse a path or the path of a page into breadcrumbs"
+        if type(page) is flask_flatpages.Page:
+            return FlatPagesIndex.get_breadcrumb_paths(page.path)
+        else:
+            return FlatPagesIndex.get_breadcrumb_paths(page)
+ 
+    def get_pages(self, paths):
+        if paths is None:
+            raise AttributeError("paths for get_pages can't be None")
+        return (self.get(p) for p in paths)
+
+    def get_sub_pages(self, path):
+        def is_subpage(path, root):
+            return (path.startswith(root) and  # is a subpage
+                ( re.match('.*index',path) is None) and # is no index
+                path_depth(path)<path_depth(root)+2
+            )
+        return (p for p in self
+               if  is_subpage(str(p.path),path))
+    
+    # List all index subpages of the current page i.e. all pages in subdirectories
+    def get_sub_ipages(self, path):
+        cc=path_depth(path)
+        ppath="index" if path=="" else pjoin2([path,"index"])
+        ps=[p for p in self
+            if (p.path.startswith(path) and # subpages of this page
+                not ( re.match('.*index',p.path) is None) and # only index
+                ( ppath != p.path) and # not identical page
+                len(p.path.split('/'))<=cc+2) ] # only one level
+        return ps 
+
+
+    # load a page from a path information
+    def get_flatpage(self,path):
+        is_index= path=='' or path.split('/')[-1]=='index'
+        if path == 'index' or path=='': 
+            is_index=True
+            path=''
+            pathi='index'
+        else:
+            if path.split('/')[-1]=='index': # last element is "index" ?
+                is_index=True
+                path='/'.join(path.split('/')[0:-1]) # remove index from path
+            pathi = u'{}/{}'.format(path, 'index') # try to add index to the path
+        page = FlatPages.get(self,pathi) # try to load index page
+        if not page is None: # if successful it is a index
+            is_index=True
+        else: 
+            page=FlatPages.get(self,path)
+        return (is_index, path, page) 
+
+
+    def load_lists_for_templates(self, path, is_index):
+        path2 = pjoin(FLATPAGES_ROOT,path)
+        if is_index == True:
+            ld=list_dir_md(path2)
+            il=list_dir_img(path2)
+            sp=list(get_sub_pages(path))
+            spi=list(get_sub_ipages(path))
+        else:
+            ld=[]
+            sp=[]
+            spi=[]
+            il=[]
+        bc=get_bc(path)
+        return (ld, il, sp, spi,bc)