Added broken pipe handling.

This commit is contained in:
Greenscreener 2019-09-24 21:13:49 +02:00
parent 3a7894e052
commit df59ddfc34

View File

@ -1,7 +1,8 @@
import socket import socket
import sys import sys
ledNames = [
ledNames = [
"power", "power",
"lan0", "lan0",
"lan1", "lan1",
@ -37,7 +38,8 @@ def setLed(index, color):
finally: finally:
file.close() file.close()
def handleData(data):
def handleData(data):
databyline = data.split("\n") databyline = data.split("\n")
try: try:
for i in range(len(databyline)): for i in range(len(databyline)):
@ -84,13 +86,15 @@ while True:
allData+=data.decode('utf-8') allData+=data.decode('utf-8')
if allData.find('\n\n') > -1: if allData.find('\n\n') > -1:
break break
eprint('all data: {}'.format(allData))\ eprint('all data: {}'.format(allData))
if not handleData(allData) is 0: if not handleData(allData) is 0:
conn.sendall("error".encode('utf-8')) conn.sendall("error".encode('utf-8'))
else: else:
conn.sendall("OK".encode('utf-8')) conn.sendall("OK".encode('utf-8'))
except BrokenPipeError:
eprint("connection from {} has broken a pipe.".format(client_address))
conn.close()
finally: finally:
# Clean up the connection # Clean up the connection
conn.close() conn.close()