vnpy/vn.datayes/prepare.sh
2015-08-14 16:26:41 +08:00

84 lines
1.6 KiB
Bash
Executable File

#!/bin/bash
dir_n=names
if [ ! -d $dir_n ]; then
mkdir $dir_n
fi
dir_c=config
if [ ! -d $dir_c ]; then
mkdir $dir_c
fi
echo [vn-past]: Configuration starts.
python - << EOF
from storage import *
import pandas as pd
import os
dc = DBConfig()
api = PyApi(Config())
mc = MongodController(dc, api)
mc._collNames['equTicker'] = mc._allEquTickers()
print '[MONGOD]: Equity tickers collected.'
mc._collNames['secID'] = mc._allSecIds()
print '[MONGOD]: Security IDs collected.'
mc._collNames['futTicker'] = mc._allFutTickers()
print '[MONGOD]: Future tickers collected.'
mc._collNames['optTicker'] = mc._allOptTickers()
print '[MONGOD]: Option symbols collected.'
mc._collNames['fudTicker'] = mc._allFudTickers()
print '[MONGOD]: Mutual Fund symbols collected.'
mc._collNames['idxTicker'] = mc._allIdxTickers()
print '[MONGOD]: Index symbols collected.'
mc._ensure_index()
EOF
echo [vn-past]: Configuration finished.
echo [vn-past]: Selected databases:
cd ./names
ls -l
echo [vn-past]: Prepare to construct[c]/update[u] databases...
read -r -p "[vn-past]: Waiting for orders[c/u]: " response
if [[ $response =~ ^([uU][pP][dD][aA][tT][eE]|[uU])$ ]]
then
echo [API]: Prepare to update data...
read -r -p "[API]: Confirm? [y/N] " response
if [[ $response =~ ^([yY][eE][sS]|[yY])$ ]]
then
cd -
chmod +x update.sh
./update.sh
else
echo [vn-past]: Do not update.
:
fi
else
echo [API]: Prepare to download Bars...
read -r -p "[API]: Confirm? [y/N] " response
if [[ $response =~ ^([yY][eE][sS]|[yY])$ ]]
then
cd -
chmod +x download.sh
./download.sh
else
echo [vn-past]: Do not download.
:
fi
fi
echo [vn-past]: Finished.