"""
行情管理
"""
from log_module import async_log_util
from log_module.log import logger_local_huaxin_l2_market

__market_info_dict = {}


class MarketInfo:
    def __init__(self, code, price, rate, buy1_price, buy1_volume, total_volume, total_bid_volume, total_ask_volume, pre_close_price):
        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
        self.pre_close_price = pre_close_price


def set_market_info(data, with_log=True):
    """
    设置行情信息
    :param with_log: 是否写入日志
    :param data: (代码, 最近的价格, 涨幅, 买1价， 买1量, 成交总量, 买入量, 卖出量, 昨日收盘价, 时间戳)
    :return:
    """
    if with_log:
        async_log_util.info(logger_local_huaxin_l2_market, f"{data}")
    __market_info_dict[data[0]] = MarketInfo(data[0], data[1], data[2], data[3], data[4], data[5], data[6], data[7], data[8])


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