From 1751158cbde923b13de75e10550b6b12f7be36dc Mon Sep 17 00:00:00 2001 From: Lars Noack Date: Tue, 15 Apr 2025 13:11:59 +0200 Subject: [PATCH] feat: added logger --- stsg/build.py | 29 ++++++++++++++++++++--------- 1 file changed, 20 insertions(+), 9 deletions(-) diff --git a/stsg/build.py b/stsg/build.py index ace5624..4eb9866 100644 --- a/stsg/build.py +++ b/stsg/build.py @@ -73,8 +73,8 @@ class ArticleTranslation: f.write(self.get_article()) - def _get_values(self) -> Dict[str, str]: - return { + def _get_values(self, return_foreign_articles: bool = True) -> Dict[str, str]: + r = { "article_content": self.article_content, "article_preview": self.article_content[:ARTICLE_PREVIEW_LENGTH] + "...", "article_url": self.url, @@ -86,9 +86,14 @@ class ArticleTranslation: "article_language_flag": self.language_flag, } + if return_foreign_articles: + r.update(ARTICLE_REFERENCE_VALUES[self.language_code]) + + return r + def get_article_values(self) -> Dict[str, str]: res = {} - for key, value in self._get_values().items(): + for key, value in self._get_values(return_foreign_articles=False).items(): res[key + ":" + self.article_overview.slug] = value return res @@ -149,8 +154,8 @@ class ArticleOverview: for ca in self.child_articles: ca.build() - def _get_values(self) -> Dict[str, str]: - return { + def _get_values(self, return_foreign_articles: bool = True) -> Dict[str, str]: + r = { "article_url": self.url, "article_title": self.slug, "article_slug": self.slug, @@ -158,9 +163,15 @@ class ArticleOverview: "article_overview_cards": self.overview_cards, } + if return_foreign_articles: + r.update(ARTICLE_REFERENCE_VALUES[""]) + + return r + + def get_article_values(self) -> Dict[str, str]: res = {} - for key, value in self._get_values().items(): + for key, value in self._get_values(return_foreign_articles=False).items(): res[key + ":" + self.slug] = value return res @@ -216,10 +227,10 @@ def build(): # build article reverence values for article_overview in ARTICLE_LAKE.values(): - ARTICLE_REFERENCE_VALUES[None].update(article_overview.get_article_values()) + logger.info("found article %s", article_overview.slug) + + ARTICLE_REFERENCE_VALUES[""].update(article_overview.get_article_values()) for language_code, at in article_overview.article_translations_map.items(): ARTICLE_REFERENCE_VALUES[language_code].update(at.get_article_values()) - print(ARTICLE_REFERENCE_VALUES) - tree.build() \ No newline at end of file