diff --git a/setstatus.py b/setstatus.py index 2993689..0f13580 100755 --- a/setstatus.py +++ b/setstatus.py @@ -5,13 +5,13 @@ # date: 26.07.2019 # email: berhsi@web.de -# client, who connects to the statusserver at port 10001 to update the krautspace -# door status. If no status is given as argument, he reads from stdin until -# input is 0 or 1. +# client, who connects to the statusserver at port 10001 to update the +# krautspace door status. If no status is given as argument, he reads from +# stdin until input is 0 or 1. import socket import ssl -from sys import exit, argv, byteorder +from sys import exit, argv def check_arguments(argv): @@ -42,7 +42,7 @@ def read_argument(): ''' status = None - while status == None: + while status is None: buf = input('Enter new status (0/1): ') if buf == '0' or buf == '1': status = bytes([int(buf)]) @@ -50,7 +50,7 @@ def read_argument(): return status -def main(*status): +def main(): HOST = 'localhost' PORT = 10001 @@ -58,34 +58,26 @@ def main(*status): CLIENT_CERT = './certs/client.crt' CLIENT_KEY = './certs/client.key' SERVER_CERT = './certs/server.crt' - BOM = byteorder STATUS = None RESPONSE = None - if status: - STATUS = status[0] - STATUS = bytes([STATUS]) - print('Status: {}'.format(STATUS)) - else: - if len(argv) == 1: - STATUS = read_argument() - else: - STATUS = check_arguments(argv) - if STATUS == None: - STATUS = read_argument() + STATUS = check_arguments(argv) + while STATUS is None: + STATUS = read_argument() - context = ssl.create_default_context(ssl.Purpose.SERVER_AUTH, cafile = SERVER_CERT) + context = ssl.create_default_context(ssl.Purpose.SERVER_AUTH, + cafile=SERVER_CERT) context.options &= ~ssl.PROTOCOL_TLS context.verify_mode = ssl.CERT_OPTIONAL # context.set_ciphers('HIGHT:!aNULL:!RC4:!DSS') - context.load_cert_chain(certfile = CLIENT_CERT, keyfile = CLIENT_KEY) + context.load_cert_chain(certfile=CLIENT_CERT, keyfile=CLIENT_KEY) print('SSL context created') with socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0) as mySocket: print('Socket created') try: - conn = context.wrap_socket(mySocket, server_side = False, \ - server_hostname = SERVER_NAME) + conn = context.wrap_socket(mySocket, server_side=False, + server_hostname=SERVER_NAME) print('Connection wrapped with ssl.context') conn.settimeout(5.0) except Exception as e: