home *** CD-ROM | disk | FTP | other *** search
# Source Generated with Decompyle++ # File: in.pyc (Python 2.6) __author__ = 'Gustavo Niemeyer <gustavo@niemeyer.net>' __license__ = 'PSF License' import datetime import calendar __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 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 def __repr__(self): s = ('MO', 'TU', 'WE', 'TH', 'FR', 'SA', 'SU')[self.weekday] if not self.n: return s return '%s(%+d)' % (s, self.n) class relativedelta: def __init__(self, dt1 = None, dt2 = None, years = 0, months = 0, days = 0, leapdays = 0, weeks = 0, hours = 0, minutes = 0, seconds = 0, microseconds = 0, year = None, month = None, day = None, weekday = None, yearday = None, nlyearday = None, hour = None, minute = None, second = None, microsecond = None): if dt1 and dt2: if not isinstance(dt1, datetime.date) or not isinstance(dt2, datetime.date): raise TypeError, 'relativedelta only diffs datetime/date' not isinstance(dt2, datetime.date) if type(dt1) is not type(dt2): if not isinstance(dt1, datetime.datetime): dt1 = datetime.datetime.fromordinal(dt1.toordinal()) elif not isinstance(dt2, datetime.datetime): dt2 = datetime.datetime.fromordinal(dt2.toordinal()) self.years = 0 self.months = 0 self.days = 0 self.leapdays = 0 self.hours = 0 self.minutes = 0 self.seconds = 0 self.microseconds = 0 self.year = None self.month = None self.day = None self.weekday = None self.hour = None self.minute = None self.second = None self.microsecond = None self._has_time = 0 months = dt1.year * 12 + dt1.month - (dt2.year * 12 + dt2.month) self._set_months(months) dtm = self.__radd__(dt2) if dt1 < dt2: while dt1 > dtm: months += 1 self._set_months(months) dtm = self.__radd__(dt2) else: while dt1 < dtm: months -= 1 self._set_months(months) dtm = self.__radd__(dt2) delta = dt1 - dtm self.seconds = delta.seconds + delta.days * 86400 self.microseconds = delta.microseconds else: self.years = years self.months = months self.days = days + weeks * 7 self.leapdays = leapdays self.hours = hours self.minutes = minutes self.seconds = seconds self.microseconds = microseconds self.year = year self.month = month self.day = day self.hour = hour self.minute = minute self.second = second self.microsecond = microsecond if type(weekday) is int: self.weekday = weekdays[weekday] else: self.weekday = weekday yday = 0 if nlyearday: yday = nlyearday elif yearday: yday = yearday if yearday > 59: self.leapdays = -1 if yday: ydayidx = [ 31, 59, 90, 120, 151, 181, 212, 243, 273, 304, 334, 366] for idx, ydays in enumerate(ydayidx): if yday <= ydays: self.month = idx + 1 if idx == 0: self.day = ydays else: self.day = yday - ydayidx[idx - 1] break continue else: raise ValueError, 'invalid year day (%d)' % yday self._fix() def _fix(self): if abs(self.microseconds) > 999999: s = self.microseconds // abs(self.microseconds) (div, mod) = divmod(self.microseconds * s, 1000000) self.microseconds = mod * s self.seconds += div * s if abs(self.seconds) > 59: s = self.seconds // abs(self.seconds) (div, mod) = divmod(self.seconds * s, 60) self.seconds = mod * s self.minutes += div * s if abs(self.minutes) > 59: s = self.minutes // abs(self.minutes) (div, mod) = divmod(self.minutes * s, 60) self.minutes = mod * s self.hours += div * s if abs(self.hours) > 23: s = self.hours // abs(self.hours) (div, mod) = divmod(self.hours * s, 24) self.hours = mod * s self.days += div * s if abs(self.months) > 11: s = self.months // abs(self.months) (div, mod) = divmod(self.months * s, 12) self.months = mod * s self.years += div * s if self.hours and self.minutes and self.seconds and self.microseconds and self.hour is not None and self.minute is not None and self.second is not None or self.microsecond is not None: self._has_time = 1 else: self._has_time = 0 def _set_months(self, months): self.months = months if abs(self.months) > 11: s = self.months // abs(self.months) (div, mod) = divmod(self.months * s, 12) self.months = mod * s self.years = div * s else: self.years = 0 def __radd__(self, other): if not isinstance(other, datetime.date): raise TypeError, 'unsupported type for add operation' isinstance(other, datetime.date) if self._has_time and not isinstance(other, datetime.datetime): other = datetime.datetime.fromordinal(other.toordinal()) if not self.year: pass year = other.year + self.years if not self.month: pass month = other.month if self.months: month += self.months if month > 12: year += 1 month -= 12 elif month < 1: year -= 1 month += 12 if not self.day: pass day = min(calendar.monthrange(year, month)[1], other.day) repl = { 'year': year, 'month': month, 'day': day } for attr in [ 'hour', 'minute', 'second', 'microsecond']: value = getattr(self, attr) if value is not None: repl[attr] = value continue days = self.days if self.leapdays and month > 2 and calendar.isleap(year): days += self.leapdays ret = other.replace(**repl) + datetime.timedelta(days = days, hours = self.hours, minutes = self.minutes, seconds = self.seconds, microseconds = self.microseconds) if self.weekday: if not self.weekday.n: pass weekday = self.weekday.weekday nth = 1 jumpdays = (abs(nth) - 1) * 7 if nth > 0: jumpdays += ((7 - ret.weekday()) + weekday) % 7 else: jumpdays += (ret.weekday() - weekday) % 7 jumpdays *= -1 ret += datetime.timedelta(days = jumpdays) return ret def __rsub__(self, other): return self.__neg__().__radd__(other) def __add__(self, other): if not isinstance(other, relativedelta): raise TypeError, 'unsupported type for add operation' isinstance(other, relativedelta) if not other.leapdays: pass if not other.year: pass if not other.month: pass if not other.day: pass if not other.weekday: pass if not other.hour: pass if not other.minute: pass if not other.second: pass if not other.second: pass return relativedelta(years = other.years + self.years, months = other.months + self.months, days = other.days + self.days, hours = other.hours + self.hours, minutes = other.minutes + self.minutes, seconds = other.seconds + self.seconds, microseconds = other.microseconds + self.microseconds, leapdays = self.leapdays, year = self.year, month = self.month, day = self.day, weekday = self.weekday, hour = self.hour, minute = self.minute, second = self.second, microsecond = self.microsecond) def __sub__(self, other): if not isinstance(other, relativedelta): raise TypeError, 'unsupported type for sub operation' isinstance(other, relativedelta) if not other.leapdays: pass if not other.year: pass if not other.month: pass if not other.day: pass if not other.weekday: pass if not other.hour: pass if not other.minute: pass if not other.second: pass if not other.second: pass return relativedelta(years = other.years - self.years, months = other.months - self.months, days = other.days - self.days, hours = other.hours - self.hours, minutes = other.minutes - self.minutes, seconds = other.seconds - self.seconds, microseconds = other.microseconds - self.microseconds, leapdays = self.leapdays, year = self.year, month = self.month, day = self.day, weekday = self.weekday, hour = self.hour, minute = self.minute, second = self.second, microsecond = self.microsecond) def __neg__(self): return relativedelta(years = -(self.years), months = -(self.months), days = -(self.days), hours = -(self.hours), minutes = -(self.minutes), seconds = -(self.seconds), microseconds = -(self.microseconds), leapdays = self.leapdays, year = self.year, month = self.month, day = self.day, weekday = self.weekday, hour = self.hour, minute = self.minute, second = self.second, microsecond = self.microsecond) def __nonzero__(self): if not (self.years) and not (self.months) and not (self.days) and not (self.hours) and not (self.minutes) and not (self.seconds) and not (self.microseconds) and not (self.leapdays) and self.year is None and self.month is None and self.day is None and self.weekday is None and self.hour is None and self.minute is None and self.second is None: pass return not (self.microsecond is None) def __mul__(self, other): f = float(other) return relativedelta(years = self.years * f, months = self.months * f, days = self.days * f, hours = self.hours * f, minutes = self.minutes * f, seconds = self.seconds * f, microseconds = self.microseconds * f, leapdays = self.leapdays, year = self.year, month = self.month, day = self.day, weekday = self.weekday, hour = self.hour, minute = self.minute, second = self.second, microsecond = self.microsecond) def __eq__(self, other): if not isinstance(other, relativedelta): return False if self.weekday or other.weekday: if not (self.weekday) or not (other.weekday): return False if self.weekday.weekday != other.weekday.weekday: return False n1 = self.weekday.n n2 = other.weekday.n if self.years == other.years and self.months == other.months and self.days == other.days and self.hours == other.hours and self.minutes == other.minutes and self.seconds == other.seconds and self.leapdays == other.leapdays and self.year == other.year and self.month == other.month and self.day == other.day and self.hour == other.hour and self.minute == other.minute and self.second == other.second: pass return self.microsecond == other.microsecond def __ne__(self, other): return not self.__eq__(other) def __div__(self, other): return self.__mul__(1 / float(other)) def __repr__(self): l = [] for attr in [ 'years', 'months', 'days', 'leapdays', 'hours', 'minutes', 'seconds', 'microseconds']: value = getattr(self, attr) if value: l.append('%s=%+d' % (attr, value)) continue for attr in [ 'year', 'month', 'day', 'weekday', 'hour', 'minute', 'second', 'microsecond']: value = getattr(self, attr) if value is not None: l.append('%s=%s' % (attr, `value`)) continue return '%s(%s)' % (self.__class__.__name__, ', '.join(l)) 
(.txt)