fixes #15

frontend/wikicms.py

@@ -5,7 +5,7 @@
 '''
 from wikibot.wikiclient import WikiClient
 from frontend.site import Site
-
+from bs4 import BeautifulSoup
 import traceback
 
 import requests
@@ -113,11 +113,22 @@ def proxy(self,path:str)->str:
         r = requests.get(url)
         return Response(r.content)
 
+    def filterEditSections(self,html):
+        '''
+        filter Edit sections from html code
+        '''
+        # https://stackoverflow.com/questions/5598524/can-i-remove-script-tags-with-beautifulsoup
+        soup = BeautifulSoup(html,'lxml')
+        # https://stackoverflow.com/questions/5041008/how-to-find-elements-by-class
+        for s in soup.select('span',{"class": "mw-editsection"}):
+            s.extract()
+        return str(soup)
 
-    def getContent(self,pagePath:str):
+    def getContent(self,pagePath:str,dofilterEditSections=True):
         ''' get the content for the given pagePath 
         Args:
-            pagePath(str): the page Pageh
+            pagePath(str): the pagePath
+            dofilterEditSections(bool): True if editSection html code should be filtered
         Returns:
             str: the HTML content for the given path
         '''
@@ -131,7 +142,11 @@ def getContent(self,pagePath:str):
                 error=self.checkPath(pagePath)
                 pageTitle=self.wikiPage(pagePath)
             if error is None:
+                if self.wiki is None:
+                    raise Exception("getContent without wiki - you might want to call open first")
                 content=self.wiki.getHtml(pageTitle)
+                if dofilterEditSections:
+                    content=self.filterEditSections(content)
         except Exception as e:
             error=self.errMsg(e)
         return pageTitle,content,error

setup.py

@@ -12,7 +12,7 @@
     long_description = open('../../README.md').read()
 
 setup(name='pyWikiCMS',
-      version='0.0.4',
+      version='0.0.5',
       description='python implementation of a Mediawiki based Content Management System',
       long_description=long_description,
       long_description_content_type='text/markdown',

templates/macros.html

@@ -84,7 +84,11 @@
 {% macro activateDataTable() -%}
     <script type="text/javascript">
     $(document).ready(function() {
-    	$('.table').DataTable();
+    	$('.table').DataTable({
+		    buttons: [
+		        'copy', 'excel', 'pdf'
+		    ]
+		});
 	});
     </script>
 {%- endmacro %}

tests/test_frontend.py

@@ -7,6 +7,7 @@
 from frontend.wikicms import Frontend
 from tests.test_webserver import TestWebServer
 
+
 class TestFrontend(unittest.TestCase):
     '''
     test the frontend
@@ -48,6 +49,27 @@ def testProxy(self):
         self.assertFalse(imageResponse is None)
         self.assertEqual("200 OK",imageResponse.status)
         self.assertEqual(79499,len(imageResponse.data))
+
+    def testIssue15(self):
+        '''
+        test Filter "edit" section buttons 
+        
+        see https://github.com/BITPlan/pyWikiCMS/issues/15
+        
+        '''
+        frontend=self.server.enableFrontend('cr')
+        unfiltered="""<span class="mw-editsection"><span class="mw-editsection-bracket">[</span><a href="/index.php?title=...;action=edit&amp;section=T-1" title="Edit section: ">edit</a><span class="mw-editsection-bracket">]</span></span>"""
+        filtered=frontend.filterEditSections(unfiltered)
+        if self.debug:
+            print(filtered)      
+        self.assertFalse('''<span class="mw-editsection">''' in filtered)
+        pageTitle,content,error=frontend.getContent('Issue15')
+        self.assertEqual("Issue15",pageTitle)
+        self.assertIsNone(error)
+        if self.debug:
+            print(content)
+        self.assertFalse('''<span class="mw-editsection">''' in content)
+
 
 if __name__ == "__main__":
     #import sys;sys.argv = ['', 'Test.testName']