diff --git a/client-clear.py b/setstatus.py similarity index 89% rename from client-clear.py rename to setstatus.py index 8d70dfa..4f7d800 100755 --- a/client-clear.py +++ b/setstatus.py @@ -49,18 +49,23 @@ def read_argument(): return status -def main(): +def main(*status): - HOST = 'nr18.space' + HOST = 'localhost' PORT = 10001 BOM = byteorder STATUS = None RESPONSE = None - if check_arguments(argv) == None: - STATUS = read_argument() + if status: + STATUS = status[0] + STATUS = bytes([STATUS]) + print('Status: {}'.format(STATUS)) else: - STATUS = check_arguments(argv) + if len(argv) == 1: + STATUS = read_argument() + else: + STATUS = check_arguments(argv) with socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0) as mySocket: print('Socket created')