"""新股自动申购，提供交易日校验、券商对账和本地幂等保护。"""

from __future__ import annotations

import logging
from datetime import datetime, time
from pathlib import Path

import config
from sdk import Client
from libs.calc import trading_time
from libs.lockfile import is_lock,write_lockfile

IPO_SESSIONS = ((time(9, 30), time(11, 30)), (time(13, 0), time(15, 0)))


def AutoBuyIpo() -> None:
    """安全执行一次新股申购，返回成功提交的证券数量。"""
    if not config.account_config.enable_auto_ipo:
        logging.info("[IPO] 自动申购未启用")
        return
    if not trading_time(datetime.now()):
        logging.info("[IPO] 非交易时间")
        return

    try:
        with Client(
            config.global_config.qmt_base_url,
            config.global_config.qmt_token,
            config.HTTP_TIMEOUT,
        ) as client:
            result = client.ipo_data("STOCK")
            for item in result:
                try:
                    if not isinstance(item, dict):
                        raise TypeError("IPO 数据项必须是字典")

                    stock = str(item.get("stock", "")).strip()
                    if not stock or stock.endswith(".BJ"):
                        continue

                    ipo_price = float(item["issuePrice"])
                    max_purchase_num = int(item["maxPurchaseNum"])
                    if ipo_price <= 0 or max_purchase_num <= 0:
                        raise ValueError("发行价或申购额度必须大于 0")

                    lock_path = Path(config.global_config.qmt_data_dir) / f"{stock}.lock"
                    if is_lock(lock_path):
                        continue

                    client.passorder(
                        op_type=23,
                        stock=stock,
                        volume=max_purchase_num,
                        pr_type=11,
                        price=ipo_price,
                        strategy_name="ipo",
                    )
                    write_lockfile(lock_path)
                    logging.info(
                        "[IPO] %s 申购，发行价：%s 可申购额度：%s",
                        stock,
                        ipo_price,
                        max_purchase_num,
                    )
                except Exception:
                    logging.exception("[IPO] 单条申购处理失败，数据=%r", item)
    except Exception:
        logging.exception("[IPO] 自动申购任务失败")