From bced9222a9caae04924e43b9ffb2d2cdb2488174 Mon Sep 17 00:00:00 2001 From: Paul Ganssle Date: Sun, 6 Nov 2016 15:41:23 -0500 Subject: Moved 'weekday' into a _common.py module. --- dateutil/relativedelta.py | 35 +++-------------------------------- 1 file changed, 3 insertions(+), 32 deletions(-) (limited to 'dateutil/relativedelta.py') diff --git a/dateutil/relativedelta.py b/dateutil/relativedelta.py index 6c71d2f..7e3bd12 100644 --- a/dateutil/relativedelta.py +++ b/dateutil/relativedelta.py @@ -8,41 +8,12 @@ from math import copysign from six import integer_types from warnings import warn -__all__ = ["relativedelta", "MO", "TU", "WE", "TH", "FR", "SA", "SU"] - - -class weekday(object): - __slots__ = ["weekday", "n"] - - def __init__(self, weekday, n=None): - self.weekday = weekday - self.n = n - - def __call__(self, n): - if n == self.n: - return self - else: - return self.__class__(self.weekday, n) - - def __eq__(self, other): - try: - if self.weekday != other.weekday or self.n != other.n: - return False - except AttributeError: - return False - return True - - __hash__ = None - - def __repr__(self): - s = ("MO", "TU", "WE", "TH", "FR", "SA", "SU")[self.weekday] - if not self.n: - return s - else: - return "%s(%+d)" % (s, self.n) +from ._common import weekday MO, TU, WE, TH, FR, SA, SU = weekdays = tuple([weekday(x) for x in range(7)]) +__all__ = ["relativedelta", "MO", "TU", "WE", "TH", "FR", "SA", "SU"] + class relativedelta(object): """ -- cgit v1.2.3