"""
华鑫 L1数据接口
"""
import hashlib
import json
import socket
import time
# 数据格式：(代码,昨日收盘价,最新价,总成交量,总成交额,[(买1价,买1量),(买2价,买2量),(买3价,买3量),(买4价,买4量),(买5价,买5量)],[(卖1价,(卖1量),((卖2价,(卖2量),((卖3价,(卖3量),((卖4价,(卖4量),((卖5价,(卖5量)],更新时间)

class SocketUtil:

    @classmethod
    def md5_encrypt(cls, value):
        md5 = hashlib.md5()
        md5.update(value.encode('utf-8'))
        return md5.hexdigest()

    @classmethod
    def create_socket(cls, addr, port):
        client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # 生成socket，连接server
        client.connect((addr, port))
        return client

    @classmethod
    def load_header(cls, data_bytes):
        slen = '##%08d' % len(data_bytes)
        return slen.encode("utf-8") + data_bytes

    @classmethod
    # 接收数据，去除头
    def recv_data(cls, sk):
        data = ""
        header_size = 10
        buf = sk.recv(1024)
        header_str = buf[:header_size]
        if buf:
            buf = buf.decode('utf-8')
            if buf.startswith("##"):
                content_length = int(buf[2:10])
                received_size = 0
                # 加上读取头的数据
                received_size += len(buf[header_size:])
                data += buf[header_size:]
                while not received_size == content_length:
                    r_data = sk.recv(10240)
                    received_size += len(r_data)
                    data += r_data.decode('utf-8')
            else:
                data = sk.recv(1024 * 1024)
                data = buf + data.decode('utf-8')
        return data, header_str


def get_current_info():
    """
    获取当前L1信息
    :return: [(代码,昨日收盘价,最新价,总成交量,总成交额,买五档（价格，成交额），卖五档（价格，成交额）,更新时间)]
    """
    sk = SocketUtil.create_socket("43.138.167.68", 11008)
    params_bytes = SocketUtil.load_header(json.dumps({"type": "get_l1_data","sign":''}).encode("utf-8"))
    sk.send(params_bytes)
    result_str, header = SocketUtil.recv_data(sk)
    result_json = json.loads(result_str)
    if result_json['code'] == 0:
        return result_json['data']
    return []


if __name__ == "__main__":
    while True:
        print(get_current_info())
        time.sleep(1.5)