Update usages of Parser all around the library

delivrance · delivrance · commit f05e79e0f4a7 · 2019-06-26T16:08:24.000+02:00
diff --git a/pyrogram/client/ext/base_client.py b/pyrogram/client/ext/base_client.py
@@ -25,7 +25,7 @@
 from threading import Lock
 
 from pyrogram import __version__
-from ..style import Markdown, HTML
+from ..parser import Parser
 from ...session.internals import MsgId
 
 
@@ -92,8 +92,7 @@ def __init__(self):
 
         self.rnd_id = MsgId
 
-        self.markdown = Markdown(self)
-        self.html = HTML(self)
+        self.parser = Parser(self)
 
         self.session = None
         self.media_sessions = {}
diff --git a/pyrogram/client/methods/messages/edit_inline_text.py b/pyrogram/client/methods/messages/edit_inline_text.py
@@ -16,6 +16,8 @@
 # You should have received a copy of the GNU Lesser General Public License
 # along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
+from typing import Union
+
 import pyrogram
 from pyrogram.api import functions
 from pyrogram.client.ext import BaseClient, utils
@@ -26,7 +28,7 @@ def edit_inline_text(
         self,
         inline_message_id: str,
         text: str,
-        parse_mode: str = "",
+        parse_mode: Union[str, None] = "",
         disable_web_page_preview: bool = None,
         reply_markup: "pyrogram.InlineKeyboardMarkup" = None
     ) -> bool:
@@ -55,13 +57,12 @@ def edit_inline_text(
         Raises:
             RPCError: In case of a Telegram RPC error.
         """
-        style = self.html if parse_mode.lower() == "html" else self.markdown
 
         return self.send(
             functions.messages.EditInlineBotMessage(
                 id=utils.unpack_inline_message_id(inline_message_id),
                 no_webpage=disable_web_page_preview or None,
                 reply_markup=reply_markup.write() if reply_markup else None,
-                **style.parse(text)
+                **self.parser.parse(text, parse_mode)
             )
         )
diff --git a/pyrogram/client/methods/messages/edit_message_text.py b/pyrogram/client/methods/messages/edit_message_text.py
@@ -29,7 +29,7 @@ def edit_message_text(
         chat_id: Union[int, str],
         message_id: int,
         text: str,
-        parse_mode: str = "",
+        parse_mode: Union[str, None] = "",
         disable_web_page_preview: bool = None,
         reply_markup: "pyrogram.InlineKeyboardMarkup" = None
     ) -> "pyrogram.Message":
@@ -63,15 +63,14 @@ def edit_message_text(
         Raises:
             RPCError: In case of a Telegram RPC error.
         """
-        style = self.html if parse_mode.lower() == "html" else self.markdown
 
         r = self.send(
             functions.messages.EditMessage(
                 peer=self.resolve_peer(chat_id),
                 id=message_id,
                 no_webpage=disable_web_page_preview or None,
                 reply_markup=reply_markup.write() if reply_markup else None,
-                **style.parse(text)
+                **self.parser.parse(text, parse_mode)
             )
         )
 
diff --git a/pyrogram/client/methods/messages/send_animation.py b/pyrogram/client/methods/messages/send_animation.py
@@ -32,7 +32,7 @@ def send_animation(
         animation: str,
         caption: str = "",
         unsave: bool = False,
-        parse_mode: str = "",
+        parse_mode: Union[str, None] = "",
         duration: int = 0,
         width: int = 0,
         height: int = 0,
@@ -130,7 +130,6 @@ def send_animation(
             RPCError: In case of a Telegram RPC error.
         """
         file = None
-        style = self.html if parse_mode.lower() == "html" else self.markdown
 
         try:
             if os.path.exists(animation):
@@ -168,7 +167,7 @@ def send_animation(
                             reply_to_msg_id=reply_to_message_id,
                             random_id=self.rnd_id(),
                             reply_markup=reply_markup.write() if reply_markup else None,
-                            **style.parse(caption)
+                            **self.parser.parse(caption, parse_mode)
                         )
                     )
                 except FilePartMissing as e:
diff --git a/pyrogram/client/methods/messages/send_audio.py b/pyrogram/client/methods/messages/send_audio.py
@@ -31,7 +31,7 @@ def send_audio(
         chat_id: Union[int, str],
         audio: str,
         caption: str = "",
-        parse_mode: str = "",
+        parse_mode: Union[str, None] = "",
         duration: int = 0,
         performer: str = None,
         title: str = None,
@@ -127,7 +127,6 @@ def send_audio(
             RPCError: In case of a Telegram RPC error.
         """
         file = None
-        style = self.html if parse_mode.lower() == "html" else self.markdown
 
         try:
             if os.path.exists(audio):
@@ -163,7 +162,7 @@ def send_audio(
                             reply_to_msg_id=reply_to_message_id,
                             random_id=self.rnd_id(),
                             reply_markup=reply_markup.write() if reply_markup else None,
-                            **style.parse(caption)
+                            **self.parser.parse(caption, parse_mode)
                         )
                     )
                 except FilePartMissing as e:
diff --git a/pyrogram/client/methods/messages/send_cached_media.py b/pyrogram/client/methods/messages/send_cached_media.py
@@ -29,7 +29,7 @@ def send_cached_media(
         chat_id: Union[int, str],
         file_id: str,
         caption: str = "",
-        parse_mode: str = "",
+        parse_mode: Union[str, None] = "",
         disable_notification: bool = None,
         reply_to_message_id: int = None,
         reply_markup: Union[
@@ -79,7 +79,6 @@ def send_cached_media(
         Raises:
             RPCError: In case of a Telegram RPC error.
         """
-        style = self.html if parse_mode.lower() == "html" else self.markdown
 
         r = self.send(
             functions.messages.SendMedia(
@@ -89,7 +88,7 @@ def send_cached_media(
                 reply_to_msg_id=reply_to_message_id,
                 random_id=self.rnd_id(),
                 reply_markup=reply_markup.write() if reply_markup else None,
-                **style.parse(caption)
+                **self.parser.parse(caption, parse_mode)
             )
         )
 
diff --git a/pyrogram/client/methods/messages/send_document.py b/pyrogram/client/methods/messages/send_document.py
@@ -32,7 +32,7 @@ def send_document(
         document: str,
         thumb: str = None,
         caption: str = "",
-        parse_mode: str = "",
+        parse_mode: Union[str, None] = "",
         disable_notification: bool = None,
         reply_to_message_id: int = None,
         reply_markup: Union[
@@ -113,7 +113,6 @@ def send_document(
             RPCError: In case of a Telegram RPC error.
         """
         file = None
-        style = self.html if parse_mode.lower() == "html" else self.markdown
 
         try:
             if os.path.exists(document):
@@ -144,7 +143,7 @@ def send_document(
                             reply_to_msg_id=reply_to_message_id,
                             random_id=self.rnd_id(),
                             reply_markup=reply_markup.write() if reply_markup else None,
-                            **style.parse(caption)
+                            **self.parser(caption, parse_mode)
                         )
                     )
                 except FilePartMissing as e:
diff --git a/pyrogram/client/methods/messages/send_message.py b/pyrogram/client/methods/messages/send_message.py
@@ -28,7 +28,7 @@ def send_message(
         self,
         chat_id: Union[int, str],
         text: str,
-        parse_mode: str = "",
+        parse_mode: Union[str, None] = "",
         disable_web_page_preview: bool = None,
         disable_notification: bool = None,
         reply_to_message_id: int = None,
@@ -74,8 +74,7 @@ def send_message(
         Raises:
             RPCError: In case of a Telegram RPC error.
         """
-        style = self.html if parse_mode.lower() == "html" else self.markdown
-        message, entities = style.parse(text).values()
+        message, entities = self.parser.parse(text, parse_mode).values()
 
         r = self.send(
             functions.messages.SendMessage(
diff --git a/pyrogram/client/methods/messages/send_photo.py b/pyrogram/client/methods/messages/send_photo.py
@@ -31,7 +31,7 @@ def send_photo(
         chat_id: Union[int, str],
         photo: str,
         caption: str = "",
-        parse_mode: str = "",
+        parse_mode: Union[str, None] = "",
         ttl_seconds: int = None,
         disable_notification: bool = None,
         reply_to_message_id: int = None,
@@ -112,7 +112,6 @@ def send_photo(
             RPCError: In case of a Telegram RPC error.
         """
         file = None
-        style = self.html if parse_mode.lower() == "html" else self.markdown
 
         try:
             if os.path.exists(photo):
@@ -139,7 +138,7 @@ def send_photo(
                             reply_to_msg_id=reply_to_message_id,
                             random_id=self.rnd_id(),
                             reply_markup=reply_markup.write() if reply_markup else None,
-                            **style.parse(caption)
+                            **self.parser.parse(caption, parse_mode)
                         )
                     )
                 except FilePartMissing as e:
diff --git a/pyrogram/client/methods/messages/send_video.py b/pyrogram/client/methods/messages/send_video.py
@@ -31,7 +31,7 @@ def send_video(
         chat_id: Union[int, str],
         video: str,
         caption: str = "",
-        parse_mode: str = "",
+        parse_mode: Union[str, None] = "",
         duration: int = 0,
         width: int = 0,
         height: int = 0,
@@ -129,7 +129,6 @@ def send_video(
             RPCError: In case of a Telegram RPC error.
         """
         file = None
-        style = self.html if parse_mode.lower() == "html" else self.markdown
 
         try:
             if os.path.exists(video):
@@ -166,7 +165,7 @@ def send_video(
                             reply_to_msg_id=reply_to_message_id,
                             random_id=self.rnd_id(),
                             reply_markup=reply_markup.write() if reply_markup else None,
-                            **style.parse(caption)
+                            **self.parser.parse(caption, parse_mode)
                         )
                     )
                 except FilePartMissing as e:
diff --git a/pyrogram/client/methods/messages/send_voice.py b/pyrogram/client/methods/messages/send_voice.py
@@ -31,7 +31,7 @@ def send_voice(
         chat_id: Union[int, str],
         voice: str,
         caption: str = "",
-        parse_mode: str = "",
+        parse_mode: Union[str, None] = "",
         duration: int = 0,
         disable_notification: bool = None,
         reply_to_message_id: int = None,
@@ -110,7 +110,6 @@ def send_voice(
             RPCError: In case of a Telegram RPC error.
         """
         file = None
-        style = self.html if parse_mode.lower() == "html" else self.markdown
 
         try:
             if os.path.exists(voice):
@@ -142,7 +141,7 @@ def send_voice(
                             reply_to_msg_id=reply_to_message_id,
                             random_id=self.rnd_id(),
                             reply_markup=reply_markup.write() if reply_markup else None,
-                            **style.parse(caption)
+                            **self.parser.parse(caption, parse_mode)
                         )
                     )
                 except FilePartMissing as e:
diff --git a/pyrogram/client/types/input_message_content/input_text_message_content.py b/pyrogram/client/types/input_message_content/input_text_message_content.py
@@ -16,9 +16,11 @@
 # You should have received a copy of the GNU Lesser General Public License
 # along with Pyrogram.  If not, see <http://www.gnu.org/licenses/>.
 
+from typing import Union
+
 from pyrogram.api import types
 from .input_message_content import InputMessageContent
-from ...style import HTML, Markdown
+from ...parser import Parser
 
 
 class InputTextMessageContent(InputMessageContent):
@@ -38,7 +40,7 @@ class InputTextMessageContent(InputMessageContent):
 
     __slots__ = ["message_text", "parse_mode", "disable_web_page_preview"]
 
-    def __init__(self, message_text: str, parse_mode: str = "", disable_web_page_preview: bool = None):
+    def __init__(self, message_text: str, parse_mode: Union[str, None] = "", disable_web_page_preview: bool = None):
         super().__init__()
 
         self.message_text = message_text
@@ -49,5 +51,5 @@ def write(self, reply_markup):
         return types.InputBotInlineMessageText(
             no_webpage=self.disable_web_page_preview or None,
             reply_markup=reply_markup.write() if reply_markup else None,
-            **(HTML() if self.parse_mode.lower() == "html" else Markdown()).parse(self.message_text)
+            **(Parser(None)).parse(self.message_text, self.parse_mode)
         )
diff --git a/pyrogram/client/types/messages_and_media/message.py b/pyrogram/client/types/messages_and_media/message.py
@@ -31,7 +31,7 @@
 from ..update import Update
 from ..user_and_chats.chat import Chat
 from ..user_and_chats.user import User
-from ...style import utils, Markdown, HTML
+from ...parser import utils, Parser
 
 
 class Str(str):
@@ -47,11 +47,11 @@ def init(self, entities):
 
     @property
     def markdown(self):
-        return Markdown.unparse(self, self.entities)
+        return Parser.unparse(self, self.entities, False)
 
     @property
     def html(self):
-        return HTML.unparse(self, self.entities)
+        return Parser.unparse(self, self.entities, True)
 
     def __getitem__(self, item):
         return utils.remove_surrogates(utils.add_surrogates(self)[item])
