"""
行情管理
"""

__market_info_dict = {}


class MarketInfo:
    def __init__(self, code, price, rate, buy1_price, buy1_volume, total_volume, total_bid_volume, total_ask_volume):
        self.code = code
        self.price = price
        self.rate = rate
        self.buy1_price = buy1_price
        self.buy1_volume = buy1_volume
        self.total_volume = total_volume
        self.total_bid_volume = total_bid_volume
        self.total_ask_volume = total_ask_volume


def set_market_info(data):
    """
    设置行情信息
    :param data: (代码, 最近的价格, 涨幅, 买1价， 买1量, 成交总量)
    :return:
    """
    __market_info_dict[data[0]] = MarketInfo(data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7])


def get_market_info(code) -> MarketInfo:
    """
    获取行情信息
    :param code:
    :return:
    """
    return __market_info_dict.get(code)