# encoding: UTF-8

from strategyEngine import *
from backtestingEngine import *
from demoStrategy import SimpleEmaStrategy
import decimal



# 回测脚本    
if __name__ == '__main__':
    #symbol = 'IF1506'
    symbol = 'a'
    
    # 创建回测引擎
    be = BacktestingEngine()
    
    # 创建策略引擎对象
    se = StrategyEngine(be.eventEngine, be, backtesting=True)
    be.setStrategyEngine(se)
    
    # 初始化回测引擎
    #be.connectMongo()
    be.connectMysql()
    #be.loadMongoDataHistory(symbol, datetime(2015,5,1), datetime.today())
    #be.loadMongoDataHistory(symbol, datetime(2012,1,9), datetime(2012,1,14))
    be.loadMysqlDataHistory(symbol, datetime(2012,1,9), datetime(2012,1,20))

    # 创建策略对象
    setting = {}
    setting['fastAlpha'] = 0.2
    setting['slowAlpha'] =  0.05
    #setting['startDate'] = datetime(year=2015, month=5, day=20)
    setting['startDate'] = datetime(year=2012, month=1, day=9)

    se.createStrategy(u'EMA演示策略', symbol, SimpleEmaStrategy, setting)

    # 启动所有策略
    se.startAll()

    # 开始回测
    be.startBacktesting()