gitea/mcstatus
1
0
Fork 0
mirror of https://github.com/Dinnerbone/mcstatus.git synced 2026-04-06 12:01:24 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #35 from zeroflow/connection

Browse Source 
Implicitly close sockets in destructor methods
This commit is contained in:
Dinner Bone
2017-04-01 17:19:13 +02:00
committed by GitHub
parent 9c4596e807 3ceb1e1135
commit 1ac30d15fd
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
mcstatus/protocol/connection.py
View File

@@ -149,6 +149,9 @@ class TCPSocketConnection(Connection):
def write(self, data): def write(self, data):
self.socket.send(data) self.socket.send(data)
def __del__(self):
self.socket.close()
class UDPSocketConnection(Connection): class UDPSocketConnection(Connection):
def __init__(self, addr, timeout=3): def __init__(self, addr, timeout=3):
@@ -175,4 +178,7 @@ class UDPSocketConnection(Connection):
def write(self, data): def write(self, data):
if isinstance(data, Connection): if isinstance(data, Connection):
data = bytearray(data.flush()) data = bytearray(data.flush())
self.socket.sendto(data, self.addr) self.socket.sendto(data, self.addr)
def __del__(self):
self.socket.close()