payment-platform/backend/app/utils/time_util.py

import re
from datetime import datetime
from typing import Any


class TimeUtil:
    """
    时间格式化工具类
    """

    DEFAULT_DATETIME_FORMAT = "%Y-%m-%d %H:%M:%S"

    @classmethod
    def object_format_datetime(cls, obj: Any) -> Any:
        """
        格式化对象中的 datetime 属性为默认字符串格式。

        参数:
        - obj (Any): 输入对象。

        返回:
        - Any: 格式化后的对象。
        """
        for attr in dir(obj):
            if not attr.startswith("_"):  # 跳过私有属性
                value = getattr(obj, attr)
                if isinstance(value, datetime):
                    setattr(obj, attr, value.strftime(cls.DEFAULT_DATETIME_FORMAT))
        return obj

    @classmethod
    def list_format_datetime(cls, lst: list[Any]) -> list[Any]:
        """
        格式化列表内每个对象的 datetime 属性。

        参数:
        - lst (List[Any]): 对象列表。

        返回:
        - list[Any]: 格式化后的对象列表。
        """
        return [cls.object_format_datetime(obj) for obj in lst]

    @classmethod
    def format_datetime_dict_list(cls, dicts: list[dict]) -> list[dict]:
        """
        递归格式化字典列表中的 datetime 值为默认字符串格式。

        参数:
        - dicts (list[dict]): 字典列表。

        返回:
        - list[dict]: 格式化后的字典列表。
        """

        def _format_value(value: Any) -> Any:
            if isinstance(value, dict):
                return {k: _format_value(v) for k, v in value.items()}
            if isinstance(value, list):
                return [_format_value(item) for item in value]
            if isinstance(value, datetime):
                return value.strftime(cls.DEFAULT_DATETIME_FORMAT)
            return value

        return [_format_value(item) for item in dicts]

    @classmethod
    def __valid_range(cls, search_str: str, start_range: int, end_range: int) -> bool:
        """
        校验范围字符串是否合法。

        参数:
        - search_str (str): 范围字符串（例如："1-5"）。
        - start_range (int): 允许的最小范围值。
        - end_range (int): 允许的最大范围值。

        返回:
        - bool: 校验是否通过。
        """
        match = re.match(r"^(\d+)-(\d+)$", search_str)
        if match:
            start, end = int(match.group(1)), int(match.group(2))
            return start_range <= start < end <= end_range
        return False

    @classmethod
    def __valid_sum(
        cls,
        search_str: str,
        start_range_a: int,
        start_range_b: int,
        end_range_a: int,
        end_range_b: int,
        sum_range: int,
    ) -> bool:
        """
        校验和字符串是否合法。

        参数:
        - search_str (str): 和字符串（例如："1/5"）。
        - start_range_a (int): 允许的最小范围值A。
        - start_range_b (int): 允许的最大范围值A。
        - end_range_a (int): 允许的最小范围值B。
        - end_range_b (int): 允许的最大范围值B。
        - sum_range (int): 允许的最大和值。

        返回:
        - bool: 校验是否通过。
        """
        match = re.match(r"^(\d+)/(\d+)$", search_str)
        if match:
            start, end = int(match.group(1)), int(match.group(2))
            return (
                start_range_a <= start <= start_range_b
                and end_range_a <= end <= end_range_b
                and start + end <= sum_range
            )
        return False

    @classmethod
    def validate_second_or_minute(cls, second_or_minute: str) -> bool:
        """
        校验秒或分钟字段的合法性。

        参数:
        - second_or_minute (str): 秒或分钟值。

        返回:
        - bool: 校验是否通过。
        """
        return bool(
            second_or_minute == "*"
            or ("-" in second_or_minute and cls.__valid_range(second_or_minute, 0, 59))
            or ("/" in second_or_minute and cls.__valid_sum(second_or_minute, 0, 58, 1, 59, 59))
            or re.match(r"^(?:[0-5]?\d|59)(?:,[0-5]?\d|59)*$", second_or_minute)
        )

    @classmethod
    def validate_hour(cls, hour: str) -> bool:
        """
        校验小时字段的合法性。

        参数:
        - hour (str): 小时值。

        返回:
        - bool: 校验是否通过。
        """
        return bool(
            hour == "*"
            or ("-" in hour and cls.__valid_range(hour, 0, 23))
            or ("/" in hour and cls.__valid_sum(hour, 0, 22, 1, 23, 23))
            or re.match(r"^(?:0|[1-9]|1\d|2[0-3])(?:,(?:0|[1-9]|1\d|2[0-3]))*$", hour)
        )

    @classmethod
    def validate_day(cls, day: str) -> bool:
        """
        校验日期字段的合法性。

        参数:
        - day (str): 日值。

        返回:
        - bool: 校验是否通过。
        """
        return bool(
            day in ["*", "?", "L"]
            or ("-" in day and cls.__valid_range(day, 1, 31))
            or ("/" in day and cls.__valid_sum(day, 1, 30, 1, 30, 31))
            or ("W" in day and re.match(r"^(?:[1-9]|1\d|2\d|3[01])W$", day))
            or re.match(
                r"^(?:0|[1-9]|1\d|2[0-9]|3[0-1])(?:,(?:0|[1-9]|1\d|2[0-9]|3[0-1]))*$",
                day,
            )
        )

    @classmethod
    def validate_month(cls, month: str) -> bool:
        """
        校验月份字段的合法性。

        参数:
        - month (str): 月值。

        返回:
        - bool: 校验是否通过。
        """
        return bool(
            month == "*"
            or ("-" in month and cls.__valid_range(month, 1, 12))
            or ("/" in month and cls.__valid_sum(month, 1, 11, 1, 11, 12))
            or re.match(r"^(?:0|[1-9]|1[0-2])(?:,(?:0|[1-9]|1[0-2]))*$", month)
        )

    @classmethod
    def validate_week(cls, week: str) -> bool:
        """
        校验星期字段的合法性。

        参数:
        - week (str): 周值。

        返回:
        - bool: 校验是否通过。
        """
        return bool(
            week in ["*", "?"]
            or ("-" in week and cls.__valid_range(week, 1, 7))
            or ("#" in week and re.match(r"^[1-7]#[1-4]$", week))
            or ("L" in week and re.match(r"^[1-7]L$", week))
            or re.match(r"^[1-7](?:(,[1-7]))*$", week)
        )

    @classmethod
    def validate_year(cls, year: str) -> bool:
        """
        校验年份字段的合法性。

        参数:
        - year (str): 年值。

        返回:
        - bool: 校验是否通过。
        """
        current_year = int(datetime.now().year)
        future_years = [current_year + i for i in range(9)]
        return bool(
            year == "*"
            or ("-" in year and cls.__valid_range(year, current_year, 2099))
            or (
                "/" in year
                and cls.__valid_sum(year, current_year, 2098, 1, 2099 - current_year, 2099)
            )
            or ("#" in year and re.match(r"^[1-7]#[1-4]$", year))
            or ("L" in year and re.match(r"^[1-7]L$", year))
            or (
                (len(year) == 4 or "," in year)
                and all(
                    int(item) in future_years and current_year <= int(item) <= 2099
                    for item in year.split(",")
                )
            )
        )

    @classmethod
    def validate_cron_expression(cls, cron_expression: str):
        """
        校验 Cron 表达式是否正确。
        * * * * * *
        | | | | | |
        | | | | | +--- 星期（0-7，0和7都表示星期日）
        | | | | +----- 月份（1-12）
        | | | +------- 日期（1-31）
        | | +--------- 小时（0-23）
        | +----------- 分钟（0-59）
        +------------- 秒（0-59），部分环境不支持秒字段。

        参数:
        - cron_expression (str): Cron 表达式。

        返回:
        - bool: 校验是否通过。
        """
        values = cron_expression.split()
        if len(values) != 6 and len(values) != 7:
            return False
        second_validation = cls.validate_second_or_minute(values[0])
        minute_validation = cls.validate_second_or_minute(values[1])
        hour_validation = cls.validate_hour(values[2])
        day_validation = cls.validate_day(values[3])
        month_validation = cls.validate_month(values[4])
        week_validation = cls.validate_week(values[5])
        validation = (
            second_validation
            and minute_validation
            and hour_validation
            and day_validation
            and month_validation
            and week_validation
        )
        if len(values) == 6:
            return validation
        if len(values) == 7:
            year_validation = cls.validate_year(values[6])
            return validation and year_validation