o g+@sBdZddlmZddlmZmZddZddZGdd d Zd S) z This module defines a Duration class. The class Duration allows to define durations in years and months and can be used as limited replacement for timedelta objects. ) timedelta) ROUND_FLOORDecimalcCs@||||}}||t}|||}||7}t||fS)z- A divmod function with boundaries. ) to_integralrint)vallowhighabdivmodrI/home/ubuntu/webapp/venv/lib/python3.10/site-packages/isodate/duration.pyfquotmod s   rcCsD|dvrdS|dvr dS|ddks|ddkr |ddkr d Sd S) zO Determines the number of days of a specific month in a specific year. ) ) irdrr)yearmonthrrrmax_days_in_months$r#c@seZdZdZ         d"ddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ e Z ddZeZddZddZddZddZd#d d!ZdS)$DurationaY A class which represents a duration. The difference to datetime.timedelta is, that this class handles also differences given in years and months. A Duration treats differences given in year, months separately from all other components. A Duration can be used almost like any timedelta object, however there are some restrictions: * It is not really possible to compare Durations, because it is unclear, whether a duration of 1 year is bigger than 365 days or not. * Equality is only tested between the two (year, month vs. timedelta) basic components. A Duration can also be converted into a datetime object, but this requires a start date or an end date. The algorithm to add a duration to a date is defined at http://www.w3.org/TR/xmlschema-2/#adding-durations-to-dateTimes rc CsRt|ts tt|}t| tstt| } ||_| |_t||||||||_dS)zN Initialise this Duration instance with the given parameters. N) isinstancerstrmonthsyearsrtdelta) selfdaysseconds microseconds millisecondsminuteshoursweeksr'r(rrr__init__As     zDuration.__init__cCs|jSN)__dict__r*rrr __getstate__ZszDuration.__getstate__cCs|j|dSr3)r4update)r*staterrr __setstate__]szDuration.__setstate__cCs t|j|S)zU Provide direct access to attributes of included timedelta instance. )getattrr))r*namerrr __getattr__`s zDuration.__getattr__cCs\g}|jr |d|j|jr!d}|jdkrd}|||j|t|jd|S)zW Return a string representation of this duration similar to timedelta. z%d yearsz %d monthsrz%d monthz, )r(appendr'r&r)join)r*paramsfmtrrr__str__fs  zDuration.__str__cCs.d|jj|jj|jj|jj|jj|j|jfS)z= Return a string suitable for repr(x) calls. z&%s.%s(%d, %d, %d, years=%d, months=%d)) __class__ __module____name__r)r+r,r-r(r'r5rrr__repr__uszDuration.__repr__cCst|j|j|jfS)zp Return a hash of this instance so that it can be used in, for example, dicts and sets. )hashr)r'r(r5rrr__hash__szDuration.__hash__cCs"t|j |j d}|j |_|S)zg A simple unary minus. Returns a new Duration instance with all it's negated. r(r')r$r(r'r))r* negdurationrrr__neg__s zDuration.__neg__c Cst|trt|j|j|j|jd}|j|j|_|SzLt|jr+t|js/td|j|j}t |dd\}}|j |j|}t ||}|j |krR|}n|j }|j t|t|t|d}|j|WStyqYnwzt|j|jd}|j||_|WStyYtSw)ze Durations can be added with Duration, timedelta, date and datetime objects. rH>fractional years or months not supported for date calculationsr r!r"day)r%r$r(r'r)float is_integer ValueErrorr"rr!r#rNreplacerAttributeErrorNotImplemented) r*other newdurationnewmonthcarrynewyearmaxdaysnewdaynewdtrrr__add__sB        zDuration.__add__cCs6t|trt|j||j|d}|j||_|StS)NrH)r%rr$r(r'r)rTr*rUrVrrr__mul__s  zDuration.__mul__cCspt|trt|j|j|j|jd}|j|j|_|Szt|j|jd}|j||_|WSty7YtSw)zj It is possible to subtract Duration and timedelta objects from Duration objects. rH)r%r$r(r'r) TypeErrorrTr^rrr__sub__s   zDuration.__sub__c Cst|trt}||_||SzLt|jrt|js"td|j |j}t |dd\}}|j |j|}t ||}|j |krE|}n|j }|jt|t|t|d}||jWStyeYtSw)a@ It is possible to subtract Duration objects from date, datetime and timedelta objects. TODO: there is some weird behaviour in date - timedelta ... if timedelta has seconds or microseconds set, then date - timedelta != date + (-timedelta) for now we follow this behaviour to avoid surprises when mixing timedeltas with Durations, but in case this ever changes in the stdlib we can just do: return -self + other instead of all the current code rKrrLrM)r%rr$r)rOr(rPr'rQr"rr!r#rNrRrrSrT) r*rUtmpdurrWrXrYrZr[r\rrr__rsub__s0      zDuration.__rsub__cCs`t|tr|jd|j|jd|jkr|j|jkrdSdS|jdkr.|jdkr.|j|kSdS)z If the years, month part and the timedelta part are both equal, then the two Durations are considered equal. rTFrr%r$r(r'r)r*rUrrr__eq__    zDuration.__eq__cCs`t|tr|jd|j|jd|jks|j|jkrdSdS|jdkr.|jdkr.|j|kSdS)z If the years, month part or the timedelta part is not equal, then the two Durations are considered not equal. rTFrrdrerrr__ne__rgzDuration.__ne__NcCsP|dur |dur td|dur|durtd|dur"|||S|||S)z Convert this duration into a timedelta object. This method requires a start datetime or end datetimem, but raises an exception if both are given. Nzstart or end requiredzonly start or end allowed)rQ)r*startendrrr totimedelta/s  zDuration.totimedelta) rrrrrrrrr)NN)rDrC __qualname____doc__r2r6r9r<rArErGrJr]__radd__r___rmul__rarcrfrhrkrrrrr$*s8  /*r$N) rmdatetimerdecimalrrrr#r$rrrrs