diff --git a/qhp-download.py b/qhp-download.py index c875d55..2071707 100644 --- a/qhp-download.py +++ b/qhp-download.py @@ -113,43 +113,52 @@ def main(): "timeframe" : period } + print("Sending request:", rq) s.send_multipart([bytes(json.dumps(rq), "utf-8")]) - parts = s.recv_multipart() + print("Awaiting response") + resp = s.recv() - print(parts[0]) - if parts[0] != b'OK': - print("Error:", parts[1]) + print(resp) + if resp != b'OK': + errmsg = s.recv_string() + print("Error:", errmsg) line_count = 0 with open(args.output_file, 'w', newline='') as f: writer = csv.writer(f) writer.writerow(['', '', '', '