akquant 0.1.0__cp39-abi3-win_amd64.whl

akquant/log.py

@@ -0,0 +1,135 @@
+import logging
+import sys
+from typing import Optional, Union
+
+# Default format: Time | Level | Message
+DEFAULT_FORMAT = "%(asctime)s | %(levelname)s | %(message)s"
+DATE_FORMAT = "%Y-%m-%d %H:%M:%S"
+
+
+class Logger:
+    r"""
+    akquant 日志封装
+
+    :description: 提供控制台与文件日志的快捷配置
+    """
+
+    _instance = None
+
+    def __init__(self):
+        self._logger = logging.getLogger("akquant")
+        self._logger.setLevel(logging.INFO)
+        self._handlers = {}  # key -> handler
+
+        # Add default console handler if not present
+        if not self._logger.handlers:
+            self.enable_console()
+
+    @classmethod
+    def get_logger(cls):
+        if cls._instance is None:
+            cls._instance = Logger()
+        return cls._instance._logger
+
+    def set_level(self, level: Union[str, int]):
+        r"""
+        设置日志等级
+
+        :param level: 日志等级字符串或整数 (DEBUG/INFO/WARNING/ERROR/CRITICAL)
+        :type level: str | int
+        """
+        self._logger.setLevel(level)
+
+    def enable_console(self, format_str: str = DEFAULT_FORMAT):
+        r"""
+        启用控制台日志
+
+        :param format_str: 日志格式字符串
+        :type format_str: str
+        """
+        if "console" in self._handlers:
+            return
+
+        handler = logging.StreamHandler(sys.stdout)
+        handler.setFormatter(logging.Formatter(format_str, datefmt=DATE_FORMAT))
+        self._logger.addHandler(handler)
+        self._handlers["console"] = handler
+
+    def disable_console(self):
+        r"""
+        禁用控制台日志
+        """
+        if "console" in self._handlers:
+            self._logger.removeHandler(self._handlers["console"])
+            del self._handlers["console"]
+
+    def enable_file(
+        self, filename: str, format_str: str = DEFAULT_FORMAT, mode: str = "a"
+    ):
+        r"""
+        启用文件日志
+
+        :param filename: 日志文件路径
+        :type filename: str
+        :param format_str: 日志格式字符串
+        :type format_str: str
+        :param mode: 文件打开模式 ('a' 追加 或 'w' 覆写)
+        :type mode: str
+        """
+        # Remove existing file handler if path matches (simple check)
+        key = f"file_{filename}"
+        if key in self._handlers:
+            return
+
+        handler = logging.FileHandler(filename, mode=mode, encoding="utf-8")
+        handler.setFormatter(logging.Formatter(format_str, datefmt=DATE_FORMAT))
+        self._logger.addHandler(handler)
+        self._handlers[key] = handler
+
+
+# Global helper functions
+def get_logger():
+    r"""
+    获取全局 logger 实例
+
+    :return: 已初始化的 logger
+    :rtype: logging.Logger
+    """
+    return Logger.get_logger()
+
+
+def set_log_level(level: Union[str, int]):
+    r"""
+    设置全局日志等级
+
+    :param level: 日志等级字符串或整数
+    :type level: str | int
+    """
+    Logger.get_logger().setLevel(level)
+
+
+def register_logger(
+    filename: Optional[str] = None, console: bool = True, level: str = "INFO"
+):
+    r"""
+    日志一体化配置
+
+    :param filename: 日志文件路径，提供则写入文件
+    :type filename: str, optional
+    :param console: 是否输出到控制台
+    :type console: bool
+    :param level: 日志等级 ("DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL")
+    :type level: str
+    """
+    logger_manager = Logger._instance or Logger()
+    Logger._instance = logger_manager
+
+    logger_manager.set_level(level.upper())
+
+    if console:
+        logger_manager.enable_console()
+    else:
+        logger_manager.disable_console()
+
+    if filename:
+        logger_manager.enable_file(filename)

akquant/sizer.py

@@ -0,0 +1,82 @@
+from abc import ABC, abstractmethod
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from .akquant import StrategyContext
+
+
+class Sizer(ABC):
+    """
+    仓位管理基类 (Sizer Base Class)
+    用于计算下单数量
+    """
+
+    @abstractmethod
+    def get_size(
+        self, price: float, cash: float, context: "StrategyContext", symbol: str
+    ) -> float:
+        """
+        计算下单数量
+
+        Args:
+            price (float): 当前价格
+            cash (float): 当前可用资金
+            context (StrategyContext): 策略上下文
+            symbol (str): 标的代码
+
+        Returns:
+            float: 下单数量
+        """
+        pass
+
+
+class FixedSize(Sizer):
+    """
+    固定数量 Sizer
+    每次交易固定数量
+    """
+
+    def __init__(self, size: float = 100.0):
+        self.size = size
+
+    def get_size(
+        self, price: float, cash: float, context: "StrategyContext", symbol: str
+    ) -> float:
+        return self.size
+
+
+class PercentSizer(Sizer):
+    """
+    百分比 Sizer
+    使用当前资金的一定百分比买入
+    """
+
+    def __init__(self, percents: float = 10.0):
+        """
+        Args:
+            percents (float): 资金百分比 (0-100)
+        """
+        self.percents = percents
+
+    def get_size(
+        self, price: float, cash: float, context: "StrategyContext", symbol: str
+    ) -> float:
+        if price <= 0:
+            return 0.0
+
+        target_cash = cash * (self.percents / 100.0)
+        return int(target_cash / price)
+
+
+class AllInSizer(Sizer):
+    """
+    全仓 Sizer
+    使用所有可用资金买入
+    """
+
+    def get_size(
+        self, price: float, cash: float, context: "StrategyContext", symbol: str
+    ) -> float:
+        if price <= 0:
+            return 0.0
+        return int(cash / price)