nemunaire/nemubot
1
0
Fork 0
Code Releases Activity

Local client now detects when server close the connection

Browse Source
This commit is contained in:
nemunaire 2017-06-23 21:20:32 +02:00
parent 02838658b0
commit 8b6f72587d
1 changed files with 36 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
nemubot/__init__.py
View File

@ -53,12 +53,20 @@ def attach(pid, socketfile):
sys.stderr.write("\n")
return 1
from select import select
import select
mypoll = select.poll()
mypoll.register(sys.stdin.fileno(), select.POLLIN | select.POLLPRI)
mypoll.register(sock.fileno(), select.POLLIN | select.POLLPRI)
try:
while True:
rl, wl, xl = select([sys.stdin, sock], [], [])
for fd, flag in mypoll.poll():
if flag & (select.POLLERR | select.POLLHUP | select.POLLNVAL):
sock.close()
print("Connection closed.")
return 1
if sys.stdin in rl:
if fd == sys.stdin.fileno():
line = sys.stdin.readline().strip()
if line == "exit" or line == "quit":
return 0
@ -86,8 +94,9 @@ def attach(pid, socketfile):
else:
sock.send(line.encode() + b'\r\n')
if sock in rl:
if fd == sock.fileno():
sys.stdout.write(sock.recv(2048).decode())
except KeyboardInterrupt:
pass
except: