Blacken and isort code

mautrix_telegram/tgclient.py

@@ -1,5 +1,5 @@
 # mautrix-telegram - A Matrix-Telegram puppeting bridge
-# Copyright (C) 2019 Tulir Asokan
+# Copyright (C) 2021 Tulir Asokan
 #
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU Affero General Public License as published by
@@ -13,24 +13,35 @@
 #
 # You should have received a copy of the GNU Affero General Public License
 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
-from typing import List, Union, Optional
+from typing import List, Optional, Union
 
 from telethon import TelegramClient, utils
-from telethon.tl.functions.messages import SendMediaRequest
-from telethon.tl.types import (InputMediaUploadedDocument, InputMediaUploadedPhoto,
-                               TypeDocumentAttribute, TypeInputMedia, TypeInputPeer,
-                               TypeMessageEntity, TypeMessageMedia, TypePeer)
-from telethon.tl.patched import Message
 from telethon.sessions.abstract import Session
+from telethon.tl.functions.messages import SendMediaRequest
+from telethon.tl.patched import Message
+from telethon.tl.types import (
+    InputMediaUploadedDocument,
+    InputMediaUploadedPhoto,
+    TypeDocumentAttribute,
+    TypeInputMedia,
+    TypeInputPeer,
+    TypeMessageEntity,
+    TypeMessageMedia,
+    TypePeer,
+)
 
 
 class MautrixTelegramClient(TelegramClient):
     session: Session
 
-    async def upload_file_direct(self, file: bytes, mime_type: str = None,
-                                 attributes: List[TypeDocumentAttribute] = None,
-                                 file_name: str = None, max_image_size: float = 10 * 1000 ** 2,
-                                 ) -> Union[InputMediaUploadedDocument, InputMediaUploadedPhoto]:
+    async def upload_file_direct(
+        self,
+        file: bytes,
+        mime_type: str = None,
+        attributes: List[TypeDocumentAttribute] = None,
+        file_name: str = None,
+        max_image_size: float = 10 * 1000 ** 2,
+    ) -> Union[InputMediaUploadedDocument, InputMediaUploadedPhoto]:
         file_handle = await super().upload_file(file, file_name=file_name)
 
         if (mime_type == "image/png" or mime_type == "image/jpeg") and len(file) < max_image_size:
@@ -42,14 +53,20 @@ class MautrixTelegramClient(TelegramClient):
             return InputMediaUploadedDocument(
                 file=file_handle,
                 mime_type=mime_type or "application/octet-stream",
-                attributes=list(attr_dict.values()))
+                attributes=list(attr_dict.values()),
+            )
 
-    async def send_media(self, entity: Union[TypeInputPeer, TypePeer],
-                         media: Union[TypeInputMedia, TypeMessageMedia],
-                         caption: str = None, entities: List[TypeMessageEntity] = None,
-                         reply_to: int = None) -> Optional[Message]:
+    async def send_media(
+        self,
+        entity: Union[TypeInputPeer, TypePeer],
+        media: Union[TypeInputMedia, TypeMessageMedia],
+        caption: str = None,
+        entities: List[TypeMessageEntity] = None,
+        reply_to: int = None,
+    ) -> Optional[Message]:
         entity = await self.get_input_entity(entity)
         reply_to = utils.get_message_id(reply_to)
-        request = SendMediaRequest(entity, media, message=caption or "", entities=entities or [],
-                                   reply_to_msg_id=reply_to)
+        request = SendMediaRequest(
+            entity, media, message=caption or "", entities=entities or [], reply_to_msg_id=reply_to
+        )
         return self._get_response_message(request, await self(request), entity)