Add support for processing language parameter in message entities

Author: pplulee

telebot/formatting.py

@@ -418,13 +418,16 @@ def apply_html_entities(text: str, entities: Optional[List], custom_subs: Option
     utf16_text = text.encode("utf-16-le")
     html_text = ""
 
-    def func(upd_text, subst_type=None, url=None, user=None, custom_emoji_id=None):
+    def func(upd_text, subst_type=None, url=None, user=None, custom_emoji_id=None, language=None):
         upd_text = upd_text.decode("utf-16-le")
         if subst_type == "text_mention":
             subst_type = "text_link"
             url = "tg://user?id={0}".format(user.id)
         elif subst_type == "mention":
             url = "https://t.me/{0}".format(upd_text[1:])
+        elif subst_type == "pre":
+            if language is not None:
+                return '<pre><code class="language-{0}">{1}</code></pre>'.format(language, upd_text)
         upd_text = upd_text.replace("&", "&amp;").replace("<", "&lt;").replace(">", "&gt;")
         if not subst_type or not _subs.get(subst_type):
             return upd_text
@@ -444,14 +447,16 @@ def func(upd_text, subst_type=None, url=None, user=None, custom_emoji_id=None):
             offset = entity.offset
 
             new_string = func(utf16_text[offset * 2: (offset + entity.length) * 2], subst_type=entity.type,
-                              url=entity.url, user=entity.user, custom_emoji_id=entity.custom_emoji_id)
+                              url=entity.url, user=entity.user, custom_emoji_id=entity.custom_emoji_id,
+                              language=entity.language)
             start_index = len(html_text)
             html_text += new_string
             offset += entity.length
             end_index = len(html_text)
         elif entity.offset == offset:
             new_string = func(utf16_text[offset * 2: (offset + entity.length) * 2], subst_type=entity.type,
-                              url=entity.url, user=entity.user, custom_emoji_id=entity.custom_emoji_id)
+                              url=entity.url, user=entity.user, custom_emoji_id=entity.custom_emoji_id,
+                              language=entity.language)
             start_index = len(html_text)
             html_text += new_string
             end_index = len(html_text)
@@ -463,7 +468,8 @@ def func(upd_text, subst_type=None, url=None, user=None, custom_emoji_id=None):
             # And we don't change it).
             entity_string = html_text[start_index: end_index].encode("utf-16-le")
             formatted_string = func(entity_string, subst_type=entity.type, url=entity.url, user=entity.user,
-                                    custom_emoji_id=entity.custom_emoji_id). \
+                                    custom_emoji_id=entity.custom_emoji_id,
+                                    language=entity.language). \
                 replace("&amp;", "&").replace("&lt;", "<").replace("&gt;", ">")
             html_text = html_text[:start_index] + formatted_string + html_text[end_index:]
             end_index = len(html_text)