diff options
| -rw-r--r-- | bemani/iidx.py | 2 | ||||
| -rw-r--r-- | news_feed.py | 6 | ||||
| -rw-r--r-- | translate.py | 4 |
3 files changed, 8 insertions, 4 deletions
diff --git a/bemani/iidx.py b/bemani/iidx.py index 67e1085..c13e05f 100644 --- a/bemani/iidx.py +++ b/bemani/iidx.py @@ -3,7 +3,7 @@ from datetime import datetime from urllib.parse import urljoin import re -IIDX_KEY_TERMS = [ +KEY_TERMS_TL = [ ("クプロ", "QPro") ] diff --git a/news_feed.py b/news_feed.py index 18c2616..ac90f0f 100644 --- a/news_feed.py +++ b/news_feed.py @@ -31,16 +31,18 @@ def get_news(news_url: str, version=None) -> list: if news_url == constants.SOUND_VOLTEX_EXCEED_GEAR_NEWS_SITE: site_data = download_site_as_html(news_url) news_posts = sorted(sound_voltex.parse_exceed_gear_news_site(site_data, constants.EAMUSEMENT_BASE_URL), key=lambda x: x['timestamp'], reverse=True) + news_posts = translate.add_translate_text_to_en(news_posts) elif news_url == constants.IIDX_PINKY_CRUSH_NEWS_SITE: site_data = download_site_as_html(news_url) news_posts = sorted(iidx.parse_pinky_crush_news_site(site_data, constants.EAMUSEMENT_BASE_URL), key=lambda x: x['timestamp'], reverse=True) - news_posts = translate.add_translate_text_to_en(news_posts) + news_posts = translate.add_translate_text_to_en(news_posts, iidx.KEY_TERMS_TL) elif news_url == constants.CHUNITHM_JP_NEWS_SITE: site_data = download_site_as_html(news_url) if version == constants.CHUNITHM_VERSION.VERSE: news_posts = sorted(chunithm_jp.parse_chuni_jp_verse_news_site(site_data), key=lambda x: x['timestamp'], reverse=True) + news_posts = translate.add_translate_text_to_en(news_posts) elif news_url == constants.CHUNITHM_INTL_NEWS_SITE: site_data = download_site_as_html(news_url) @@ -51,6 +53,7 @@ def get_news(news_url: str, version=None) -> list: site_data = download_site_as_html(news_url) if version == constants.MAIMAIDX_VERSION.PRISM_PLUS: news_posts = sorted(maimaidx_jp.parse_maimaidx_jp_prism_plus_news_site(site_data), key=lambda x: x['timestamp'], reverse=True) + news_posts = translate.add_translate_text_to_en(news_posts) elif news_url == constants.MAIMAIDX_INTL_NEWS_SITE: scraper = SiteScraper(headless=True) @@ -63,6 +66,7 @@ def get_news(news_url: str, version=None) -> list: site_data = download_site_as_html(news_url) if version == constants.ONGEKI_VERSION.REFRESH: news_posts = sorted(ongeki_jp.parse_ongeki_refresh_news_site(site_data), key=lambda x: x['timestamp'], reverse=True) + news_posts = translate.add_translate_text_to_en(news_posts) else: news_posts = [] return news_posts diff --git a/translate.py b/translate.py index 64ba018..ecd2680 100644 --- a/translate.py +++ b/translate.py @@ -103,12 +103,12 @@ def add_translate_text_to_en(news_post: dict, overrides: list=[]) -> dict: translated_posts = [] translation_cache = _load_translation_cache() for post in news_post: - headline = post["headline"] + headline = post.get("headline") if headline: for override in overrides: headline = headline.replace(override[0], override[1]) post["en_headline"] = request_google_translate(headline, translation_cache=translation_cache) - content = post["content"] + content = post.get("content") if content: for override in overrides: content = content.replace(override[0], override[1]) |