From 2f28bf2d05e512bef7af3f76972ceef944ff836f Mon Sep 17 00:00:00 2001
From: Konrad <klux2@hs-mittweida.de>
Date: Thu, 26 Sep 2019 16:25:25 +0200
Subject: [PATCH] added possibility to translate messages with the google

---
 HSMensaW_botA.py | 31 +++++++++++++++++++++++++------
 1 file changed, 25 insertions(+), 6 deletions(-)

diff --git a/HSMensaW_botA.py b/HSMensaW_botA.py
index 6f9aa01..4bfb66e 100644
--- a/HSMensaW_botA.py
+++ b/HSMensaW_botA.py
@@ -13,6 +13,7 @@ from typing import Any
 from urllib.error import HTTPError
 
 import telepot
+from googletrans import Translator
 from telepot.aio import DelegatorBot
 from telepot.aio.delegate import per_chat_id, create_open, pave_event_space
 from telepot.aio.loop import MessageLoop
@@ -41,6 +42,7 @@ info_str = "*Mensa-Bot der Hochschule Mittweida (beta)*\nDieser Bot versendet je
            "[vecteezy.com](https://de.vecteezy.com)"
 status = ""
 essen = []
+essen_eng = []
 var = True
 datum = None
 
@@ -131,11 +133,16 @@ class HSMensaW(telepot.aio.helper.ChatHandler):
                     write_config()
 
 
-async def send_essen(chat_id: int) -> None:
-    global datum, essen
-    nachricht = "Speiseplan am %s:\n" % datum.strftime("%A, den %d. %B %Y")
-    for i in essen:
-        nachricht += "- " + str(i).replace(".", ",") + "\n\n"
+async def send_essen(chat_id: int, deutsch: bool = True) -> None:
+    global datum, essen, essen_eng
+    if deutsch:
+        nachricht = "Speiseplan am %s:\n" % datum.strftime("%A, den %d. %B %Y")
+        for i in essen:
+            nachricht += "- " + str(i).replace(".", ",") + "\n\n"
+    else:
+        nachricht = "Menu on %s:\n" % datum.strftime("%A, %d %B %Y")
+        for i in essen_eng:
+            nachricht += "- " + str(i) + "\n\n"
     await send_message(bot_obj=bot, chat_id=chat_id, msg=nachricht, parse_mode="markdown")
 
 
@@ -183,8 +190,9 @@ async def get_essen(only_today: bool) -> None:
     :param only_today: only current day; no download if the mensa is closed
     :return:
     """
-    global datum, essen, var  # , ctx
+    global datum, essen, essen_eng, var  # , ctx
     essen = []
+    essen_eng = []
     try:
         # response = urllib.request.urlopen(url, context=ctx)
         response = urllib.request.urlopen(url)
@@ -221,15 +229,24 @@ async def get_essen(only_today: bool) -> None:
     datum = date_dt
     menus = et.findall("./menus/day[date='" + datum.isoformat() + "']/menu")
 
+    essen_transl = []
+    preise = []
     for i in menus:
         kategorie = i.findall("type")[0].text
+        essen_transl.append(kategorie)
         name = i.findall("description")[0].text.rstrip("()1234567890, ")
+        essen_transl.append(name)
         try:
             preis = float(i.findall("./prices/price[label='Studenten']/value")[0].text.replace(",", "."))
         except ValueError:
             preis = -1
+        preise.append(preis)
         essen.append(Essen(name, preis, kategorie))
 
+    translations = translator.translate(essen_transl)
+    for i in range(int(len(translations) / 2)):
+        essen_eng.append(Essen(translations[i * 2 + 1].text, preise[i], translations[i * 2].text))
+
 
 def get_bot_id() -> int:
     global botID
@@ -304,6 +321,8 @@ signal.signal(signal.SIGINT, shutdown)
 
 locale.setlocale(locale.LC_TIME, 'de_DE.UTF-8')
 
+translator = Translator()
+
 try:
     with open(config_filename, 'r') as config_file:
         config = json.load(config_file)