# 数据管理
import datetime
import json
import os

from utils import tool
from utils import log_util


class CodeDataManager:
    def save_data(self, data):
        code = data["code"]
        create_at = data["created_at"]
        data["created_at"] = create_at.strftime("%Y-%m-%d %H:%M:%S")
        log_util.save_l1_data(code, data)

    def get_datas(self, code):
        datas = log_util.get_l1_datas(code)
        # 根据创建时间排序
        datas = sorted(datas, key=lambda tup: int(tup["created_at"].split(" ")[1].replace(":", "")))
        return datas

    def get_lack_datas_time_range(self, datas, min_time=None):
        def is_trade_time(time_):
            if int("092500") <= int(time_.replace(":", "")) <= int("113000") or int("130000") <= int(
                    time_.replace(":", "")) <= int("150000"):
                return True
            return False

        ranges = []

        now_time_str = tool.get_now_time_str()
        if tool.trade_time_sub(now_time_str, "15:00:00") > 0:
            now_time_str = "15:00:00"

        if tool.trade_time_sub(now_time_str, "13:00:00") < 0 and tool.trade_time_sub(now_time_str, "11:30:00") > 0:
            now_time_str = "11:30:00"

        # 还没到时间
        if tool.trade_time_sub(now_time_str, "09:25:00") < 0:
            return ranges

        last_time = "09:25:01" if min_time is None else min_time
        # tool.trade_time_add_second(now_time_str, 0 - 18 * 60)
        # if tool.trade_time_sub(last_time, "09:25:00") < 0:
        #    last_time = "09:25:00"

        if len(datas) == 0:
            ranges.append((last_time, now_time_str))
            return ranges

        for data in datas:
            # 如果上次时间与本次时间间隔相差30s以上才补全
            time_ = data["created_at"].split(" ")[1]
            if tool.trade_time_sub(time_, last_time) < 0:
                continue
            if tool.trade_time_sub(time_, last_time) > 30:
                if len(ranges) == 0 or ranges[-1][0] != last_time:
                    if is_trade_time(last_time) and is_trade_time(time_):
                        start_time = tool.trade_time_add_second(last_time, 3)
                        end_time = tool.trade_time_add_second(time_, -3)
                        if tool.trade_time_sub(end_time, start_time) > 0:
                            ranges.append((start_time, end_time))
            last_time = time_
        return ranges