senset/vnpy
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[Add]adjust timestamp of bar data from RQData, close #1573

Browse Source
This commit is contained in:
vn.py 2019-04-30 14:29:35 +08:00
parent dbf37c629a
commit 791fb0d840
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
vnpy/trader/rqdata.py
View File

@ -16,6 +16,12 @@ INTERVAL_VT2RQ = {
Interval.DAILY: "1d",
}
INTERVAL_ADJUSTMENT_MAP = {
Interval.MINUTE: timedelta(minutes=1),
Interval.HOUR: timedelta(hours=1),
Interval.DAILY: timedelta() # no need to adjust for daily bar
}
class RqdataClient:
"""
@ -102,7 +108,11 @@ class RqdataClient:
if not rq_interval:
return None
end += timedelta(1) # For querying night trading period data
# For adjust timestamp from bar close point (RQData) to open point (VN Trader)
adjustment = INTERVAL_ADJUSTMENT_MAP[interval]
# For querying night trading period data
end += timedelta(1)
df = rqdata_get_price(
rq_symbol,
@ -118,7 +128,7 @@ class RqdataClient:
symbol=symbol,
exchange=exchange,
interval=interval,
datetime=row.name.to_pydatetime(),
datetime=row.name.to_pydatetime() - adjustment,
open_price=row["open"],
high_price=row["high"],
low_price=row["low"],