mirror of
https://github.com/fastapiadmin/FastapiAdmin.git
synced 2026-09-20 20:39:55 +00:00
zhangtao
1d54dca76c
refactor: 优化代码结构和可读性 feat: 添加http_limit模块实现请求限制功能 fix: 修复异步任务中使用time.sleep的问题 chore: 更新依赖项并添加pytest测试框架 docs: 更新项目描述信息 perf: 优化Redis序列化方式使用JSON替代pickle test: 添加测试相关配置和依赖
287 lines
8.7 KiB
Python
287 lines
8.7 KiB
Python
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
|