From 0aee560efe392aefe83b76f928746a6cbaeb13f5 Mon Sep 17 00:00:00 2001
From: Yuval Adam <_@yuv.al>
Date: Sat, 15 May 2021 15:46:07 +0300
Subject: Add localization support

---
 alarmpy.py | 23 ++++++++++++++++++++++-
 1 file changed, 22 insertions(+), 1 deletion(-)

(limited to 'alarmpy.py')

diff --git a/alarmpy.py b/alarmpy.py
index bce4714..55f66c6 100644
--- a/alarmpy.py
+++ b/alarmpy.py
@@ -1,4 +1,5 @@
 import click
+import json
 import requests
 
 from datetime import datetime
@@ -17,12 +18,14 @@ class Alarm:
 
     def __init__(
         self,
+        language="he",
         delay=1,
         routine_delay=60 * 5,
         alarm_id=False,
         repeat_alarms=False,
         quiet=False,
     ):
+        self.language = language
         self.delay = delay
         self.last_routine_delay = routine_delay
         self.alarm_id = alarm_id
@@ -33,10 +36,15 @@ class Alarm:
         self.last_routine_output = 0
 
         self.session = self.init_session()
+        self.labels = self.load_labels()
 
     def init_session(self):
         return requests.Session()
 
+    def load_labels(self):
+        with open("labels.json", "r") as f:
+            return json.load(f)
+
     def start(self):
         while True:
             try:
@@ -105,13 +113,26 @@ class Alarm:
 
     def output_alarms(self, cities, alarm_id):
         self.output_leading_timestamp()
-        cities_str = ", ".join(cities)
+        cities_str = self.localize_cities(cities)
         click.secho(f"{cities_str} ", fg="red", nl=not self.alarm_id)
         if self.alarm_id:
             click.secho(f"({alarm_id})")
 
+    def localize_cities(self, cities):
+        localized_cities = [
+            self.labels.get(city, {}).get(f"label_{self.language}", city)
+            for city in cities
+        ]
+        return ", ".join(localized_cities)
+
 
 @click.command()
+@click.option(
+    "--language",
+    default="he",
+    type=click.Choice(["en", "he", "ar", "ru"]),
+    help="Alert language ",
+)
 @click.option("--delay", default=1, help="Polling delay in seconds")
 @click.option(
     "--routine-delay", default=60 * 5, help="Routine message delay in seconds"
-- 
cgit v1.3.1