50ETF期权波动率交易样例 ETF Opt Volatility Trade

# coding:utf-8
#!/usr/bin/env python
from PoboAPI import *
import datetime
import time
import numpy as np
#用poboquant python实现，在poboquant上运行，如果有问题 可加群 726895887 咨询
#这是一个卖空波动率的50ETF期权策略样例,在日线级别测试
#开始时间，用于初始化一些参数
def OnStart(context) :
    print "I\'m starting..."
    #设定一个全局变量品种
    g.code = "510050.SHSE"
    #订阅实时数据，用于驱动OnQuote事件
    SubscribeQuote(g.code)
    #订阅K线数据，用于驱动OnBar事件
    SubscribeBar(g.code, BarType.Day)
    #登录交易账号，需在主页用户管理中设置账号，并把证券测试替换成您的账户名称
    context.myacc = None
    if context.accounts.has_key("回测期权") :
        print "登录交易账号[回测期权]"
        if context.accounts["回测期权"].Login() :
            context.myacc = context.accounts["回测期权"]
def Getop(code):
    dyndata = GetQuote(code)
    now1 = dyndata.now
    now50 = round(now1,1) + 0.05
    cutime = GetCurrentTime()
    if cutime.day >15 and cutime.month<12:
        tim = cutime.month + 1
        month_time = datetime.datetime(month=tim, year=cutime.year,day = 20)
    elif cutime.day >15 and cutime.month==12:
        tim = 1
        yea = cutime.year + 1
        month_time = datetime.datetime(month=tim, year=yea,day = 20) 
    else:
        month_time = cutime
    atmopc = GetAtmOptionContract(code,month_time,now50,0)
    atmopp = GetAtmOptionContract(code,month_time,now50,1)
    return atmopc,atmopp
  
def stime(op):
    info1 = GetContractInfo(op)
    kill = info1['行权到期日']
    print str(type(kill))
    cutime = GetCurrentTime()
    if cutime.day <10:
        cutim = str(cutime.year) + str(cutime.month) + '0' +str(cutime.day)
    elif cutime.month <10:
        cutim = str(cutime.year) + '0' + str(cutime.month) +str(cutime.day)
    else:
        cutim = str(cutime.year) + str(cutime.month) +str(cutime.day)
    #cu = int(cutim)
    n = (kill - cutime.date()).days #这里的到期日计算进行了更新 于 20190219
    return n
    
#实时行情事件，当有新行情出现时调用该事件
def OnQuote(context, code) :
    #过滤掉不需要的行情通知
    if code != g.code :
        return

    #获取最新行情
    dyndata = GetQuote(g.code)
    if dyndata :
        #.now指最新价，详细属性见API文档i
        now1 = dyndata.now
        #打印最新价
        log.info("最新价: " + str(dyndata.now) + str(dyndata.time))
    posi = context.myacc.GetPositions()
    print len(posi)
    
    b = CreateCalcObj()
    option = PBObj()
    option.EndDate = GetCurrentTime()
    option.Count = 60
    #获取最近60天的收盘价列表
    klist = GetHisDataByField(g.code, BarType.Day, "close", option)
    if len(klist)>0:
        Kl = np.array(klist, dtype=np.double)
        c=b.GetVolatility(Kl)
        print "历史波动率"
        print c

    if len(posi) == 0 and c<0.35:
        opc,opp = Getop(g.code)
        print str(opc)
        dync = GetQuote(opc)
        dynp = GetQuote(opp)
        if dync != None:
            log.info("最新价2: " + str(dync.now))
            log.info("最新价3: " + str(dynp.now))
            context.myacc.InsertOrder(opc, BSType.SellOpen, dync.now, 100)
            context.myacc.InsertOrder(opp, BSType.SellOpen, dynp.now, 100)
    elif len(posi) >1:
        print len(posi)
        opcode1 = posi[0].contract
        opcode2 = posi[1].contract
        dyn1 = GetQuote(opcode1)
        dyn2 = GetQuote(opcode2)
        info1 = GetContractInfo(opcode1)
        pr1 = info1['行权价格']
        ki1 = info1['行权到期日']
        info2 = GetContractInfo(opcode2)
        pr2 = info2['行权价格']
        sy = int(stime(opcode1))
        print str(pr1) + '行权价格'
        print str(pr2) + '行权价格2'
        print sy
        if sy<3:
            context.myacc.InsertOrder(opcode1, BSType.BuyClose, dyn1.now, 100)
            context.myacc.InsertOrder(opcode2, BSType.BuyClose, dyn2.now, 100)
        elif now1 >= pr1 or now1 <= (pr1-0.15):
            context.myacc.InsertOrder(opcode1, BSType.BuyClose, dyn1.now, 100)
            context.myacc.InsertOrder(opcode2, BSType.BuyClose, dyn2.now, 100)
#委托回报事件，当有委托回报时调用
def OnOrderChange(context, AccountName, order) :
    #打印委托信息，id是编号，volume是数量，详细见API文档
    print "委托编号： " + order.id + "   账号名称： " + AccountName
    print "Vol: " + str(order.volume) + " Price: " + str(order.price)