diff options
Diffstat (limited to 'src/pyssg/builder.py')
-rw-r--r-- | src/pyssg/builder.py | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/pyssg/builder.py b/src/pyssg/builder.py index 8472b20..d2163d2 100644 --- a/src/pyssg/builder.py +++ b/src/pyssg/builder.py @@ -28,6 +28,8 @@ class HTMLBuilder: self.md_files: list[str] = None self.html_files: list[str] = None + self.all_pages: list[Page] = None + def build(self) -> None: self.dirs = get_dir_structure(self.src, ['templates']) @@ -40,6 +42,9 @@ class HTMLBuilder: parser: MDParser = MDParser(self.src, self.md_files, self.db) parser.parse() + # just to be able to extract all pages out of this class + self.all_pages = parser.all_pages + # create the article index self.__create_article_index(parser.all_tags, parser.all_pages) @@ -52,6 +57,10 @@ class HTMLBuilder: self.__create_tags(parser.all_tags, parser.all_pages) + def get_pages(self) -> list[Page]: + return self.all_pages + + def __create_dir_structure(self) -> None: for d in self.dirs: # for the dir structure, |