diff --git a/qhp-download.py b/qhp-download.py index 83c789f..c875d55 100644 --- a/qhp-download.py +++ b/qhp-download.py @@ -62,6 +62,17 @@ class BarAggregator: self.low = min(low, self.low) self.close = close self.volume += volume + return None + + def get_bar(self): + b_open = self.open_ + b_high = self.high + b_low = self.low + b_close = self.close + b_volume = self.volume + b_timestamp = self.timestamp + + return (b_timestamp, b_open, b_high, b_low, b_close, b_volume) def main(): parser = argparse.ArgumentParser(description='QHP client') @@ -91,6 +102,10 @@ def main(): if args.time_delta: timedelta = datetime.timedelta(seconds=int(args.time_delta)) + agg = None + if args.rescale: + agg = BarAggregator(int(args.rescale)) + rq = { "ticker" : symbol, "from" : start_time.strftime("%Y-%m-%dT%H:%M:%S"), @@ -101,15 +116,16 @@ def main(): s.send_multipart([bytes(json.dumps(rq), "utf-8")]) parts = s.recv_multipart() + print(parts[0]) if parts[0] != b'OK': print("Error:", parts[1]) + line_count = 0 with open(args.output_file, 'w', newline='') as f: writer = csv.writer(f) writer.writerow(['', '', '', '