o ]Ni@s4ddlZddlZddlZeeZGdddZdS)Nc@sHeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS) TokenManagerz? Manages Keepa API tokens, rate limiting, and refills. cCsL||_d|_d|_d|_d|_d|_t|j|_t|_t ddS)N<2di,z5TokenManager initialized without a blocking API call.) api_keyREFILL_RATE_PER_MINUTEMIN_TIME_BETWEEN_CALLS_SECONDSMIN_TOKEN_THRESHOLDtokens max_tokenstimelast_api_call_timestamplast_refill_timestamploggerinfo)selfrr4/var/www/agentarbitrage/keepa_deals/token_manager.py__init__ s zTokenManager.__init__cCsrt}||j}|dkr5|d|j}|dkr7t|j|j||_||_td|dd|jddSdSdS)zZ Calculates and adds tokens that have been refilled since the last check. rrz Refilled .2fz tokens. Current tokens: N)r rrminr r rdebug)rnowseconds_elapsed refill_amountrrr_refill_tokenss  zTokenManager._refill_tokenscCs||j|kS)zu Checks if there are enough tokens for a call without waiting. Also triggers a refill check. )rr )restimated_costrrrhas_enough_tokens*s zTokenManager.has_enough_tokensc CsXt}||j}||jkr"|j|}td|ddt||d}|jdkrKd|j}t ||j d}t d|jdd|d n1|j|j kr{|j d }||j}t ||j d}t d |j d |jdd|d |d n |dkr|j dkrt||nt dtd|td|d|jddS)a Checks if an API call can be made and waits if necessary. This method implements a "controlled deficit" strategy based on Keepa API behavior. Keepa API Rule: A call can be made as long as the token balance is positive. The call is allowed to drive the balance into a negative value. Our Strategy (Optimized): 1. **Hard Stop at Zero:** If tokens are <= 0, we must wait until they refill to a positive number. 2. **Aggressive Consumption:** If `current_tokens` > `MIN_TOKEN_THRESHOLD` (50), we allow the call immediately, even if it creates a deficit. We do NOT wait for a full bucket. This prevents "starvation" when an upserter task is simultaneously consuming the refill trickle. 3. **Smart Recovery:** If `current_tokens` drops below the threshold, we wait only until it recovers to (`threshold + buffer`), not `max_tokens`. zRate limit: Pausing for rz seconds.r rzZero or negative tokens. Have: z. Waiting for z! seconds to recover to 10 tokens.rzLow tokens (Below Threshold z ). Have: z seconds to recover to .zOZero refill rate, cannot wait for tokens. Pausing for 15 minutes as a fallback.iz1Permission granted for API call. Estimated cost: z. Current tokens: N)r rr rrsleeprr mathceilrwarningr error)rrrtime_since_last_call wait_durationwait_time_seconds tokens_neededrecovery_targetrrrrequest_permission_for_call2sH                z(TokenManager.request_permission_for_callcCsLddlm}td||j}|rd|vr||ddStddS)z} Authoritatively fetches the current token status from the Keepa API and updates the internal state. )get_token_statusz5Performing authoritative token sync with Keepa API... tokensLeftz;Failed to sync tokens. API did not return valid token data.N) keepa_apir-rrr_sync_tokens_from_responser%)rr- status_datarrr sync_tokensus    zTokenManager.sync_tokenscCs:|j}t||_t|_td|dd|jddS)zZ Authoritatively sets the token count from a provided API response value. zIToken count authoritatively synced from API response. Previous estimate: rz , New value: N)r floatr rrr)rtokens_left_from_apiold_token_countrrrr0s  z'TokenManager._sync_tokens_from_responsecCst|_||dS)zk Updates the token count and timestamp after an API call using the authoritative response. N)r rr0)rr4rrrupdate_after_calls zTokenManager.update_after_callN) __name__ __module__ __qualname____doc__rrrr+r2r0r6rrrrrsC r)r r"logging getLoggerr7rrrrrrs