gdZddlZddlmZddlmZddlmZGddZ Gdd e Z Gd d e Z Gd d e Z Gdde Z y)z' Provides various throttling policies. N)cache)ImproperlyConfigured) api_settingsc"eZdZdZdZdZdZy) BaseThrottlez& Rate throttling of requests. ctd)zT Return `True` if the request should be allowed, `False` otherwise. z#.allow_request() must be overriddenNotImplementedErrorselfrequestviews W/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/rest_framework/throttling.py allow_requestzBaseThrottle.allow_requests""GHHc^|jjd}|jjd}tj}|C|dk(s||S|j d}|t |t | }|jS|rdj|j S|S)z Identify the machine making the request by parsing HTTP_X_FORWARDED_FOR if present and number of proxies is > 0. If not use all of HTTP_X_FORWARDED_FOR if it is available, if not use REMOTE_ADDR. HTTP_X_FORWARDED_FOR REMOTE_ADDRr,) METAgetr NUM_PROXIESsplitminlenstripjoin)r r xff remote_addr num_proxiesaddrs client_addrs r get_identzBaseThrottle.get_idents ll56ll&&}5 "..  "a3;""IIcNE[#e*!= =>K$$& &'*rwwsyy{#; ;rcy)zm Optionally, return a recommended number of seconds to wait before the next request. Nr s rwaitzBaseThrottle.wait*s rN)__name__ __module__ __qualname____doc__rr$r(r&rrrr sI <&rrc|eZdZdZeZej ZdZdZ e jZ dZ dZdZdZdZd Zd Zd Zy) SimpleRateThrottlea A simple cache implementation, that only requires `.get_cache_key()` to be overridden. The rate (requests / seconds) is set by a `rate` attribute on the Throttle class. The attribute is a string of the form 'number_of_requests/period'. Period should be one of: ('s', 'sec', 'm', 'min', 'h', 'hour', 'd', 'day') Previous request information used for throttling is stored in the cache. zthrottle_%(scope)s_%(ident)sNct|dds|j|_|j|j\|_|_y)Nrate)getattrget_rater0 parse_rate num_requestsdurationr's r__init__zSimpleRateThrottle.__init__Ds7tVT* DI+/??499+E(4=rctd)z Should return a unique cache-key which can be used for throttling. Must be overridden. May return `None` if the request should not be throttled. z#.get_cache_key() must be overriddenr r s r get_cache_keyz SimpleRateThrottle.get_cache_keyIs""GHHrct|dds$d|jjz}t| |j|j S#t $rd|j z}t|wxYw)zR Determine the string representation of the allowed request rate. scopeNz9You must set either `.scope` or `.rate` for '%s' throttlez+No default throttle rate set for '%s' scope)r1 __class__r)rTHROTTLE_RATESr:KeyError)r msgs rr2zSimpleRateThrottle.get_rateRsttWd+N>>**+C&s+ + ,&&tzz2 2 ,?$**LC&s+ + ,s A $A0ch|y|jd\}}t|}ddddd|d}||fS) z Given the request rate string, return a two tuple of: , )NN/<iiQ)smhdr)rint)r r0numperiodr4r5s rr3zSimpleRateThrottle.parse_rateasH <jjo V3x $U;F1IFh''rcr|jy|j|||_|jy|jj |jg|_|j |_|j ry|j d|j|jz krP|j j|j r*|j d|j|jz krPt|j |jk\r|jS|jS)z Implement the check to see if the request should be throttled. On success calls `throttle_success`. On failure calls `throttle_failure`. T)r0r8keyrrhistorytimernowr5poprr4throttle_failurethrottle_successr s rrz SimpleRateThrottle.allow_requestms 99 %%gt4 88 zz~~dhh3 ::<llt||B/488dmm3KK LL   llt||B/488dmm3KK t||  1 1 1((* *$$&&rc|jjd|j|jj |j |j|j y)zd Inserts the current request's timestamp along with the key into the cache. rT)rMinsertrOrsetrLr5r's rrRz#SimpleRateThrottle.throttle_successs? Atxx( txxt}}=rcy)zP Called when a request to the API has failed due to throttling. Fr&r's rrQz#SimpleRateThrottle.throttle_failuresrc|jr*|j|j|jdz z }n |j}|jt |jz dz}|dkry|t |z S)zG Returns the recommended next request time in seconds. rKrArN)rMr5rOr4rfloat)r remaining_durationavailable_requestss rr(zSimpleRateThrottle.waitsq <rr.ceZdZdZdZdZy)AnonRateThrottlez Limits the rate of API calls that may be made by a anonymous users. The IP address of the request will be used as the unique cache key. anonc|jr|jjry|j|j|j |dzSNr:ident)useris_authenticatedr]r:r$r s rr8zAnonRateThrottle.get_cache_keysC <r{sY 470##Lp>p>f ) $ ) ,- +- r