Skip to content
Extraits de code Groupes Projets
Valider c106c99b rédigé par Tulir Asokan's avatar Tulir Asokan
Parcourir les fichiers

Maybe make import.py work on Python 3.6

parent 5e3fc57b
Aucune branche associée trouvée
Aucune étiquette associée trouvée
Aucune requête de fusion associée trouvée
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
# This Source Code Form is subject to the terms of the Mozilla Public # This Source Code Form is subject to the terms of the Mozilla Public
# License, v. 2.0. If a copy of the MPL was not distributed with this # License, v. 2.0. If a copy of the MPL was not distributed with this
# file, You can obtain one at http://mozilla.org/MPL/2.0/. # file, You can obtain one at http://mozilla.org/MPL/2.0/.
from typing import Dict, TypedDict, Optional from typing import Dict, Optional, TYPE_CHECKING
from io import BytesIO from io import BytesIO
import argparse import argparse
import os.path import os.path
...@@ -70,6 +70,7 @@ async def upload(data: bytes, mimetype: str, filename: str) -> str: ...@@ -70,6 +70,7 @@ async def upload(data: bytes, mimetype: str, filename: str) -> str:
return (await resp.json())["content_uri"] return (await resp.json())["content_uri"]
if TYPE_CHECKING:
class MatrixMediaInfo(TypedDict): class MatrixMediaInfo(TypedDict):
w: int w: int
h: int h: int
...@@ -93,7 +94,7 @@ def convert_image(data: bytes) -> (bytes, int, int): ...@@ -93,7 +94,7 @@ def convert_image(data: bytes) -> (bytes, int, int):
return new_file.getvalue(), w, h return new_file.getvalue(), w, h
async def reupload_document(client: TelegramClient, document: Document) -> MatrixStickerInfo: async def reupload_document(client: TelegramClient, document: Document) -> 'MatrixStickerInfo':
print(f"Reuploading {document.id}", end="", flush=True) print(f"Reuploading {document.id}", end="", flush=True)
data = await client.download_media(document, file=bytes) data = await client.download_media(document, file=bytes)
print(".", end="", flush=True) print(".", end="", flush=True)
...@@ -168,7 +169,7 @@ async def reupload_pack(client: TelegramClient, pack: StickerSetFull) -> None: ...@@ -168,7 +169,7 @@ async def reupload_pack(client: TelegramClient, pack: StickerSetFull) -> None:
except FileNotFoundError: except FileNotFoundError:
pass pass
reuploaded_documents: Dict[int, MatrixStickerInfo] = {} reuploaded_documents: Dict[int, 'MatrixStickerInfo'] = {}
for document in pack.documents: for document in pack.documents:
try: try:
reuploaded_documents[document.id] = already_uploaded[document.id] reuploaded_documents[document.id] = already_uploaded[document.id]
......
0% Chargement en cours ou .
You are about to add 0 people to the discussion. Proceed with caution.
Veuillez vous inscrire ou vous pour commenter