Add missing async/await

This commit is contained in:
Dan 2019-08-01 10:43:09 +02:00 committed by GitHub
parent 8700e3a0f3
commit eddff4769c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -30,7 +30,7 @@ class Parser:
self.html = HTML(client) self.html = HTML(client)
self.markdown = Markdown(client) self.markdown = Markdown(client)
def parse(self, text: str, mode: Union[str, None] = object): async def parse(self, text: str, mode: Union[str, None] = object):
text = str(text).strip() text = str(text).strip()
if mode == object: if mode == object:
@ -48,13 +48,13 @@ class Parser:
mode = mode.lower() mode = mode.lower()
if mode == "combined": if mode == "combined":
return self.markdown.parse(text) return await self.markdown.parse(text)
if mode in ["markdown", "md"]: if mode in ["markdown", "md"]:
return self.markdown.parse(text, True) return await self.markdown.parse(text, True)
if mode == "html": if mode == "html":
return self.html.parse(text) return await self.html.parse(text)
raise ValueError('parse_mode must be one of {} or None. Not "{}"'.format( raise ValueError('parse_mode must be one of {} or None. Not "{}"'.format(
", ".join('"{}"'.format(m) for m in pyrogram.Client.PARSE_MODES[:-1]), ", ".join('"{}"'.format(m) for m in pyrogram.Client.PARSE_MODES[:-1]),