LonamiWebs/Telethon
1
1
Fork 0
mirror of https://github.com/LonamiWebs/Telethon.git synced 2024-11-25 10:53:44 +03:00
Code Issues Packages Projects Releases Wiki Activity

Use memoryview() on TcpClient.write() instead slicing the data

Browse Source
This commit is contained in:
Lonami Exo 2017-06-09 11:50:30 +02:00
parent c6645a555d
commit ba5bfa105f
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
telethon/network/tcp_client.py
View File

@ -50,9 +50,10 @@ class TcpClient:
# Ensure that only one thread can send data at once
with self._lock:
total_sent = 0
while total_sent < len(data):
sent = self._socket.send(data[total_sent:])
view = memoryview(data)
total_sent, total = 0, len(data)
while total_sent < total:
sent = self._socket.send(view[total_sent:])
if sent == 0:
raise ConnectionResetError(
'The server has closed the connection.')