Skip to content

Commit 7bad2f7

Browse files
committed
Add Aiohttp Logging
1 parent 8459357 commit 7bad2f7

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

discord/http.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -410,9 +410,24 @@ async def close(self) -> None:
410410
# login management
411411

412412
async def static_login(self, token: str) -> user.User:
413+
import logging
414+
415+
async def on_request_start(session, context, params: aiohttp.TraceRequestStartParams):
416+
# breakpoint()
417+
logging.getLogger('aiohttp.client').debug(f'Starting request <{params}> <{session}> <{context}>')
418+
419+
async def on_request_chunk_sent(session, context, params: aiohttp.TraceRequestChunkSentParams):
420+
with open("output.txt", "a") as file:
421+
file.write(str(params.chunk))
422+
# logging.getLogger('aiohttp.client').debug(f'Sent Chunk <{params}>')
423+
424+
trace_config = aiohttp.TraceConfig()
425+
trace_config.on_request_start.append(on_request_start)
426+
trace_config.on_request_chunk_sent.append(on_request_chunk_sent)
427+
413428
# Necessary to get aiohttp to stop complaining about session creation
414429
self.__session = aiohttp.ClientSession(
415-
connector=self.connector, ws_response_class=DiscordClientWebSocketResponse
430+
connector=self.connector, ws_response_class=DiscordClientWebSocketResponse, trace_configs=[trace_config]
416431
)
417432
old_token = self.token
418433
self.token = token

0 commit comments

Comments
 (0)