|
from datetime import ( |
|
datetime, |
|
time, |
|
timedelta, |
|
) |
|
from typing import ( |
|
Any, |
|
Collection, |
|
Literal, |
|
TypeVar, |
|
overload, |
|
) |
|
|
|
import numpy as np |
|
|
|
from pandas._libs.tslibs.nattype import NaTType |
|
from pandas._typing import ( |
|
OffsetCalendar, |
|
Self, |
|
npt, |
|
) |
|
|
|
from .timedeltas import Timedelta |
|
|
|
_BaseOffsetT = TypeVar("_BaseOffsetT", bound=BaseOffset) |
|
_DatetimeT = TypeVar("_DatetimeT", bound=datetime) |
|
_TimedeltaT = TypeVar("_TimedeltaT", bound=timedelta) |
|
|
|
_relativedelta_kwds: set[str] |
|
prefix_mapping: dict[str, type] |
|
|
|
class ApplyTypeError(TypeError): ... |
|
|
|
class BaseOffset: |
|
n: int |
|
normalize: bool |
|
def __init__(self, n: int = ..., normalize: bool = ...) -> None: ... |
|
def __eq__(self, other) -> bool: ... |
|
def __ne__(self, other) -> bool: ... |
|
def __hash__(self) -> int: ... |
|
@property |
|
def kwds(self) -> dict: ... |
|
@property |
|
def base(self) -> BaseOffset: ... |
|
@overload |
|
def __add__(self, other: npt.NDArray[np.object_]) -> npt.NDArray[np.object_]: ... |
|
@overload |
|
def __add__(self, other: BaseOffset) -> Self: ... |
|
@overload |
|
def __add__(self, other: _DatetimeT) -> _DatetimeT: ... |
|
@overload |
|
def __add__(self, other: _TimedeltaT) -> _TimedeltaT: ... |
|
@overload |
|
def __radd__(self, other: npt.NDArray[np.object_]) -> npt.NDArray[np.object_]: ... |
|
@overload |
|
def __radd__(self, other: BaseOffset) -> Self: ... |
|
@overload |
|
def __radd__(self, other: _DatetimeT) -> _DatetimeT: ... |
|
@overload |
|
def __radd__(self, other: _TimedeltaT) -> _TimedeltaT: ... |
|
@overload |
|
def __radd__(self, other: NaTType) -> NaTType: ... |
|
def __sub__(self, other: BaseOffset) -> Self: ... |
|
@overload |
|
def __rsub__(self, other: npt.NDArray[np.object_]) -> npt.NDArray[np.object_]: ... |
|
@overload |
|
def __rsub__(self, other: BaseOffset): ... |
|
@overload |
|
def __rsub__(self, other: _DatetimeT) -> _DatetimeT: ... |
|
@overload |
|
def __rsub__(self, other: _TimedeltaT) -> _TimedeltaT: ... |
|
@overload |
|
def __mul__(self, other: np.ndarray) -> np.ndarray: ... |
|
@overload |
|
def __mul__(self, other: int): ... |
|
@overload |
|
def __rmul__(self, other: np.ndarray) -> np.ndarray: ... |
|
@overload |
|
def __rmul__(self, other: int) -> Self: ... |
|
def __neg__(self) -> Self: ... |
|
def copy(self) -> Self: ... |
|
@property |
|
def name(self) -> str: ... |
|
@property |
|
def rule_code(self) -> str: ... |
|
@property |
|
def freqstr(self) -> str: ... |
|
def _apply(self, other): ... |
|
def _apply_array(self, dtarr: np.ndarray) -> np.ndarray: ... |
|
def rollback(self, dt: datetime) -> datetime: ... |
|
def rollforward(self, dt: datetime) -> datetime: ... |
|
def is_on_offset(self, dt: datetime) -> bool: ... |
|
def __setstate__(self, state) -> None: ... |
|
def __getstate__(self): ... |
|
@property |
|
def nanos(self) -> int: ... |
|
def is_anchored(self) -> bool: ... |
|
|
|
def _get_offset(name: str) -> BaseOffset: ... |
|
|
|
class SingleConstructorOffset(BaseOffset): |
|
@classmethod |
|
def _from_name(cls, suffix: None = ...): ... |
|
def __reduce__(self): ... |
|
|
|
@overload |
|
def to_offset(freq: None, is_period: bool = ...) -> None: ... |
|
@overload |
|
def to_offset(freq: _BaseOffsetT, is_period: bool = ...) -> _BaseOffsetT: ... |
|
@overload |
|
def to_offset(freq: timedelta | str, is_period: bool = ...) -> BaseOffset: ... |
|
|
|
class Tick(SingleConstructorOffset): |
|
_creso: int |
|
_prefix: str |
|
def __init__(self, n: int = ..., normalize: bool = ...) -> None: ... |
|
@property |
|
def delta(self) -> Timedelta: ... |
|
@property |
|
def nanos(self) -> int: ... |
|
|
|
def delta_to_tick(delta: timedelta) -> Tick: ... |
|
|
|
class Day(Tick): ... |
|
class Hour(Tick): ... |
|
class Minute(Tick): ... |
|
class Second(Tick): ... |
|
class Milli(Tick): ... |
|
class Micro(Tick): ... |
|
class Nano(Tick): ... |
|
|
|
class RelativeDeltaOffset(BaseOffset): |
|
def __init__(self, n: int = ..., normalize: bool = ..., **kwds: Any) -> None: ... |
|
|
|
class BusinessMixin(SingleConstructorOffset): |
|
def __init__( |
|
self, n: int = ..., normalize: bool = ..., offset: timedelta = ... |
|
) -> None: ... |
|
|
|
class BusinessDay(BusinessMixin): ... |
|
|
|
class BusinessHour(BusinessMixin): |
|
def __init__( |
|
self, |
|
n: int = ..., |
|
normalize: bool = ..., |
|
start: str | time | Collection[str | time] = ..., |
|
end: str | time | Collection[str | time] = ..., |
|
offset: timedelta = ..., |
|
) -> None: ... |
|
|
|
class WeekOfMonthMixin(SingleConstructorOffset): |
|
def __init__( |
|
self, n: int = ..., normalize: bool = ..., weekday: int = ... |
|
) -> None: ... |
|
|
|
class YearOffset(SingleConstructorOffset): |
|
def __init__( |
|
self, n: int = ..., normalize: bool = ..., month: int | None = ... |
|
) -> None: ... |
|
|
|
class BYearEnd(YearOffset): ... |
|
class BYearBegin(YearOffset): ... |
|
class YearEnd(YearOffset): ... |
|
class YearBegin(YearOffset): ... |
|
|
|
class QuarterOffset(SingleConstructorOffset): |
|
def __init__( |
|
self, n: int = ..., normalize: bool = ..., startingMonth: int | None = ... |
|
) -> None: ... |
|
|
|
class BQuarterEnd(QuarterOffset): ... |
|
class BQuarterBegin(QuarterOffset): ... |
|
class QuarterEnd(QuarterOffset): ... |
|
class QuarterBegin(QuarterOffset): ... |
|
class MonthOffset(SingleConstructorOffset): ... |
|
class MonthEnd(MonthOffset): ... |
|
class MonthBegin(MonthOffset): ... |
|
class BusinessMonthEnd(MonthOffset): ... |
|
class BusinessMonthBegin(MonthOffset): ... |
|
|
|
class SemiMonthOffset(SingleConstructorOffset): |
|
def __init__( |
|
self, n: int = ..., normalize: bool = ..., day_of_month: int | None = ... |
|
) -> None: ... |
|
|
|
class SemiMonthEnd(SemiMonthOffset): ... |
|
class SemiMonthBegin(SemiMonthOffset): ... |
|
|
|
class Week(SingleConstructorOffset): |
|
def __init__( |
|
self, n: int = ..., normalize: bool = ..., weekday: int | None = ... |
|
) -> None: ... |
|
|
|
class WeekOfMonth(WeekOfMonthMixin): |
|
def __init__( |
|
self, n: int = ..., normalize: bool = ..., week: int = ..., weekday: int = ... |
|
) -> None: ... |
|
|
|
class LastWeekOfMonth(WeekOfMonthMixin): ... |
|
|
|
class FY5253Mixin(SingleConstructorOffset): |
|
def __init__( |
|
self, |
|
n: int = ..., |
|
normalize: bool = ..., |
|
weekday: int = ..., |
|
startingMonth: int = ..., |
|
variation: Literal["nearest", "last"] = ..., |
|
) -> None: ... |
|
|
|
class FY5253(FY5253Mixin): ... |
|
|
|
class FY5253Quarter(FY5253Mixin): |
|
def __init__( |
|
self, |
|
n: int = ..., |
|
normalize: bool = ..., |
|
weekday: int = ..., |
|
startingMonth: int = ..., |
|
qtr_with_extra_week: int = ..., |
|
variation: Literal["nearest", "last"] = ..., |
|
) -> None: ... |
|
|
|
class Easter(SingleConstructorOffset): ... |
|
|
|
class _CustomBusinessMonth(BusinessMixin): |
|
def __init__( |
|
self, |
|
n: int = ..., |
|
normalize: bool = ..., |
|
weekmask: str = ..., |
|
holidays: list | None = ..., |
|
calendar: OffsetCalendar | None = ..., |
|
offset: timedelta = ..., |
|
) -> None: ... |
|
|
|
class CustomBusinessDay(BusinessDay): |
|
def __init__( |
|
self, |
|
n: int = ..., |
|
normalize: bool = ..., |
|
weekmask: str = ..., |
|
holidays: list | None = ..., |
|
calendar: OffsetCalendar | None = ..., |
|
offset: timedelta = ..., |
|
) -> None: ... |
|
|
|
class CustomBusinessHour(BusinessHour): |
|
def __init__( |
|
self, |
|
n: int = ..., |
|
normalize: bool = ..., |
|
weekmask: str = ..., |
|
holidays: list | None = ..., |
|
calendar: OffsetCalendar | None = ..., |
|
start: str | time | Collection[str | time] = ..., |
|
end: str | time | Collection[str | time] = ..., |
|
offset: timedelta = ..., |
|
) -> None: ... |
|
|
|
class CustomBusinessMonthEnd(_CustomBusinessMonth): ... |
|
class CustomBusinessMonthBegin(_CustomBusinessMonth): ... |
|
class OffsetMeta(type): ... |
|
class DateOffset(RelativeDeltaOffset, metaclass=OffsetMeta): ... |
|
|
|
BDay = BusinessDay |
|
BMonthEnd = BusinessMonthEnd |
|
BMonthBegin = BusinessMonthBegin |
|
CBMonthEnd = CustomBusinessMonthEnd |
|
CBMonthBegin = CustomBusinessMonthBegin |
|
CDay = CustomBusinessDay |
|
|
|
def roll_qtrday( |
|
other: datetime, n: int, month: int, day_opt: str, modby: int |
|
) -> int: ... |
|
|
|
INVALID_FREQ_ERR_MSG: Literal["Invalid frequency: {0}"] |
|
|
|
def shift_months( |
|
dtindex: npt.NDArray[np.int64], |
|
months: int, |
|
day_opt: str | None = ..., |
|
reso: int = ..., |
|
) -> npt.NDArray[np.int64]: ... |
|
|
|
_offset_map: dict[str, BaseOffset] |
|
|