mirror of
https://github.com/saymrwulf/zipline.git
synced 2026-05-14 20:58:10 +00:00
39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
import datetime
|
|
import sys
|
|
import zipline.util as qutil
|
|
from zipline.finance.data import DataLoader
|
|
|
|
def print_usage():
|
|
print """
|
|
Usage is:
|
|
python loaddata.py (pt | lt | lh | ld | ei | bm | si | help)
|
|
|
|
pt - purge trade collection from the db
|
|
lt - load trades (minute bars) to the db
|
|
lh - load trades (hour bars) to the db
|
|
ld - load trades (daily close) to the db
|
|
ei - ensure all indexes on all collections in tick and algo db
|
|
tr - load treasury rates
|
|
bm - load benchmark data
|
|
si - load security info (sid, symbol, qualifier)
|
|
help - display this message
|
|
"""
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
if len(sys.argv) == 2:
|
|
qutil.configure_logging()
|
|
operation = sys.argv[1]
|
|
if(operation not in['pt','lt','lh','ld','ei','si', 'tr','bm'] or operation == 'help'):
|
|
print_usage()
|
|
else:
|
|
ts = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
|
|
pidfile = "/tmp/loaddata-{stamp}.pid".format(stamp=ts)
|
|
daemon = DataLoader(pidfile,operation)
|
|
qutil.LOGGER.info("DataLoader starting.")
|
|
daemon.run()
|
|
sys.exit(0)
|
|
else:
|
|
print_usage()
|
|
sys.exit(2)
|