i18n class was added

telebot/util.py

@@ -1,4 +1,6 @@
 # -*- coding: utf-8 -*-
+import gettext
+import os
 import random
 import re
 import string
@@ -21,6 +23,7 @@ try:
     # noinspection PyPackageRequirements
     from PIL import Image
     from io import BytesIO
+
     pil_imported = True
 except:
     pil_imported = False
@@ -32,12 +35,14 @@ logger = logging.getLogger('TeleBot')
 thread_local = threading.local()
 
 content_type_media = [
-    'text', 'audio', 'animation', 'document', 'photo', 'sticker', 'video', 'video_note', 'voice', 'contact', 'dice', 'poll',
+    'text', 'audio', 'animation', 'document', 'photo', 'sticker', 'video', 'video_note', 'voice', 'contact', 'dice',
+    'poll',
     'venue', 'location'
 ]
 
 content_type_service = [
-    'new_chat_members', 'left_chat_member', 'new_chat_title', 'new_chat_photo', 'delete_chat_photo', 'group_chat_created',
+    'new_chat_members', 'left_chat_member', 'new_chat_title', 'new_chat_photo', 'delete_chat_photo',
+    'group_chat_created',
     'supergroup_chat_created', 'channel_chat_created', 'migrate_to_chat_id', 'migrate_from_chat_id', 'pinned_message',
     'proximity_alert_triggered', 'voice_chat_scheduled', 'voice_chat_started', 'voice_chat_ended',
     'voice_chat_participants_invited', 'message_auto_delete_timer_changed'
@@ -49,6 +54,7 @@ update_types = [
     "my_chat_member", "chat_member", "chat_join_request"
 ]
 
+
 class WorkerThread(threading.Thread):
     count = 0
 
@@ -305,9 +311,12 @@ def smart_split(text: str, chars_per_string: int=MAX_MESSAGE_LENGTH) -> List[str
 
         part = text[:chars_per_string]
 
-        if "\n" in part: part = _text_before_last("\n")
-        elif ". " in part: part = _text_before_last(". ")
-        elif " " in part: part = _text_before_last(" ")
+        if "\n" in part:
+            part = _text_before_last("\n")
+        elif ". " in part:
+            part = _text_before_last(". ")
+        elif " " in part:
+            part = _text_before_last(" ")
 
         parts.append(part)
         text = text[len(part):]
@@ -450,15 +459,19 @@ def deprecated(warn: bool=True, alternative: Optional[Callable]=None):
     :param warn: If True a warning is logged else an info
     :param alternative: The new function to use instead
     """
+
     def decorator(function):
         def wrapper(*args, **kwargs):
-            info = f"`{function.__name__}` is deprecated." + (f" Use `{alternative.__name__}` instead" if alternative else "")
+            info = f"`{function.__name__}` is deprecated." + (
+                f" Use `{alternative.__name__}` instead" if alternative else "")
             if not warn:
                 logger.info(info)
             else:
                 logger.warning(info)
             return function(*args, **kwargs)
+
         return wrapper
+
     return decorator
 
 
@@ -477,6 +490,7 @@ def webhook_google_functions(bot, request):
     else:
         return 'Bot ON'
 
+
 def antiflood(function, *args, **kwargs):
     """
     Use this function inside loops in order to avoid getting TooManyRequests error.
@@ -499,3 +513,68 @@ def antiflood(function, *args, **kwargs):
             msg = function(*args, **kwargs)
     finally:
         return msg
+
+
+def find_translations(path, domain):
+    """
+    Looks for translations with passed 'domain' in passed 'path'
+    """
+    if not os.path.exists(path):
+        raise RuntimeError(f"Translations directory by path: {path!r} was not found")
+
+    result = {}
+
+    for name in os.listdir(path):
+        translations_path = os.path.join(path, name, 'LC_MESSAGES')
+
+        if not os.path.isdir(translations_path):
+            continue
+
+        po_file = os.path.join(translations_path, domain + '.po')
+        mo_file = po_file[:-2] + 'mo'
+
+        if os.path.isfile(po_file) and not os.path.isfile(mo_file):
+            raise FileNotFoundError(f"Translations for: {name!r} were not compiled!")
+
+        with open(mo_file, 'rb') as file:
+            result[name] = gettext.GNUTranslations(file)
+
+    return result
+
+
+class I18N:
+    """
+    This class provides high-level tool for internationalization
+    It is based on gettext util.
+    """
+
+    def __init__(self, translations_path, domain_name: str):
+        self.path = translations_path
+        self.domain = domain_name
+        self.translations = find_translations(self.path, self.domain)
+
+    @property
+    def available_translations(self):
+        return list(self.translations)
+
+    def gettext(self, text: str, lang: str = None):
+        """
+        Singular translations
+        """
+        if not lang or lang not in self.translations:
+            return text
+
+        translator = self.translations[lang]
+        return translator.gettext(text)
+
+    def ngettext(self, singular: str, plural: str, lang: str = None, n=1):
+        """
+        Plural translations
+        """
+        if not lang or lang not in self.translations:
+            if n == 1:
+                return singular
+            return plural
+
+        translator = self.translations[lang]
+        return translator.ngettext(singular, plural, n)