rssguard/resources/scripts/scrapers/translate-rss2.py

39 lines
1.2 KiB
Python
Executable File

# Translates entries of RSS 2.0 feed into different locale.
#
# Make sure to have all dependencies installed:
# pip3 install googletrans==4.0.0-rc1
#
# You must provide raw RSS 2.0 UTF-8 feed XML data as input, for example with curl:
# curl 'https://phys.org/rss-feed/' | python ./translate-rss2.py "en" "pt_BR"
#
# You must provide two additional command line arguments:
# translate-rss2.py [FROM-LANGUAGE] [TO-LANGUAGE]
import sys
import time
import xml.etree.ElementTree as ET
from googletrans import Translator
lang_from = sys.argv[1]
lang_to = sys.argv[2]
sys.stdin.reconfigure(encoding='utf-8')
rss_data = sys.stdin.read()
rss_document = ET.fromstring(rss_data)
translator = Translator()
def translate_string(to_translate):
translated_text = translator.translate(to_translate, src = lang_from, dest = lang_to)
time.sleep(0.2)
return translated_text.text
def process_article(article):
title = article.find("title")
title.text = translate_string(title.text)
contents = article.find("description")
contents.text = translate_string(" ".join(contents.itertext()))
for article in rss_document.findall(".//item"):
process_article(article)
print(ET.tostring(rss_document, encoding = "unicode"))