forked from genewildish/Mainline
David Gwilliam
757c854584
- Fix pre-existing lint errors in engine/ modules using ruff --unsafe-fixes - Add hk.pkl with pre-commit and pre-push hooks using ruff builtin - Configure hooks to use 'uv run' prefix for tool execution - Update mise.toml to include hk and pkl tools - All 73 tests pass fix: apply ruff auto-fixes and add hk git hooks - Fix pre-existing lint errors in engine/ modules using ruff --unsafe-fixes - Add hk.pkl with pre-commit and pre-push hooks using ruff builtin - Configure hooks to use 'uv run' prefix for tool execution - Update mise.toml to include hk and pkl tools - Use 'hk install --mise' for proper mise integration - All 73 tests pass
44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
"""
|
|
Google Translate wrapper and location→language detection.
|
|
Depends on: sources (for LOCATION_LANGS).
|
|
"""
|
|
|
|
import json
|
|
import re
|
|
import urllib.parse
|
|
import urllib.request
|
|
|
|
from engine.sources import LOCATION_LANGS
|
|
|
|
_TRANSLATE_CACHE = {}
|
|
|
|
|
|
def detect_location_language(title):
|
|
"""Detect if headline mentions a location, return target language."""
|
|
title_lower = title.lower()
|
|
for pattern, lang in LOCATION_LANGS.items():
|
|
if re.search(pattern, title_lower):
|
|
return lang
|
|
return None
|
|
|
|
|
|
def translate_headline(title, target_lang):
|
|
"""Translate headline via Google Translate API (zero dependencies)."""
|
|
key = (title, target_lang)
|
|
if key in _TRANSLATE_CACHE:
|
|
return _TRANSLATE_CACHE[key]
|
|
try:
|
|
q = urllib.parse.quote(title)
|
|
url = (
|
|
"https://translate.googleapis.com/translate_a/single"
|
|
f"?client=gtx&sl=en&tl={target_lang}&dt=t&q={q}"
|
|
)
|
|
req = urllib.request.Request(url, headers={"User-Agent": "mainline/0.1"})
|
|
resp = urllib.request.urlopen(req, timeout=5)
|
|
data = json.loads(resp.read())
|
|
result = "".join(p[0] for p in data[0] if p[0]) or title
|
|
except Exception:
|
|
result = title
|
|
_TRANSLATE_CACHE[key] = result
|
|
return result
|