Hazel/STSG
1
2
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: added logger

Browse Source
This commit is contained in:
Hazel 2025-04-15 13:11:59 +02:00
parent c45cebe497
commit 1751158cbd
1 changed files with 20 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
stsg/build.py
View File

@ -73,8 +73,8 @@ class ArticleTranslation:
f.write(self.get_article()) f.write(self.get_article())
def _get_values(self) -> Dict[str, str]: def _get_values(self, return_foreign_articles: bool = True) -> Dict[str, str]:
return { r = {
"article_content": self.article_content, "article_content": self.article_content,
"article_preview": self.article_content[:ARTICLE_PREVIEW_LENGTH] + "...", "article_preview": self.article_content[:ARTICLE_PREVIEW_LENGTH] + "...",
"article_url": self.url, "article_url": self.url,
@ -86,9 +86,14 @@ class ArticleTranslation:
"article_language_flag": self.language_flag, "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]: def get_article_values(self) -> Dict[str, str]:
res = {} 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 res[key + ":" + self.article_overview.slug] = value
return res return res
@ -149,8 +154,8 @@ class ArticleOverview:
for ca in self.child_articles: for ca in self.child_articles:
ca.build() ca.build()
def _get_values(self) -> Dict[str, str]: def _get_values(self, return_foreign_articles: bool = True) -> Dict[str, str]:
return { r = {
"article_url": self.url, "article_url": self.url,
"article_title": self.slug, "article_title": self.slug,
"article_slug": self.slug, "article_slug": self.slug,
@ -158,9 +163,15 @@ class ArticleOverview:
"article_overview_cards": self.overview_cards, "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]: def get_article_values(self) -> Dict[str, str]:
res = {} 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 res[key + ":" + self.slug] = value
return res return res
@ -216,10 +227,10 @@ def build():
# build article reverence values # build article reverence values
for article_overview in ARTICLE_LAKE.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(): for language_code, at in article_overview.article_translations_map.items():
ARTICLE_REFERENCE_VALUES[language_code].update(at.get_article_values()) ARTICLE_REFERENCE_VALUES[language_code].update(at.get_article_values())
print(ARTICLE_REFERENCE_VALUES)
tree.build() tree.build()