style(mypy): apply linter changes

This commit is contained in:
mpa 2020-04-22 10:00:59 +04:00
parent 518e88df97
commit 23799929b8
5 changed files with 26 additions and 17 deletions

View file

@ -110,7 +110,7 @@ from ..types import (
WebhookInfo,
)
from .session.aiohttp import AiohttpSession
from .session.base import BaseSession, PT
from .session.base import PT, BaseSession
T = TypeVar("T")
@ -121,7 +121,10 @@ class Bot(ContextInstanceMixin["Bot"]):
"""
def __init__(
self, token: str, session: Optional[BaseSession[PT]] = None, parse_mode: Optional[str] = None
self,
token: str,
session: Optional[BaseSession[PT]] = None,
parse_mode: Optional[str] = None,
) -> None:
validate_token(token)

View file

@ -4,18 +4,18 @@ from typing import (
Any,
AsyncGenerator,
Callable,
Optional,
Iterable,
TypeVar,
Type,
Tuple,
Dict,
Iterable,
List,
Optional,
Tuple,
Type,
TypeVar,
Union,
cast,
)
from aiohttp import ClientSession, ClientTimeout, FormData, BasicAuth, TCPConnector
from aiohttp import BasicAuth, ClientSession, ClientTimeout, FormData, TCPConnector
from aiogram.api.methods import Request, TelegramMethod
@ -60,8 +60,10 @@ def _prepare_connector(chain_or_plain: _ProxyType) -> Tuple[Type["TCPConnector"]
if isinstance(chain_or_plain, str) or (
isinstance(chain_or_plain, tuple) and len(chain_or_plain) == 2
):
chain_or_plain = cast(_ProxyBasic, chain_or_plain)
return ProxyConnector, _retrieve_basic(chain_or_plain)
chain_or_plain = cast(_ProxyChain, chain_or_plain)
infos: List[ProxyInfo] = []
for basic in chain_or_plain:
infos.append(ProxyInfo(**_retrieve_basic(basic)))
@ -73,7 +75,7 @@ class AiohttpSession(BaseSession[_ProxyType]):
def __init__(
self,
api: TelegramAPIServer = PRODUCTION,
json_loads: Optional[Callable[[str, ...], Any]] = None,
json_loads: Optional[Callable[..., Any]] = None,
json_dumps: Optional[Callable[..., str]] = None,
proxy: Optional[_ProxyType] = None,
):

View file

@ -4,7 +4,7 @@ import abc
import datetime
import json
from types import TracebackType
from typing import Any, AsyncGenerator, Callable, Optional, Type, TypeVar, Union, Generic
from typing import Any, AsyncGenerator, Callable, Generic, Optional, Type, TypeVar, Union
from aiogram.utils.exceptions import TelegramAPIError
@ -19,7 +19,7 @@ class BaseSession(abc.ABC, Generic[PT]):
def __init__(
self,
api: Optional[TelegramAPIServer] = None,
json_loads: Optional[Callable[[str, ...], Any]] = None,
json_loads: Optional[Callable[..., Any]] = None,
json_dumps: Optional[Callable[..., str]] = None,
proxy: Optional[PT] = None,
) -> None:

View file

@ -61,8 +61,8 @@ class TestAiohttpSession:
session = AiohttpSession(
proxy=[
"socks4://proxy.url/",
'socks5://proxy.url/',
'http://user:password@127.0.0.1:3128'
"socks5://proxy.url/",
"http://user:password@127.0.0.1:3128",
]
)
@ -72,8 +72,12 @@ class TestAiohttpSession:
assert isinstance(session._connector_init["proxy_infos"], list)
assert isinstance(session._connector_init["proxy_infos"][0], aiohttp_socks.ProxyInfo)
assert session._connector_init["proxy_infos"][0].proxy_type is aiohttp_socks.ProxyType.SOCKS4
assert session._connector_init["proxy_infos"][1].proxy_type is aiohttp_socks.ProxyType.SOCKS5
assert (
session._connector_init["proxy_infos"][0].proxy_type is aiohttp_socks.ProxyType.SOCKS4
)
assert (
session._connector_init["proxy_infos"][1].proxy_type is aiohttp_socks.ProxyType.SOCKS5
)
assert session._connector_init["proxy_infos"][2].proxy_type is aiohttp_socks.ProxyType.HTTP
aiohttp_session = await session.create_session()

View file

@ -1,6 +1,6 @@
import types
import datetime
from typing import AsyncContextManager, AsyncGenerator, Any
import types
from typing import Any, AsyncContextManager, AsyncGenerator
import pytest