50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
import selectors
|
|
import socket
|
|
import types
|
|
import time
|
|
HOST = '127.0.0.1' # The server's hostname or IP address
|
|
PORT = 80 # The port used by the server
|
|
sel = selectors.DefaultSelector()
|
|
i = 0
|
|
message = ""
|
|
def service_connection(key, mask, message):
|
|
sock = key.fileobj
|
|
data = key.data
|
|
if mask & selectors.EVENT_READ:
|
|
recv_data = sock.recv(1024) # Should be ready to read
|
|
if recv_data:
|
|
print('received', repr(recv_data),flush=True)
|
|
|
|
if mask & selectors.EVENT_WRITE:
|
|
if message:
|
|
data.outb += message
|
|
message = ""
|
|
print('sending', repr(data.outb))
|
|
sent = sock.send(data.outb.encode('utf-8')) # Should be ready to write
|
|
data.outb = data.outb[sent:]
|
|
|
|
def start_connection(host, port):
|
|
server_addr = (host, port)
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
sock.setblocking(False)
|
|
sock.connect_ex(server_addr)
|
|
events = selectors.EVENT_READ | selectors.EVENT_WRITE
|
|
data = types.SimpleNamespace(outb='')
|
|
sel.register(sock, events, data=data)
|
|
|
|
|
|
start_connection(HOST, PORT)
|
|
while True:
|
|
events = sel.select(timeout=None)
|
|
for key, mask in events:
|
|
if key.data is None:
|
|
pass
|
|
else:
|
|
service_connection(key, mask, message)
|
|
message = ""
|
|
|
|
i += 1
|
|
if i % 10 == 0:
|
|
message = f"hello {i}\n"
|
|
|
|
|