upload project source code
This commit is contained in:
281
后端源码/yifan.action-ai.cn/app/utils/time_util.py
Normal file
281
后端源码/yifan.action-ai.cn/app/utils/time_util.py
Normal file
@@ -0,0 +1,281 @@
|
||||
# -*- 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
|
||||
Reference in New Issue
Block a user