75 lines
2.0 KiB
Python
75 lines
2.0 KiB
Python
import selectors
|
|
import socket
|
|
import types
|
|
import time
|
|
import sys
|
|
import select
|
|
import tty
|
|
import termios
|
|
def isData():
|
|
return select.select([sys.stdin], [], [], 0) == ([sys.stdin], [], [])
|
|
|
|
|
|
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)
|
|
|
|
old_settings = termios.tcgetattr(sys.stdin)
|
|
|
|
try:
|
|
tty.setcbreak(sys.stdin.fileno())
|
|
|
|
temp_message = ""
|
|
while 1:
|
|
events = sel.select(timeout=None)
|
|
for key, mask in events:
|
|
if key.data is None:
|
|
pass
|
|
else:
|
|
service_connection(key, mask, message)
|
|
message = ""
|
|
|
|
if isData():
|
|
c = sys.stdin.read(1)
|
|
|
|
if c == '\x1b': # x1b is ESC
|
|
break
|
|
temp_message = temp_message+c
|
|
if c == '\n':
|
|
message = temp_message
|
|
temp_message = ""
|
|
finally:
|
|
termios.tcsetattr(sys.stdin, termios.TCSADRAIN, old_settings)
|
|
|
|
|
|
|
|
|