U ]gO @s@UdZddlmZddlmZddlmZmZmZm Z m Z m Z m Z ddl mZmZdddd d ZGd d d eZeed deddeddeddeddedddZiZded<GdddeZeGddded ZGd!d"d"eZGd#d$d$eZGd%d&d&eZGd'd(d(eZGd)d*d*eZGd+d,d,eZd-S). ) annotations)total_ordering)Dict NamedTupleOptionalTupleTypeUnioncast)ClassVarListzUnion[bytes, str, int]str)valuereturncCst|tr|St|S)z- converts a byte/str or int to a str ) isinstancebytesdecoder)rr1/tmp/pip-unpacked-wheel-kizsipjx/limits/limits.py safe_string s rc@seZdZUded<ded<dS) GranularityintsecondsrnameN)__name__ __module__ __qualname____annotations__rrrrrs riQdayi'monthiyearihour<minutesecond)rr r!r"r$r&zDict[str, Type[RateLimitItem]] GRANULARITIEScs(eZdZdddddfdd ZZS)RateLimitItemMetarzTuple[type, ...]z(Dict[str, Union[Granularity, List[str]]])rparentsdctrcsHd|krg|d<t||||}d|krDttt|t|dd<|S)N __slots__ GRANULARITYr%)super__new__r r RateLimitItemr')clsrr)r*Z granularity __class__rrr.(szRateLimitItemMeta.__new__)rrrr. __classcell__rrr1rr('sr(c@seZdZUdZdddgZded<d$d d d d d dZed ddddZd dddZ d d dddZ dddddZ d dddZ ddddd Z d dd!d"Zd#S)%r/a] defines a Rate limited resource which contains the characteristic namespace, amount and granularity multiples of the rate limiting window. :param amount: the rate limit amount :param multiples: multiple of the 'per' :attr:`GRANULARITY` (e.g. 'n' per 'm' seconds) :param namespace: category for the specific rate limit namespaceamount multipleszClassVar[Granularity]r,r%LIMITERrz Optional[int]r)r5r6r4cCs"||_t||_t|pd|_dS)Nr%)r4rr5r6)selfr5r6r4rrr__init__Os zRateLimitItem.__init__bool)granularity_stringrcCs||jjkS)z Checks if this instance matches a *granularity_string* of type ``n per hour``, ``n per minute`` etc, by comparing with :attr:`GRANULARITY` )lowerr,r)r0r;rrrcheck_granularity_stringVs z&RateLimitItem.check_granularity_string)rcCs|jj|jS)zM :return: the duration the limit is enforced for in seconds. )r,rr6r8rrr get_expiryaszRateLimitItem.get_expiry) identifiersrcGs>ddd|Dt|jt|j|jjg}|jd|S)a" Constructs a key for the current limit and any additional identifiers provided. :param identifiers: a list of strings to append to the key :return: a string key identifying this resource with each identifier appended with a '/' delimiter. /cSsg|] }t|qSr)r).0krrr rsz)RateLimitItem.key_for..)joinrr5r6r,rr4)r8r@ remainderrrrkey_forhs   zRateLimitItem.key_forobject)otherrcCs2t|tr.|j|jko,|j|jko,|j|jkSdS)NF)rr/r5r,r6r8rIrrr__eq__|s    zRateLimitItem.__eq__cCs|jd|jd|jjS)Nz per r)r5r6r,rr>rrr__repr__szRateLimitItem.__repr__cCs|jj|jjkSN)r,rrJrrr__lt__szRateLimitItem.__lt__cCst|j|j|j|jfSrM)hashr4r5r6r,r>rrr__hash__szRateLimitItem.__hash__N)r%r7)rrr__doc__r+rr9 classmethodr=r?rGrKrLrNrPrrrrr/;s    r/) metaclassc@seZdZdZedZdS)RateLimitItemPerYearz) per year rate limited resource. r!NrrrrQ TIME_TYPESr,rrrrrTsrTc@seZdZdZedZdS)RateLimitItemPerMonthz* per month rate limited resource. r NrUrrrrrWsrWc@seZdZdZedZdS)RateLimitItemPerDayz( per day rate limited resource. rNrUrrrrrXsrXc@seZdZdZedZdS)RateLimitItemPerHourz) per hour rate limited resource. r"NrUrrrrrYsrYc@seZdZdZedZdS)RateLimitItemPerMinutez+ per minute rate limited resource. r$NrUrrrrrZsrZc@seZdZdZedZdS)RateLimitItemPerSecondz+ per second rate limited resource. r&NrUrrrrr[sr[N)rQ __future__r functoolsrtypingrrrrr r r Z limits.typingr r rrdictrVr'rtyper(r/rTrWrXrYrZr[rrrrs0  $  S