220 lines
6.7 KiB
Python
220 lines
6.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
|
|
import re
|
|
from datetime import datetime
|
|
from app.utils.time_util import TimeUtil
|
|
|
|
|
|
class CronUtil:
|
|
"""
|
|
Cron表达式工具类
|
|
"""
|
|
|
|
@classmethod
|
|
def __valid_range(cls, search_str: str, start_range: int, end_range: int) -> bool:
|
|
"""
|
|
校验范围表达式的合法性。
|
|
|
|
参数:
|
|
- search_str (str): 范围表达式。
|
|
- 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): 和表达式。
|
|
- start_range_a (int): 开始范围A。
|
|
- start_range_b (int): 开始范围B。
|
|
- end_range_a (int): 结束范围A。
|
|
- 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: 校验是否通过。
|
|
"""
|
|
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) -> bool:
|
|
"""
|
|
校验小时字段的合法性。
|
|
|
|
参数:
|
|
- 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) -> bool:
|
|
"""
|
|
校验日期字段的合法性。
|
|
|
|
参数:
|
|
- 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) -> bool:
|
|
"""
|
|
校验月份字段的合法性。
|
|
|
|
参数:
|
|
- 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) -> bool:
|
|
"""
|
|
校验星期字段的合法性。
|
|
|
|
参数:
|
|
- 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) -> bool:
|
|
"""
|
|
校验年份字段的合法性。
|
|
|
|
参数:
|
|
- year (str): 年值。
|
|
|
|
返回:
|
|
- bool: 校验是否通过。
|
|
"""
|
|
current_year = int(TimeUtil.now_u8().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) -> bool | None:
|
|
"""
|
|
校验 Cron 表达式是否正确。
|
|
|
|
参数:
|
|
- cron_expression (str): Cron 表达式。
|
|
|
|
返回:
|
|
- bool | None: 校验是否通过。
|
|
"""
|
|
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
|