From d95f679f520673188b1025752b4ed50ba014da84 Mon Sep 17 00:00:00 2001 From: Bulat Kurbanov Date: Sun, 6 Aug 2023 15:30:02 +0200 Subject: [PATCH] Fix --- fastapi_file_server/app/serializers.py | 2 +- fastapi_file_server/app/services/storages.py | 10 ++++------ 2 files changed, 5 insertions(+), 7 deletions(-) diff --git a/fastapi_file_server/app/serializers.py b/fastapi_file_server/app/serializers.py index 71d16ec..7726584 100644 --- a/fastapi_file_server/app/serializers.py +++ b/fastapi_file_server/app/serializers.py @@ -10,7 +10,7 @@ class UploadBackend(enum.StrEnum): class Data(TypedDict): - chat_id: str | int + chat_id: int message_id: int diff --git a/fastapi_file_server/app/services/storages.py b/fastapi_file_server/app/services/storages.py index 1a07780..b0f36e9 100644 --- a/fastapi_file_server/app/services/storages.py +++ b/fastapi_file_server/app/services/storages.py @@ -1,4 +1,4 @@ -from typing import AsyncIterator, Optional, Union +from typing import AsyncIterator, Optional import telethon.client import telethon.errors @@ -9,9 +9,7 @@ from core.config import env_config class BaseStorage: - def __init__( - self, channel_id: Union[str, int], app_id: int, api_hash: str, session: str - ): + def __init__(self, channel_id: int, app_id: int, api_hash: str, session: str): self.channel_id = channel_id self.client = telethon.client.TelegramClient(session, app_id, api_hash) @@ -26,7 +24,7 @@ class BaseStorage: file: telethon.hints.FileLike, file_size: int, caption: Optional[str] = None, - ) -> Optional[tuple[Union[str, int], int]]: + ) -> Optional[tuple[int, int]]: try: uploaded_file = await self.client.upload_file(file, file_size=file_size) @@ -87,7 +85,7 @@ class UserStorage(BaseStorage): class BotStorage(BaseStorage): def __init__( self, - channel_id: Union[str, int], + channel_id: int, app_id: int, api_hash: str, session: str,