wiki: Find other embedded files other than images

[ipfire.org.git] / src / backend / wiki.py

diff --git a/src/backend/wiki.py b/src/backend/wiki.py
index 66b867fd5cd27f94c8bc87fa60c400c9837a8d87..9bd19eca9debd767a6a823332bf2278da358d6f8 100644 (file)
--- a/src/backend/wiki.py
+++ b/src/backend/wiki.py
@@ -791,7 +791,10 @@ class WikiRenderer(misc.Object):
                        # Make the URL absolute
                        url = self.backend.wiki.make_path(self.path, url)
 
-                       files.append(url)
+                       # Check if this is a file (it could also just be a page)
+                       file = self.backend.wiki.get_file_by_path(url)
+                       if file:
+                               files.append(url)
 
                return files
 
@@ -834,6 +837,12 @@ class LinkedFilesExtractor(markdown.treeprocessors.Treeprocessor):
 
                        self.md.files.append(src)
 
+               # Find all links
+               for link in root.findall(".//a"):
+                       href = link.get("href")
+
+                       self.md.files.append(href)
+
 
 class LinkedFilesExtractorExtension(markdown.extensions.Extension):
     def extendMarkdown(self, md):