python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Make endless long-polling

Browse source
This commit is contained in:
Alex Root Junior 2021-06-19 01:16:51 +03:00
parent 5296724a0f
commit ac1f0efde8
7 changed files with 245 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

18
aiogram/client/session/aiohttp.py
View file

@ -1,5 +1,6 @@
from __future__ import annotations
import asyncio
from typing import (
TYPE_CHECKING,
Any,
@ -14,11 +15,12 @@ from typing import (
cast,
)
from aiohttp import BasicAuth, ClientSession, FormData, TCPConnector
from aiohttp import BasicAuth, ClientError, ClientSession, FormData, TCPConnector
from aiogram.methods import Request, TelegramMethod
from ...methods.base import TelegramType
from ...utils.exceptions.network import NetworkError
from .base import UNSET, BaseSession
if TYPE_CHECKING: # pragma: no cover
@ -139,11 +141,15 @@ class AiohttpSession(BaseSession):
url = self.api.api_url(token=bot.token, method=request.method)
form = self.build_form_data(request)
async with session.post(
url, data=form, timeout=self.timeout if timeout is None else timeout
) as resp:
raw_result = await resp.text()
try:
async with session.post(
url, data=form, timeout=self.timeout if timeout is None else timeout
) as resp:
raw_result = await resp.text()
except asyncio.TimeoutError:
raise NetworkError(method=call, message="Request timeout error")
except ClientError as e:
raise NetworkError(method=call, message=f"{type(e).__name__}: {e}")
response = self.check_response(method=call, status_code=resp.status, content=raw_result)
return cast(TelegramType, response.result)