"""
下单队列管理
"""

# 尚未处理
import json
import random
import time

from utils import tool
from db import mysql_data_delegate as mysql_data

STATE_NOT_PROCESS = 0
STATE_NOT_BUY = 1
STATE_BUY = 2


@tool.singleton
class PlaceOrderRecordManager:
    """
    下单记录管理
    """

    def __init__(self, day):
        self.day = day
        # (ID,代码,板块信息, 大单信息, 时间, 价格, 涨幅)
        self.not_process_records = []
        self.mysql_db = mysql_data.Mysqldb()
        self.__load_data()

    def __load_data(self):
        results = self.mysql_db.select_all(
            f"select * from low_suction_place_order_records where day ='{self.day}' and state = {STATE_NOT_PROCESS} order by time_str")
        self.not_process_records.clear()
        if results:
            self.not_process_records = [(r[0], r[2], json.loads(r[3]), r[6], r[7], float(r[8]), float(r[9])) for r in
                                        results]

    def set_buy(self, id):
        record = None
        for r in self.not_process_records:
            if r[0] == id:
                record = r
        if record:
            self.not_process_records.remove(record)
            self.mysql_db.execute(
                f"update low_suction_place_order_records set state={STATE_BUY},update_time=now() where id ='{id}'")
            # 将后续还未处理的同代码的拒绝
            ids = [r[0] for r in self.not_process_records if r[1] == record[1]]
            for id_ in ids:
                self.set_not_buy(id_, "已买入")

    def set_not_buy(self, id, desc):
        for r in self.not_process_records:
            if r[0] == id:
                self.not_process_records.remove(r)
        self.mysql_db.execute(
            f"update low_suction_place_order_records set state={STATE_NOT_BUY},update_time=now(), state_desc='{desc}' where id ='{id}'")

    def add_record(self, code, time_str, plate_infos, big_order_info_str, price, rate):
        _id = f"{self.day}_{code}_{time_str.replace(':', '')}_{random.randint(0, 10000)}"
        data = (_id, code, plate_infos, big_order_info_str, time_str, price, rate)
        self.mysql_db.execute(
            f"insert into low_suction_place_order_records(id,day,code,plates_info,state,  big_orders_info, time_str,price, rate, create_time) values('{_id}','{self.day}','{code}','{json.dumps(plate_infos, ensure_ascii=False)}',{STATE_NOT_PROCESS},'{big_order_info_str}','{time_str}', {price},{rate},now())")
        self.not_process_records.append(data)

    def get_not_process_record(self):
        if not self.not_process_records:
            return None
        return self.not_process_records[0]

    def get_not_process_record_by_id(self, _id):
        for r in self.not_process_records:
            if r[0] == _id:
                return r
        return None

    def get_not_process_records_by_code(self, code):
        return [r for r in self.not_process_records if r[1] == code]


if __name__ == "__main__":
    manager = PlaceOrderRecordManager("2025-07-07")
    print(manager.not_process_records)
    manager.add_record("000333", '09:30:32', {"家电": ["000111", "000222"]}, "500w/600w", 10.23, 5.2)
    # manager.add_record("000333", '09:30:30', {"家电": ["000111", "000222"]}, "500w/600w")

    # manager.set_not_buy("2025-07-07_000333_1751880205848_3395", "人工驳回")
    print(manager.not_process_records)