From fe590476a35810602a727ae08fc2c8998c5dfe2f Mon Sep 17 00:00:00 2001
From: Denis Tereshkin <denis@kasan.ws>
Date: Mon, 20 Apr 2020 11:44:03 +0700
Subject: [PATCH] New option: time-delta

---
 qhp-download.py | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/qhp-download.py b/qhp-download.py
index e7340c4..161f0f3 100644
--- a/qhp-download.py
+++ b/qhp-download.py
@@ -18,6 +18,7 @@ def main():
     parser.add_argument('-y', '--symbol', action='store', dest='symbol', help='Symbol to download', required=True)
     parser.add_argument('-f', '--from', action='store', dest='from_', help='Starting date', required=True)
     parser.add_argument('-t', '--to', action='store', dest='to', help='Ending date', required=True)
+    parser.add_argument('-d', '--time-delta', action='store', dest='time_delta', help='Add given time delta (in seconds)', required=False)
 
     args = parser.parse_args()
 
@@ -32,6 +33,10 @@ def main():
     start_time = datetime.datetime.strptime(args.from_, "%Y%m%d")
     end_time = datetime.datetime.strptime(args.to, "%Y%m%d")
 
+    timedelta = datetime.timedelta()
+    if args.time_delta:
+        timedelta = datetime.timedelta(seconds=int(args.time_delta))
+
     rq = {
         "ticker" : symbol,
         "from" : start_time.strftime("%Y-%m-%dT%H:%M:%S"),
@@ -42,11 +47,14 @@ def main():
     s.send_multipart([bytes(json.dumps(rq), "utf-8")])
     parts = s.recv_multipart()
 
+    if parts[0] != b'OK':
+       print("Error:", parts[1])
+
     line_count = 0
-    with open(args.output_file, 'w') as f:
+    with open(args.output_file, 'w', newline='') as f:
         writer = csv.writer(f)
         writer.writerow(['<TICKER>', '<PER>', '<DATE>', '<TIME>', '<OPEN>', '<HIGH>', '<LOW>', '<CLOSE>', '<VOLUME>'])
-        for line in struct.iter_unpack("<qddddQ", parts[0]):
+        for line in struct.iter_unpack("<qddddQ", parts[1]):
             line_count += 1
 
             timestamp = int(line[0])
@@ -55,7 +63,7 @@ def main():
             low = float(line[3])
             close = float(line[4])
             volume = int(line[5])
-            dt = datetime.datetime.utcfromtimestamp(timestamp)
+            dt = datetime.datetime.utcfromtimestamp(timestamp) + timedelta
 
             writer.writerow([symbol, period, dt.strftime('%Y%m%d'), dt.strftime('%H%M%S'), str(open_), str(high), str(low), str(close), str(volume)])