summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/pyssg/parser.py19
1 files changed, 10 insertions, 9 deletions
diff --git a/src/pyssg/parser.py b/src/pyssg/parser.py
index d75c923..f2d23eb 100644
--- a/src/pyssg/parser.py
+++ b/src/pyssg/parser.py
@@ -67,14 +67,15 @@ class MDParser:
self.all_tags.sort(key=itemgetter(0))
self.updated_pages.sort(reverse=True)
self.all_pages.sort(reverse=True)
- # TODO: fix this in case it doesn't work lol
- for i, p in enumerate(self.all_pages):
- try:
- prev_page: Page = self.all_pages[i - 1]
- p.previous = prev_page
- except IndexError: pass
- try:
- next_page: Page = self.all_pages[i + 1]
+ pages_amount: int = len(self.all_pages)
+ # note that prev and next are switched because of the reverse rodering
+ # of all_pages
+ for i, p in enumerate(self.all_pages):
+ if i != 0:
+ next_page: Page = self.all_pages[i - 1]
p.next = next_page
- except IndexError: pass
+
+ if i != pages_amount - 1:
+ prev_page: Page = self.all_pages[i + 1]
+ p.previous = prev_page