# -*- coding: utf-8 -*- 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()} elif isinstance(value, list): return [_format_value(item) for item in value] elif 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): """ 校验秒或分钟字段的合法性。 参数: - second_or_minute (str): 秒或分钟值。 返回: - bool: 校验是否通过。 """ if ( 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) ): return True return False @classmethod def validate_hour(cls, hour: str): """ 校验小时字段的合法性。 参数: - hour (str): 小时值。 返回: - bool: 校验是否通过。 """ if ( 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) ): return True return False @classmethod def validate_day(cls, day: str): """ 校验日期字段的合法性。 参数: - day (str): 日值。 返回: - bool: 校验是否通过。 """ if ( 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) ): return True return False @classmethod def validate_month(cls, month: str): """ 校验月份字段的合法性。 参数: - month (str): 月值。 返回: - bool: 校验是否通过。 """ if ( 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) ): return True return False @classmethod def validate_week(cls, week: str): """ 校验星期字段的合法性。 参数: - week (str): 周值。 返回: - bool: 校验是否通过。 """ if ( 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) ): return True return False @classmethod def validate_year(cls, year: str): """ 校验年份字段的合法性。 参数: - year (str): 年值。 返回: - bool: 校验是否通过。 """ current_year = int(datetime.now().year) future_years = [current_year + i for i in range(9)] if ( 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(',')) ) ): return True return False @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