airflow weekday 源码
airflow weekday 代码
文件路径：/airflow/utils/weekday.py
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements. See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership. The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License. You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied. See the License for the
# specific language governing permissions and limitations
# under the License.
"""Get the ISO standard day number of the week from a given day string"""
from __future__ import annotations
import enum
from typing import Iterable
@enum.unique
class WeekDay(enum.IntEnum):
"""Python Enum containing Days of the Week"""
MONDAY = 1
TUESDAY = 2
WEDNESDAY = 3
THURSDAY = 4
FRIDAY = 5
SATURDAY = 6
SUNDAY = 7
@classmethod
def get_weekday_number(cls, week_day_str: str):
"""
Return the ISO Week Day Number for a Week Day
:param week_day_str: Full Name of the Week Day. Example: "Sunday"
:return: ISO Week Day Number corresponding to the provided Weekday
"""
sanitized_week_day_str = week_day_str.upper()
if sanitized_week_day_str not in cls.__members__:
raise AttributeError(f'Invalid Week Day passed: "{week_day_str}"')
return cls[sanitized_week_day_str]
@classmethod
def convert(cls, day: str | WeekDay) -> int:
"""Helper function that returns the day number in the week"""
if isinstance(day, WeekDay):
return day
return cls.get_weekday_number(week_day_str=day)
@classmethod
def validate_week_day(
cls,
week_day: str | WeekDay | Iterable[str] | Iterable[WeekDay],
) -> set[int]:
"""Validate each item of iterable and create a set to ease compare of values"""
if not isinstance(week_day, Iterable):
if isinstance(week_day, WeekDay):
week_day = {week_day}
else:
raise TypeError(
f"Unsupported Type for week_day parameter: {type(week_day)}."
"Input should be iterable type:"
"str, set, list, dict or Weekday enum type"
)
if isinstance(week_day, str):
week_day = {week_day}
return {cls.convert(item) for item in week_day}
相关信息
相关文章
0
赞
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
8、 golang
-
9、 openharmony
-
10、 Vue中input框自动聚焦