GeezerVison is showing github.com — vscode-python/pythonFiles/testing_tools/socket_manager.py at main · dafeda/vscode-python

Expand file tree

socket_manager.py

Copy path

More file actions

Latest commit

History

50 lines (39 loc) · 1.22 KB

socket_manager.py

Copy path

File metadata and controls

50 lines (39 loc) · 1.22 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

# Licensed under the MIT License.

import socket

import sys

class SocketManager(object):

"""Create a socket and connect to the given address.

The address is a (host: str, port: int) tuple.

Example usage:

```

with SocketManager(("localhost", 6767)) as sock:

request = json.dumps(payload)

result = s.socket.sendall(request.encode("utf-8"))

```

"""

def __init__(self, addr):

self.addr = addr

self.socket = None

def __enter__(self):

return self.connect()

def __exit__(self, *_):

self.close()

def connect(self):

self.socket = socket.socket(

socket.AF_INET, socket.SOCK_STREAM, socket.IPPROTO_TCP

)

if sys.platform == "win32":

addr_use = socket.SO_EXCLUSIVEADDRUSE

else:

addr_use = socket.SO_REUSEADDR

self.socket.setsockopt(socket.SOL_SOCKET, addr_use, 1)

self.socket.connect(self.addr)

return self

def close(self):

if self.socket:

try:

self.socket.shutdown(socket.SHUT_RDWR)

except Exception:

pass

self.socket.close()