Update

kastaid · Feb 1, 2024 · 8a104be · 8a104be
1 parent 88e5acf
commit 8a104be
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 8 deletions.
diff --git a/getter/core/base_client.py b/getter/core/base_client.py
@@ -57,6 +57,7 @@ def __init__(
     ):
         self._dialogs = []
         self._plugins = {}
+        self.log = LOG
         kwargs["api_id"] = api_id
         kwargs["api_hash"] = api_hash
         kwargs["request_retries"] = 3

diff --git a/getter/core/patched/__init__.py b/getter/core/patched/__init__.py
@@ -5,6 +5,6 @@
 # Please read the GNU Affero General Public License in
 # < https://github.com/kastaid/getter/blob/main/LICENSE/ >.
 
-from .client import Client
+from .client import TelegramClient
 from .conversation import Conversation
 from .message import Message
diff --git a/getter/core/patched/client.py b/getter/core/patched/client.py
@@ -5,7 +5,6 @@
 # Please read the GNU Affero General Public License in
 # < https://github.com/kastaid/getter/blob/main/LICENSE/ >.
 
-import logging
 import typing
 from asyncio import sleep
 from random import randrange
@@ -15,17 +14,12 @@
 from getter.core.constants import MAX_MESSAGE_LEN
 from getter.core.functions import get_chat_id, get_text, get_user
 from getter.core.patcher import patch, patchable
-from getter.logger import LOG
 
 delattr(fun.account, "DeleteAccountRequest")
 
 
 @patch(telethon.client.telegramclient.TelegramClient)
-class Client(telethon.client.telegramclient.TelegramClient):
-    @patchable(True)
-    def log(self) -> logging:
-        return LOG
-
+class TelegramClient:
     @patchable()
     async def get_id(self, entity: hints.EntityLike) -> int:
         try: