From cd3dc7c3cd81bbd3b99ca49234bc06ce67ea7fe9 Mon Sep 17 00:00:00 2001 From: Kylmakalle Date: Sun, 8 Oct 2023 15:59:53 +0300 Subject: [PATCH] Typing improvements --- aiogram/dispatcher/dispatcher.py | 4 ++-- aiogram/webhook/aiohttp_server.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/aiogram/dispatcher/dispatcher.py b/aiogram/dispatcher/dispatcher.py index ad4cefd0..e3eb654a 100644 --- a/aiogram/dispatcher/dispatcher.py +++ b/aiogram/dispatcher/dispatcher.py @@ -6,7 +6,7 @@ import signal import warnings from asyncio import CancelledError, Event, Future, Lock from contextlib import suppress -from typing import Any, AsyncGenerator, Dict, List, Optional, Union +from typing import Any, AsyncGenerator, Dict, List, Optional, Set, Union from .. import loggers from ..client.bot import Bot @@ -95,7 +95,7 @@ class Dispatcher(Router): self._running_lock = Lock() self._stop_signal: Optional[Event] = None self._stopped_signal: Optional[Event] = None - self._handle_update_tasks = set() + self._handle_update_tasks: Set[asyncio.Task[Any]] = set() def __getitem__(self, item: str) -> Any: return self.workflow_data[item] diff --git a/aiogram/webhook/aiohttp_server.py b/aiogram/webhook/aiohttp_server.py index 2378d704..7caa0e15 100644 --- a/aiogram/webhook/aiohttp_server.py +++ b/aiogram/webhook/aiohttp_server.py @@ -2,7 +2,7 @@ import asyncio import secrets from abc import ABC, abstractmethod from asyncio import Transport -from typing import Any, Awaitable, Callable, Dict, Optional, Tuple, cast +from typing import Any, Awaitable, Callable, Dict, Optional, Set, Tuple, cast from aiohttp import MultipartWriter, web from aiohttp.abc import Application @@ -98,7 +98,7 @@ class BaseRequestHandler(ABC): self.dispatcher = dispatcher self.handle_in_background = handle_in_background self.data = data - self._background_feed_update_tasks = set() + self._background_feed_update_tasks: Set[asyncio.Task[Any]] = set() def register(self, app: Application, /, path: str, **kwargs: Any) -> None: """