From 2cd3d64bb89981b5910a3e2a8388292ccbd3a671 Mon Sep 17 00:00:00 2001
From: Denis <bestloveapples@gmail.com>
Date: Thu, 24 Jul 2025 01:50:25 +0300
Subject: [PATCH] =?UTF-8?q?=D0=9D=D0=B5=D0=B1=D0=BE=D0=BB=D1=8C=D1=88?=
 =?UTF-8?q?=D0=B0=D1=8F=20=D0=BE=D0=BF=D1=82=D0=B8=D0=BC=D0=B8=D0=B7=D0=B0?=
 =?UTF-8?q?=D1=86=D0=B8=D1=8F?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 maxapi/connection/base.py | 24 +++++++++++++-----------
 1 file changed, 13 insertions(+), 11 deletions(-)

diff --git a/maxapi/connection/base.py b/maxapi/connection/base.py
index 33214d1..98c49be 100644
--- a/maxapi/connection/base.py
+++ b/maxapi/connection/base.py
@@ -5,12 +5,13 @@ from typing import TYPE_CHECKING
 from uuid import uuid4
 
 import aiofiles
-import aiohttp
-
 import puremagic
+
 from pydantic import BaseModel
+from aiohttp import ClientSession, ClientConnectionError, FormData
 
 from ..exceptions.invalid_token import InvalidToken
+from ..exceptions.max import MaxConnection
 
 from ..types.errors import Error
 from ..enums.http_method import HTTPMethod
@@ -37,7 +38,7 @@ class BaseConnection:
 
     def __init__(self):
         self.bot: 'Bot' = None
-        self.session: aiohttp.ClientSession = None
+        self.session: ClientSession = None
 
     async def request(
             self,
@@ -64,7 +65,7 @@ class BaseConnection:
         """
         
         if not self.bot.session:
-            self.bot.session = aiohttp.ClientSession(self.bot.API_URL)
+            self.bot.session = ClientSession(self.bot.API_URL)
 
         try:
             r = await self.bot.session.request(
@@ -72,10 +73,11 @@ class BaseConnection:
                 url=path.value if isinstance(path, ApiPath) else path, 
                 **kwargs
             )
-        except aiohttp.ClientConnectorDNSError as e:
-            return logger_connection.error(f'Ошибка при отправке запроса: {e}')
+        except ClientConnectionError as e:
+            raise MaxConnection(f'Ошибка при отправке запроса: {e}')
         
         if r.status == 401:
+            await self.bot.session.close()
             raise InvalidToken('Неверный токен!')
 
         if not r.ok:
@@ -122,7 +124,7 @@ class BaseConnection:
         basename = os.path.basename(path)
         _, ext = os.path.splitext(basename)
 
-        form = aiohttp.FormData()
+        form = FormData()
         form.add_field(
             name='data',
             value=file_data,
@@ -130,7 +132,7 @@ class BaseConnection:
             content_type=f"{type.value}/{ext.lstrip('.')}"
         )
 
-        async with aiohttp.ClientSession() as session:
+        async with ClientSession() as session:
             response = await session.post(
                 url=url, 
                 data=form
@@ -168,7 +170,7 @@ class BaseConnection:
 
         basename = f'{uuid4()}{ext}'
 
-        form = aiohttp.FormData()
+        form = FormData()
         form.add_field(
             name='data',
             value=buffer,
@@ -176,7 +178,7 @@ class BaseConnection:
             content_type=mime_type
         )
 
-        async with aiohttp.ClientSession() as session:
+        async with ClientSession() as session:
             response = await session.post(
                 url=url, 
                 data=form
@@ -203,7 +205,7 @@ class BaseConnection:
             'Authorization': f'Bearer {token}'
         }
 
-        async with aiohttp.ClientSession() as session:
+        async with ClientSession() as session:
             async with session.get(url, headers=headers) as response:
                 
                 if response.status == 200: