基于期限利差策略的债券回测分析

在上篇文章期限利差怎么看文章中详细介绍了期限利差策略的内容,本次将介绍代码实现,重要的是理解整个交易的逻辑,为什么要这样去做?对以后写代码会有很大帮助。

首先导入相关库:

from typing import List, Dict
import numpy as np
import datetime

from base.constant import Direction, Offset, BusinessType
from bond.base.utility import ArrayManager
from base.constant import Direction, Offset, BusinessType
from bond.base.object import TickData, OrderData, TradeData, BarData, QuoteRequest, FactorData
from bond.base.template import StrategyTemplate
from bond.base.inter_base import BaseInterface
from bond.data.data_http import HttpDataHandler
from bond.base.inter_kslib import KSLibInterface, KSQuoteType
from util.common_exception import CommonException

导入typing、numpy、datetime库,导入内部债券相关基本库做准备。

接下来定义好相关变量,所有都有说明:

class PairTradingStrategy(StrategyTemplate):
    #合约1为长期债券,合约2为短期债券
    variety_list = ['leg1_symbol', 'leg2_symbol']#用于套利的两个合约的参数名称
    parameters = ['up_threshold', 'up_threshold_sell','down_threshold','down_threshold_sell', 'spread_length']#可变输入参数
    variables = []

    up_threshold = 0.8#利差突破债券上限的分位数,买入债券1,卖出债券2
    up_threshold_sell=0.6#利差从债券上限回落的分位数,平仓
    down_threshold = 0.2#利差突破债券下限的分位数,买入债券2,卖出债券1
    down_threshold_sell=0.45#利差从债券下限回升的分位数,平仓
    spread_length = 50#至少需要累积的利差数据长度

    market_1 = []#存放债券1的行情数据————是一个object,key为leg1_symbol,value为price、yield等
    market_2 = []#存放债券2的行情数据————是一个object,key为leg1_symbol,value为price、yield等
    close_diff_list = []#存贮利差的list
    time_list_filter = []# 过滤重复的时间戳
    target_pos = {}#记录目标手数,防止因为债券行情导致的成交不到而不停发单

    order_mess = QuoteRequest()
    hdh = HttpDataHandler()
    kslib = KSLibInterface()

    # -------------------分隔线--------------------

    leg1_symbol = '200210.BOND'  #下单腿,用于成交的腿,20年国开10年期债券
    leg2_symbol = '220305.BOND'  #用于计算价差的另一腿,22年进出口5年期债券
    start_date = '2023-02-25'  # %Y-%m-%d
    end_date = '2023-08-25'  # %Y-%m-%d

相关的输入界面如图:(注意:输入界面的所有参数都是可变的,输入会覆盖原定义的值)

定义含参数后,进行初始化:

    def __init__(self, strategy_engine, strategy_name: str, vt_symbols: List[str], setting: dict = None,factors: List[str] = None):#注意:因子的初始化因为后面有接触因子回测这里就先初始化

        #父类初始化,策略引擎、合约初始化
        super().__init__(strategy_engine, strategy_name, vt_symbols,setting)

        # Obtain contract info
        self.leg1_symbol, self.leg2_symbol = vt_symbols


    def on_init(self):
        """
        时间戳、利差、仓位、行情数据初始化,输出日志
        """
        self.write_log("策略初始化")
        self.load_bars(0)
        self.close_diff_list.clear()
        self.time_list_filter.clear()

        for s in self.vt_symbols:
            self.target_pos[s] = 0

    def on_start(self):
        """
        Callback when strategy is started.
        """
        self.write_log("策略启动")
        self.put_event()

    def on_stop(self):
        """
        Callback when strategy is stopped.
        """
        self.write_log("策略停止")
        self.put_event()

    def on_tick(self, tick: TickData):
        pass
        """
        Callback of new tick data update, 实时数据
        """

下面获取行情数据进行数据处理:

    def on_bars(self, bars: Dict[str, BarData]):

        self.cancel_all()#取消所有订单
        if self.leg1_symbol not in bars or self.leg2_symbol not in bars:
            return

        # bar_trade = bars[self.leg1_symbol]
        # bar_base = bars[self.leg2_symbol]
        """
        注意:必须要在前面先定义market_1和market_2两个对象 =[],
        原因:如果这里没有获取到行情,后续对market_1 or market_2处理时报错没找到对象
        {也就是说:所有后续用到的变量最好是先定义}
        """

        #获取bar线数据
        for key in bars:
            if key == self.leg1_symbol and bars[key].close_price > 80:
                self.market_1.append(bars[key])
                self.write_log("执行on_bars: %s - %s" % (key, str(bars[key])))
            elif key == self.leg2_symbol and bars[key].close_price > 80:
                self.market_2.append(bars[key])
                self.write_log("执行on_bars: %s - %s" % (key, str(bars[key])))

        """
        注意:bar里面是两个合约,每次获取一条行情,所以说下面的if与for对齐,
        如果合约1或合约2没获取到行情就返回,并且上面for里面的每个if和else对应,为一组,上面是if和elif,不能是两个if,因为每次只能进一个选项,不是if就是elif
        下面的if若也在for里面,每次获取两个合约中任意一个后就会判断是否为空,不可置疑,肯定有一个为空(每次对合约1或合约2获取一条行情数据),这样就导致永远无法继续执行下去,到这里就退出来行情函数
        """

        if len(self.market_1) == 0 or len(self.market_2) == 0:
            return
        
        """
        注意:两个合约获取的日期必须是在同一天,同一天的行情才能进行比较
        所以后续的所有交易买卖都在这个前提下进行
        而且要注意market_1和market_2是对象,里面存储大量的行情,所以是对获取到的最新行情进行处理,也就是market_1[-1],如果是market_1就每次是对其第一条行情进行处理
        """

        if self.market_1[-1].datetime == self.market_2[-1].datetime:
            bar_close_trade = self.market_1[-1].yield_rate
            self.write_log("%s ytm:  %s" % (self.leg1_symbol,str(bar_close_trade)))
            bar_close_base = self.market_2[-1].yield_rate
            self.write_log("%s ytm:  %s" % (self.leg2_symbol,str(bar_close_base)))
            close_diff = bar_close_trade - bar_close_base
            self.write_log("%s - %s 利差ytm:  %s" % (self.leg1_symbol,self.leg2_symbol,str(close_diff)))

            if len(self.close_diff_list) < self.spread_length:  # 当利差保存数量小于最小值,则保存利差并跳过
                self.close_diff_list.append(close_diff)  # 保存利差
                return

            else:
                spread_up = np.quantile(np.array(self.close_diff_list), self.up_threshold)  # 上分位数
                spread_up_sell = np.quantile(np.array(self.close_diff_list), self.up_threshold_sell)  # 平仓的回落下分位数
                spread_down = np.quantile(np.array(self.close_diff_list), self.down_threshold)  # 下分位数
                spread_down_sell = np.quantile(np.array(self.close_diff_list),self.down_threshold_sell)  # 平仓的回升上分位数
                self.write_log("利差动态上下限: %s - %s - %s - %s" % (spread_up,spread_up_sell,spread_down,spread_down_sell))
                self.close_diff_list.pop(0)  # 利差滚动存放
                self.close_diff_list.append(close_diff)

                """
                定义了一个计算利差所处分位数的函数,计算分位数时就会调用
                """

                def calculate_percentile(current_value, spread_list):
                    sorted_spread = sorted(spread_list)
                    index = sorted_spread.index(current_value)
                    percentile = (index / len(spread_list)) * 100
                    return percentile
                percentile = calculate_percentile(close_diff, self.close_diff_list)
                self.write_log("当前利差所处分位数:  %s" % (str(percentile)))

进行好上述操作后,下面就要判断买卖操作逻辑了,具体逻辑如下:

                """
                    策略逻辑如下:

                    如果利差(10年-1年) > 以往利差的0.8上分位数,此时利率处在高位,预计未来向均值回归,利差缩小,要买入长期限债,就买入1手10年期债券,做空1手1年期短期债,

                    利差数据序列继续读利差数据,如果利差继续保持0.8分位数以上就继续买入,仓位不做限制,原本有仓位也买入,

                    若未来有一天利差缩小向均值回归,利差 < 0.6分位数,此时只要利差读到的数据满足该条件就一次性卖出全部仓位,加了个条件leg1_pos > 0有仓位才平仓,没仓位的话不会做空,

                    不用担心影响到接下来的利差骤降, < spread_down(0.2),不能做空合约1,从0.8分位数到0.2分位数必然会先被利差<0.6分位数逻辑识别到先对合约1平仓,再运行利差<0.2开空仓,

                    因为4个if是平齐的,假设运行了rate_diff > spread_up0.8分位数,势必会运行到rate_diff > spread_down_sell 0.45分位数,但后者的前提是合约1的仓位 < 0,

                    运行> spread_up0.8分位数后合约1的仓位肯定 >0,所以不用担心它们之间会有影响,做多1次合约1又做多一次合约1的错误逻辑
                
                """

代码如下:

                bar_trade_price =self.market_1[-1].close_price
                bar_base_price =self.market_2[-1].close_price
                
                #利率上行周期,买长卖短
                if close_diff > spread_up:  # 当价差大于布林带上界,买价差,买trade,卖base
                    leg1_pos = self.get_pos(self.leg1_symbol)
                    self.write_log("准备报单,1买2卖时,当前合约1价格:  %s" % (str(bar_trade_price)))
                    self.write_log("准备报单,1买2卖时,当前合约2价格:  %s" % (str(bar_base_price)))
                    if leg1_pos == 0 or leg1_pos > 0 :  # 当前无论有无仓位,对合约1开多仓
                        self.write_log("准备报单了,1买2卖")
                        self.order_mess.clear()#清理报单
                        self.order_mess.vt_symbol = self.leg1_symbol#选择合约对象
                        self.order_mess.bid_price = bar_trade_price #报单价格
                        self.order_mess.bid_volume = 100#报单量
                        self.order_mess.settle_type = '2'#确定结算类型
                        self.order_mess.quote_type = '2'#确定报单类型
                        self.quote(BusinessType.XBOND, self.order_mess)#这里是债券报单,所以是双向报单,对于API接口中buy、sell、short、cover等报单类型不适用,这些都是对期货合约定义的,债券只需要quote即可
                        self.write_log("合约1仓位:  %s" % (str(self.get_pos(self.leg1_symbol))))
                        self.order_mess.clear()
                        self.order_mess.vt_symbol = self.leg2_symbol
                        self.order_mess.ask_price = bar_base_price
                        self.order_mess.ask_volume = 100
                        self.order_mess.settle_type = '2'
                        self.order_mess.quote_type = '2'
                        self.quote(BusinessType.XBOND, self.order_mess)#买1同时也要卖2,同时操作,调参数时调整成交委托比来实现成交量的改变
                        self.write_log("合约2仓位:  %s" % (str(self.get_pos(self.leg2_symbol))))

                if close_diff < spread_up_sell:
                    leg1_pos = self.get_pos(self.leg1_symbol)
                    self.write_log("1买2卖,平仓,当前合约1价格:  %s" % (str(bar_trade_price)))
                    self.write_log("1买2卖,平仓,当前合约2价格:  %s" % (str(bar_base_price)))
                    if leg1_pos > 0  :  # 当前有仓位,就先平仓
                        self.write_log("准备报单了,1买2卖,平仓")
                        self.order_mess.clear()
                        self.order_mess.vt_symbol = self.leg1_symbol
                        self.order_mess.ask_price = bar_trade_price
                        self.order_mess.ask_volume = abs(self.get_pos(self.leg1_symbol))
                        self.order_mess.settle_type = '2'
                        self.order_mess.quote_type = '2'
                        self.quote(BusinessType.XBOND, self.order_mess)
                        self.write_log("合约1仓位:  %s" % (str(self.get_pos(self.leg1_symbol))))
                        self.order_mess.clear()
                        self.order_mess.vt_symbol = self.leg2_symbol
                        self.order_mess.bid_price = bar_base_price
                        self.order_mess.bid_volume = abs(self.get_pos(self.leg2_symbol))
                        self.order_mess.settle_type = '2'
                        self.order_mess.quote_type = '2'
                        self.quote(BusinessType.XBOND, self.order_mess)
                        self.write_log("合约2仓位:  %s" % (str(self.get_pos(self.leg2_symbol))))

                #利率下行周期,卖长买短
                if close_diff < spread_down:  # 当价差小于布林带下界,卖价差,卖trade买base
                    leg1_pos = self.get_pos(self.leg1_symbol)
                    self.write_log("准备报单,1卖2买,当前合约1价格:  %s" % (str(bar_trade_price)))
                    self.write_log("准备报单,1卖2买,当前合约2价格:  %s" % (str(bar_base_price)))
                    if leg1_pos == 0 or leg1_pos < 0 : 
                        self.write_log("准备报单了,1卖2买")
                        self.order_mess.clear()
                        self.order_mess.vt_symbol = self.leg1_symbol
                        self.order_mess.ask_price = bar_trade_price
                        self.order_mess.ask_volume = 100
                        self.order_mess.settle_type = '2'
                        self.order_mess.quote_type = '2'
                        self.quote(BusinessType.XBOND, self.order_mess)
                        self.write_log("合约1仓位:  %s" % (str(self.get_pos(self.leg1_symbol))))
                        self.order_mess.clear()
                        self.order_mess.vt_symbol = self.leg2_symbol
                        self.order_mess.bid_price = bar_base_price
                        self.order_mess.bid_volume = 100
                        self.order_mess.settle_type = '2'
                        self.order_mess.quote_type = '2'
                        self.quote(BusinessType.XBOND, self.order_mess)
                        self.write_log("合约2仓位:  %s" % (str(self.get_pos(self.leg2_symbol))))

                if close_diff > spread_down_sell:
                    leg1_pos = self.get_pos(self.leg1_symbol)
                    self.write_log("1卖2买,平仓,当前合约1价格:  %s" % (str(bar_trade_price)))
                    self.write_log("1卖2买,平仓,当前合约2价格:  %s" % (str(bar_base_price)))
                    if leg1_pos < 0:  # 当前空仓位,就平仓
                        self.write_log("准备报单了,1卖2买,平仓")
                        self.order_mess.clear()
                        self.order_mess.vt_symbol = self.leg1_symbol
                        self.order_mess.bid_price = bar_trade_price
                        self.order_mess.bid_volume =abs(self.get_pos(self.leg1_symbol))
                        self.order_mess.settle_type = '2'
                        self.order_mess.quote_type = '2'
                        self.quote(BusinessType.XBOND, self.order_mess)
                        self.write_log("合约1仓位:  %s" % (str(self.get_pos(self.leg1_symbol))))
                        self.order_mess.clear()
                        self.order_mess.vt_symbol = self.leg2_symbol
                        self.order_mess.ask_price = bar_base_price
                        self.order_mess.ask_volume = abs(self.get_pos(self.leg2_symbol))
                        self.order_mess.settle_type = '2'
                        self.order_mess.quote_type = '2'
                        self.quote(BusinessType.XBOND, self.order_mess)
                        self.write_log("合约2仓位:  %s" % (str(self.get_pos(self.leg2_symbol))))
                        #self.sell(BusinessType.XBOND, self.order_mess)

        self.put_event()

整个策略的交易逻辑就体现在上述代码里面了。

策略回测结果如图:

国开债-进出口到期收益率图示2022.02-2023.08

 

国开债-进出口收益率利差图示2022.02-2023.08

回测参数

品种:220205.BOND(22附息国开债05);220305.BOND(22附息进出口债05)
回测区间:2022年2月10日-2023年8月30日
起始资金:1000000
粒度:每天

策略参数

up_threshold = 0.8

up_threshold_sell=0.6

down_threshold = 0.2

down_threshold_sell=0.45

spread_length = 50

回测结果

统计指标

年化收益(%)

5.8582

总收益率(%)

9.5366

最大回撤(%)

-8.6538

期初资金

1000000

期末资金

1095366.35

总盈亏

95366.35

总成交笔数

323

买次数

卖次数

收益回撤比

1.1

夏普比率

0.13

收益概览

收益概览图示2022.02.10-2023.8.30

结果解析

2022-05-10至2022-05-23成交记录

操作记录

这部分是所有获取行情数据里面该有的对象:

#!/usr/bin/python3
# -*- coding: UTF-8 -*-
# add by gaolong.liu


"""
基础数据对象
"""

from dataclasses import dataclass, field
from datetime import datetime
from logging import INFO

from base.constant import Direction, Exchange, Interval, Offset, Product, OptionType, OrderType, \
    StopOrderStatus, Side, BusinessType, QuoteStatus, QuotationType

# ACTIVE_STATUSES = set([Status.SUBMITTING, Status.NOTTRADED, Status.PARTTRADED])
ACTIVE_STATUSES = set([QuoteStatus.NORMAL, QuoteStatus.NEW, QuoteStatus.TRADED])


@dataclass
class BaseData:
    """
    Any data object needs a gateway_name as source
    and should inherit base data.
    """

    gateway_name: str


@dataclass
class MarketFuture():
    """
    Tick data contains information about:
        * last trade in market
        * orderbook snapshot
        * intraday market statistics.
    """

    symbol: str
    exchange: Exchange
    datetime: datetime

    name: str = ""
    volume: int = 0
    turnover: float = 0
    open_interest: float = 0
    last_price: float = 0
    last_volume: int = 0
    limit_up: float = 0
    limit_down: float = 0

    open_price: float = 0
    high_price: float = 0
    low_price: float = 0
    pre_close: float = 0

    bid_price_1: float = 0
    bid_price_2: float = 0
    bid_price_3: float = 0
    bid_price_4: float = 0
    bid_price_5: float = 0
    bid_price_6: float = 0
    bid_price_7: float = 0
    bid_price_8: float = 0
    bid_price_9: float = 0
    bid_price_10: float = 0

    ask_price_1: float = 0
    ask_price_2: float = 0
    ask_price_3: float = 0
    ask_price_4: float = 0
    ask_price_5: float = 0
    ask_price_6: float = 0
    ask_price_7: float = 0
    ask_price_8: float = 0
    ask_price_9: float = 0
    ask_price_10: float = 0

    bid_volume_1: int = 0
    bid_volume_2: int = 0
    bid_volume_3: int = 0
    bid_volume_4: int = 0
    bid_volume_5: int = 0
    bid_volume_6: int = 0
    bid_volume_7: int = 0
    bid_volume_8: int = 0
    bid_volume_9: int = 0
    bid_volume_10: int = 0

    ask_volume_1: int = 0
    ask_volume_2: int = 0
    ask_volume_3: int = 0
    ask_volume_4: int = 0
    ask_volume_5: int = 0
    ask_volume_6: int = 0
    ask_volume_7: int = 0
    ask_volume_8: int = 0
    ask_volume_9: int = 0
    ask_volume_10: int = 0

@dataclass
class MarketLevelBond():
    swap_trade_volume: int = 0
    swap_entry_px: float = 0
    bond_yield_type: str = ""
    bond_yield: float = 0
    bond_entry_price: float = 0
    bond_settle_type: str = ""
    bond_quote_type: str = ""
    bond_delivery_type: str = ""
    bond_clear_method: str = ""
    bond_quote_id: str = ""
    bond_maker_six_code: str = ""
    bond_maker_trader_code: str = ""
    bond_mdEntryDate: str = ""
    bond_mdEntryTime: str = ""
    md_entry_size: int = 0

@dataclass
class MarketBond():
    msg_type: str = ""
    market_indicator: str = ""
    book_type: str = ""
    book_type_sub: str = ""
    security_type: str = ""
    security_id: str = ""
    symbol: str = ""
    bond_code_val: str = ""
    market_depth: int = 0
    message_id: str = ""
    message_source: str = ""
    create_time_ms: str = ""
    create_time: str = ""
    transact_time: str = ""
    sender_comp_id: str = ""
    sender_sub_id: str = ""
    #行情类型(报价类型)
    quotation_type: QuotationType = None

    # 档位行情信息
    bid_1: MarketLevelBond = None
    ask_1: MarketLevelBond = None
    bid_2: MarketLevelBond = None
    ask_2: MarketLevelBond = None
    bid_3: MarketLevelBond = None
    ask_3: MarketLevelBond = None
    bid_4: MarketLevelBond = None
    ask_4: MarketLevelBond = None
    bid_5: MarketLevelBond = None
    ask_5: MarketLevelBond = None
    bid_6: MarketLevelBond = None
    ask_6: MarketLevelBond = None
    bid_7: MarketLevelBond = None
    ask_7: MarketLevelBond = None
    bid_8: MarketLevelBond = None
    ask_8: MarketLevelBond = None
    bid_9: MarketLevelBond = None
    ask_9: MarketLevelBond = None
    bid_10: MarketLevelBond = None
    ask_10: MarketLevelBond = None

@dataclass
class TickData(BaseData):
    """
    Tick data contains information about:
        * 通用字段
        * 债券部分字段
        * 期货、贵金属部分字典
    """

    symbol: str
    exchange: Exchange
    name: str = ""
    datetime: datetime = None
    market_bond: MarketBond = None
    market_future: MarketFuture = None
    localtime: datetime = None

    def __post_init__(self):
        """"""
        self.vt_symbol = f"{self.symbol}.{self.exchange.value}"


@dataclass
class BarData(BaseData):
    """
    Candlestick bar data of a certain trading period.
    """

    symbol: str
    exchange: Exchange
    datetime: datetime

    interval: Interval = None
    volume: int = 0
    turnover: float = 0
    open_interest: float = 0
    # 开盘价
    open_price: float = 0
    # 最高价
    high_price: float = 0
    # 最低价
    low_price: float = 0
    # 收盘价
    close_price: float = 0
    # 收益率
    yield_rate: float = 0
    # 全价
    full_price: float = 0

    def __post_init__(self):
        """"""
        self.vt_symbol = f"{self.symbol}.{self.exchange.value}"

@dataclass
class FactorData(BaseData):
    """
    Candlestick factor data of a certain trading period.
    """

    factor_schedule: str
    factor_class: str
    datetime: datetime
    factor_value: str = None

@dataclass
class TradeData(BaseData):
    """
    Trade data contains information of a fill of an order. One order
    can have several trade fills.
    """
    # 债券代码
    symbol: str
    # 交易所
    exchange: Exchange
    # 委托单ID
    orderid: str
    # 成交单ID
    tradeid: str
    # 方向(期货)
    direction: Direction = None
    # 方向(债券)
    side: Side = None
    # 开平
    offset: Offset = Offset.NONE
    # 成交价格
    price: float = 0
    # 成交收益率
    yield_rate: float = 0
    # 成交量(手)
    volume: int = 0
    # 成交日期
    datetime: datetime = None
    # 面值(元)
    face_value: float = 1.0

    def __post_init__(self):
        """"""
        self.vt_symbol = f"{self.symbol}.{self.exchange.value}"
        self.vt_orderid = f"{self.gateway_name}.{self.orderid}"
        self.vt_tradeid = f"{self.gateway_name}.{self.tradeid}"


@dataclass
class StopOrder:
    vt_symbol: str
    direction: Direction
    offset: Offset
    price: float
    volume: int
    stop_orderid: str
    strategy_name: str
    datetime: datetime
    lock: bool = False
    net: bool = False
    vt_orderids: list = field(default_factory=list)
    status: StopOrderStatus = StopOrderStatus.WAITING

@dataclass
class PositionData(BaseData):
    """
    Positon data is used for tracking each individual position holding.
    """

    symbol: str
    exchange: Exchange
    direction: Direction

    volume: int = 0
    frozen: float = 0
    price: float = 0
    pnl: float = 0
    yd_volume: int = 0

    def __post_init__(self):
        """"""
        self.vt_symbol = f"{self.symbol}.{self.exchange.value}"
        self.vt_positionid = f"{self.vt_symbol}.{self.direction.value}"


@dataclass
class AccountData(BaseData):
    """
    Account data contains information about balance, frozen and
    available.
    """

    accountid: str

    balance: float = 0
    frozen: float = 0

    def __post_init__(self):
        """"""
        self.available = self.balance - self.frozen
        self.vt_accountid = f"{self.gateway_name}.{self.accountid}"


@dataclass
class LogData(BaseData):
    """
    Log data is used for recording log messages on GUI or in log files.
    """

    msg: str
    level: int = INFO

    def __post_init__(self):
        """"""
        self.time = datetime.now()


@dataclass
class ContractData(BaseData):
    """
    Contract data contains basic information about each contract traded.
    """

    symbol: str
    exchange: Exchange
    name: str
    product: Product
    size: float
    pricetick: float

    min_volume: int = 1           # minimum trading volume of the contract
    stop_supported: bool = False    # whether server supports stop order
    net_position: bool = False      # whether gateway uses net position volume
    history_data: bool = False      # whether gateway provides bar history data

    option_strike: float = 0
    option_underlying: str = ""     # vt_symbol of underlying contract
    option_type: OptionType = None
    option_listed: datetime = None
    option_expiry: datetime = None
    option_portfolio: str = ""
    option_index: str = ""          # for identifying options with same strike price

    def __post_init__(self):
        """"""
        self.vt_symbol = f"{self.symbol}.{self.exchange.value}"


@dataclass
class OrderData(BaseData):
    """
    Order data contains information for tracking lastest status
    of a specific order.
    """

    symbol: str
    exchange: Exchange
    orderid: str
    ordertype: BusinessType
    data: BaseData
    tradeid: str = ""
    time: str = ""
    datetime: datetime = None
    updatetime: str = ""
    status: QuoteStatus = QuoteStatus.NEW
    orderrejectreason: str = ""
    SysLocalOrderID: str = ""
    # 触发委托的场景序号(bar\tick\factor)
    scene_sn: str = ""

    def __post_init__(self):
        """"""
        self.vt_symbol = f"{self.symbol}.{self.exchange.value}"
        self.vt_orderid = f"{self.gateway_name}.{self.orderid}"

    def is_active(self) -> bool:
        """
        Check if the order is active.
        """
        if self.status in ACTIVE_STATUSES:
            return True
        else:
            return False

    def create_cancel_request(self) -> "CancelRequest":
        """
        Create cancel request object from order.
        """
        req = CancelRequest(
            orderid=self.orderid, symbol=self.symbol, exchange=self.exchange
        )
        return req

    def addTradeQty(self,tradeQty:int=0, lastQty:int=0, side:Side=None):
        self.data.update_last_qty(tradeQty,lastQty,side)

    def getTradeQty(self, side:Side=None):
        return self.data.get_last_qty(side)

    def isTraded_All(self):
        return self.data.isTraded_All()

@dataclass
class QuoteData(BaseData):
    """
    Quote data contains information for tracking lastest status
    of a specific quote.
    """

    symbol: str
    exchange: Exchange
    quoteid: str

    bid_price: float = 0.0
    bid_volume: int = 0
    ask_price: float = 0.0
    ask_volume: int = 0
    bid_offset: Offset = Offset.NONE
    ask_offset: Offset = Offset.NONE
    status: QuoteStatus = QuoteStatus.NEW
    datetime: datetime = None
    reference: str = ""

    def __post_init__(self):
        """"""
        self.vt_symbol = f"{self.symbol}.{self.exchange.value}"
        self.vt_quoteid = f"{self.gateway_name}.{self.quoteid}"

    def is_active(self) -> bool:
        """
        Check if the quote is active.
        """
        return self.status in ACTIVE_STATUSES

    def create_cancel_request(self) -> "CancelRequest":
        """
        Create cancel request object from quote.
        """
        req = CancelRequest(
            orderid=self.quoteid, symbol=self.symbol, exchange=self.exchange
        )
        return req


@dataclass
class SubscribeRequest:
    """
    Request sending to specific gateway for subscribing tick data update.
    """

    symbol: str
    exchange: Exchange

    def __post_init__(self):
        """"""
        self.vt_symbol = f"{self.symbol}.{self.exchange.value}"

@dataclass
class CancelRequest:
    """
    Request sending to specific gateway for canceling an existing order.
    """

    orderid: str
    symbol: str
    exchange: Exchange

    def __post_init__(self):
        """"""
        self.vt_symbol = f"{self.symbol}.{self.exchange.value}"

@dataclass
class QuoteRequest:
    """
    Request sending to specific gateway for creating a new quote.
    """

    vt_symbol: str = ""
    bid_yield: float = 0
    bid_price: float = 0
    bid_volume: int = 0
    ask_yield: float = 0
    ask_price: float = 0
    ask_volume: int = 0
    settle_type: str = ""
    valid_unit_time: str = ""
    clear_method: str = ""
    delivery_type: str = ""
    # 报价方式,字典:1-集中匹配,2-连续匹配
    quote_type: str = ""
    # 订单类型,字典:2-限价订单,b-FAK(点击成交)
    order_type: str = ""
    real_time_flag: str = ""
    direction: Direction = None
    offset: Offset = Offset.NONE
    reference: str = ""
    # 当前行情ID(现券做市FAK点击时填写)
    md_quote_id: str = ""
    # 当前行情的业务发生时间
    transact_time: str = ""
    # 面值(元)
    face_value: float = 0.0

    # def __post_init__(self):
    #     """"""
    #     self.vt_symbol = f"{self.symbol}.{self.exchange.value}"

    def clear(self):
        self.vt_symbol: str = ""
        self.bid_yield: float = 0
        self.bid_price: float = 0
        self.bid_volume: int = 0
        self.ask_yield: float = 0
        self.ask_price: float = 0
        self.ask_volume: int = 0
        self.settle_type: str = ""
        self.valid_unit_time: str = ""
        self.clear_method: str = ""
        self.delivery_type: str = ""
        self.quote_type: str = ""
        self.order_type: str = ""
        self.real_time_flag: str = ""
        self.direction: Direction = None
        self.offset: Offset = Offset.NONE
        self.reference: str = ""
        self.md_quote_id = ""
        self.transact_time = ""

    # def create_quote_data(self, quoteid: str, gateway_name: str) -> QuoteData:
    #     """
    #     Create quote data from request.
    #     """
    #     quote = QuoteData(
    #         symbol=self.symbol,
    #         exchange=self.exchange,
    #         quoteid=quoteid,
    #         bid_price=self.bid_price,
    #         bid_volume=self.bid_volume,
    #         ask_price=self.ask_price,
    #         ask_volume=self.ask_volume,
    #         bid_offset=self.bid_offset,
    #         ask_offset=self.ask_offset,
    #         reference=self.reference,
    #         gateway_name=gateway_name,
    #     )
    #     return quote

@dataclass
class BarOverview:
    """
    Overview of bar data stored in database.
    """

    symbol: str = ""
    exchange: str = ""
    interval: str = ""
    count: int = 0
    start: datetime = None
    end: datetime = None


@dataclass
class QuoteBondMaker(BaseData):
    """
    现券做市报价
    """
    orderid: str
    symbol: str
    yield_bid: float = 0
    yield_offer: float = 0
    order_qty_bid: int = 0
    order_qty_offer: int = 0
    price_bid: float = 0
    price_offer: float = 0
    strategy_id: str = ""
    invest_group_name: str = ""
    trader_code: str = ""
    valid_unit_time: str = ""
    clear_method: str = ""
    settle_type: str = ""
    delivery_type: str = ""
    # 订单类型,字典:2-限价订单,b-FAK(点击成交)
    order_type: str = ""
    run_id: str = ""
    run_unit_id: str = ""
    exchange: Exchange = Exchange.NONE
    last_qty_bid: int = 0
    last_qty_offer: int = 0
    # 当前行情ID(现券做市FAK点击时填写)
    md_quote_id: str = ""
    # 对应行情的业务发生时间
    transact_time: str = ""
    # 面值(元)
    face_value: float = 100.0

    def __post_init__(self):
        """"""
        self.vt_symbol = f"{self.symbol}.{self.exchange.value}"
        self.vt_orderid = f"{self.gateway_name}.{self.orderid}"

    # 更新已成交量
    def update_last_qty(self, trade_qty: int = 0, last_qty: int = 0, side: Side = None):
        # 依据成交量计算总成交量
        if trade_qty != 0:
            if side == Side.DOUBLE_BUY or side == Side.SINGLE_BUY:
                self.last_qty_bid += trade_qty
            elif side == Side.DOUBLE_SELL or side == Side.SINGLE_SELL:
                self.last_qty_offer += trade_qty
        # 更新总成交量为已成交量
        if last_qty != 0:
            if side == Side.DOUBLE_BUY or side == Side.SINGLE_BUY:
                self.last_qty_bid = last_qty
            elif side == Side.DOUBLE_SELL or side == Side.SINGLE_SELL:
                self.last_qty_offer = last_qty

    # 获得成交量
    def get_last_qty(self, side: Side = None):
        if side == Side.DOUBLE_BUY or side == Side.SINGLE_BUY:
            return self.last_qty_bid
        elif side == Side.DOUBLE_SELL or side == Side.SINGLE_SELL:
            return self.last_qty_offer

    def isTraded_All(self):
        return self.last_qty_offer == self.order_qty_offer and self.last_qty_bid == self.order_qty_bid

    def toJson(self):
        data = {
            "api_id": self.orderid,
            "symbol": self.symbol,
            "exchange": self.exchange.value,
            "vt_symbol": self.vt_symbol,
            "yield_bid": self.yield_bid,
            "yield_offer": self.yield_offer,
            "order_qty_bid": self.order_qty_bid,
            "order_qty_offer": self.order_qty_offer,
            "price_bid": self.price_bid,
            "price_offer": self.price_offer,
            "strategy_id": self.strategy_id,
            "invest_group_name": self.invest_group_name,
            "trader_code": self.trader_code,
            "valid_unit_time": self.valid_unit_time,
            "clear_method": self.clear_method,
            "settle_type": self.settle_type,
            "order_type": self.order_type,
            "run_id": self.run_id,
            "run_unit_id": self.run_unit_id,
            "delivery_type": self.delivery_type,
            "md_quote_id": self.md_quote_id,
            "transact_time": self.transact_time,
            "face_value": self.face_value
        }
        return data

@dataclass
class QuoteXBond(BaseData):
    """
    现券匿名报价
    """
    orderid: str
    symbol: str
    side: Side = None
    yield_rate: float = 0
    order_qty: int = 0
    price: float = 0
    strategy_id: str = ""
    invest_group_name: str = ""
    trader_code: str = ""
    settle_type: str = ""
    # 报价方式,字典:1-集中匹配,2-连续匹配
    match_type: str = ""
    # 订单类型,字典:2-限价订单,b-FAK(点击成交)
    order_type: str = ""
    run_id: str = ""
    run_unit_id: str = ""
    side_id: str = ""
    exchange: Exchange = Exchange.NONE
    last_qty: int = 0
    # 面值(元)
    face_value: float = 100.0

    def __post_init__(self):
        """"""
        self.vt_symbol = f"{self.symbol}.{self.exchange.value}"
        self.vt_orderid = f"{self.gateway_name}.{self.orderid}"

    # 更新已成交量
    def update_last_qty(self, trade_qty: int = 0, last_qty: int = 0, side: Side = None):
        # 依据成交量计算总成交量
        if trade_qty != 0:
            self.last_qty += trade_qty
        # 更新总成交量为已成交量
        if last_qty != 0:
            self.last_qty = last_qty

    # 获得成交量
    def get_last_qty(self, side: Side = None):
        return self.last_qty

    def isTraded_All(self):
        return self.last_qty == self.order_qty

    def toJson(self):
        data = {
            "api_id": self.orderid,
            "symbol": self.symbol,
            "exchange": self.exchange.value,
            "vt_symbol": self.vt_symbol,
            "side": self.side.value,
            "yield_rate": self.yield_rate,
            "order_qty": self.order_qty,
            "price": self.price,
            "strategy_id": self.strategy_id,
            "invest_group_name": self.invest_group_name,
            "trader_code": self.trader_code,
            "side_id": self.side_id,
            "settle_type": self.settle_type,
            "run_id": self.run_id,
            "run_unit_id": self.run_unit_id,
            "match_type": self.match_type,
            "order_type": self.order_type,
            "face_value": self.face_value
        }
        return data

@dataclass
class QuoteXSwap(BaseData):
    """
    利率互换匿名报价
    """
    orderid: str
    symbol: str
    side: Side = None
    order_qty: int = 0
    price: float = 0
    strategy_id: str = ""
    invest_group_name: str = ""
    trader_code: str = ""
    valid_unit_time: str = ""
    # 报价方式,字典:1-集中匹配,2-连续匹配
    match_type: str = ""
    real_time_flag: str = ""
    run_id: str = ""
    run_unit_id: str = ""
    side_id: str = ""
    expire_time: str = ""
    exchange: Exchange = Exchange.NONE
    last_qty: int = 0

    def __post_init__(self):
        """"""
        self.vt_symbol = f"{self.symbol}.{self.exchange.value}"
        self.vt_orderid = f"{self.gateway_name}.{self.orderid}"

    # 更新已成交量
    def update_last_qty(self, trade_qty:int=0, last_qty:int=0, side:Side=None):
        # 依据成交量计算总成交量
        if trade_qty != 0:
            self.last_qty += trade_qty
        # 更新总成交量为已成交量
        if last_qty != 0:
            self.last_qty = last_qty

    # 获得成交量
    def get_last_qty(self, side: Side = None):
        return self.last_qty

    def isTraded_All(self):
        return self.last_qty == self.order_qty

    def toJson(self):
        data = {
            "api_id": self.orderid,
            "symbol": self.symbol,
            "exchange": self.exchange.value,
            "vt_symbol": self.vt_symbol,
            "real_time_flag": self.real_time_flag,
            "side": self.side.value,
            "order_qty": self.order_qty,
            "price": self.price,
            "expire_time": self.expire_time,
            "strategy_id": self.strategy_id,
            "invest_group_name": self.invest_group_name,
            "trader_code": self.trader_code,
            "side_id": self.side_id,
            "valid_unit_time": self.valid_unit_time,
            "run_id": self.run_id,
            "run_unit_id": self.run_unit_id,
            "match_type": self.match_type
        }
        return data

@dataclass
class QuoteFuture(BaseData):
    """
    期货报价
    """
    orderid: str
    symbol: str
    type: OrderType = OrderType.LIMIT
    direction: Direction = None
    offset: Offset = Offset.NONE
    price: float = 0
    volume: int = 0
    traded: int = 0
    exchange: Exchange = Exchange.NONE

    def __post_init__(self):
        """"""
        self.vt_symbol = f"{self.symbol}.{self.exchange.value}"
        self.vt_orderid = f"{self.gateway_name}.{self.orderid}"

    # 更新已成交量
    def update_last_qty(self, trade_qty:int=0, last_qty:int=0, side:Side=None):
        # 依据成交量计算总成交量
        if trade_qty != 0:
            self.traded += trade_qty
        # 更新总成交量为已成交量
        if last_qty !=0:
            self.traded = last_qty

    # 获得成交量
    def get_last_qty(self, side: Side = None):
        return self.traded

    def isTraded_All(self):
        return self.traded == self.volume

    def toJson(self):
        data = {
            "type": self.type.value,
            "direction": self.direction.value,
            "offset": self.offset.value,
            "price": self.price,
            "volume": self.volume,
            "traded": self.traded
        }
        return data


@dataclass
class BondInfo:
    """
    债券基本信息
    """
    # 债券代码
    security_id: str
    # 债券名称
    name: str
    # 债券类型
    bond_type: str
    # 面值(元)
    face_value: float
    # 债券起息日
    dated_date: datetime
    # 到期兑付日
    maturity_date: datetime
    # 付息频率(0-半年,1-年,2-到期,3-月,4-季)
    frequency: str
    # 息票类型(0-固定,1-浮动,2-零息,3-贴现)
    coupon_rate_type: str
    # 票面利率
    coupon: float
    # 计息基准(0-实际/实际,1-实际/360,2-30/360,3-实际/365,5-实际/365F)
    day_count: str
    # 到期期限
    term_to_maturity: str = None

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值