:j vdZddlZddlmZddlmZmZmZmZddl m Z m Z eGddZ Gd d Z y) a GitHub Credentials Manager This module manages GitHub session tokens and API tokens with load balancing. It provides thread-safe access to multiple credentials for improved concurrency. Key Features: - Load balanced credential distribution - Support for both session tokens and API tokens - Thread-safe credential access - Usage statistics and monitoring - Automatic fallback between credential types N) dataclass)DictListOptionalTuple)BalancerStrategyceZdZUdZeed<eed<eed<eed<eed<eed<eed<d Zee ed <d Z ee ed <e d e fd Z e d e fdZe d efdZy )CredentialStatszCredential usage statisticstotal_requestssession_requeststoken_requestssessions_count tokens_countsession_percentagetoken_percentageN session_stats token_statsreturnc |jdkDS)zCheck if sessions are availabler)rselfs 7/root/.openclaw/workspace/harvester/tools/credential.py has_sessionszCredentialStats.has_sessions&s""Q&&c |jdkDS)zCheck if tokens are availabler)rrs r has_tokenszCredentialStats.has_tokens+s  1$$rc4|j|jzS)zTotal number of credentials)rrrs rtotal_credentialsz!CredentialStats.total_credentials0s""T%6%666r)__name__ __module__ __qualname____doc__int__annotations__floatrrrrpropertyboolrrr rrr r s%$(M8D>("&K$& 'd''%D%%7377rr ceZdZdZddeedeedefdZdeefdZdeefdZ dd e de eeffd Z de eeffd Z de fd Zde fd Zde fdZdeeddfdZdeeddfdZddZdefdZdefdZdefdZy) Credentialsz.GitHub credentials manager with load balancingsessionstokensstrategyc|s |s td|r|jng|_|r|jng|_|dk(rtj ntj }|jrt|j|nd|_|jrt|j|nd|_ tj|_ d|_ d|_d|_y)zInitialize credentials manager Args: sessions: List of GitHub session tokens tokens: List of GitHub API tokens strategy: Load balancing strategy ("round_robin" or "random") z.At least one session or token must be provided round_robinNr) ValueErrorcopyr-r.r ROUND_ROBINRANDOMr session_balancertoken_balancer threadingLocklockr rr)rr-r.r/ strategy_enums r__init__zCredentials.__init__9sMN N+3  '-fkkm2 19M0I,,x KO-- F]aFJkkht{{MBW[NN$  !rrc|jsy|j5|xjdz c_|xjdz c_|jj cdddS#1swYyxYw)z{Get next session token Returns: Optional[str]: Session token or None if no sessions available Nr)r6r:r rgetrs r get_sessionzCredentials.get_sessionSs` $$ YY /   1 $   ! !Q & !((,,. / / / AA((A1c|jsy|j5|xjdz c_|xjdz c_|jj cdddS#1swYyxYw)zqGet next API token Returns: Optional[str]: API token or None if no tokens available Nr)r7r:r rr>rs r get_tokenzCredentials.get_tokenas` "" YY -   1 $    1 $ &&**, - - -r@ prefer_tokenc|r"|jr|j}|r|dfS|jr|j}|r|dfS|s"|jr|j}|r|dfSt d)a#Get next credential with type preference Args: prefer_token: Whether to prefer API tokens over sessions Returns: Tuple[str, str]: (credential_value, credential_type) Raises: RuntimeError: If no credentials are available tokensessionzNo credentials available)r7rBr6r? RuntimeError)rrCrErFs rget_credentialzCredentials.get_credentialos} D//NN$Eg~%  &&(G )) 3 3NN$Eg~%566rc&|jdS)zGet any available credential Returns: Tuple[str, str]: (credential_value, credential_type) Raises: RuntimeError: If no credentials are available T)rC)rHrs rget_anyzCredentials.get_anys"""55rc,t|jS)zkCheck if sessions are available Returns: bool: True if sessions are available )r)r-rs rrzCredentials.has_sessionss DMM""rc,t|jS)zgCheck if tokens are available Returns: bool: True if tokens are available )r)r.rs rrzCredentials.has_tokenss DKK  rcF|jxs|jS)zyCheck if any credentials are available Returns: bool: True if any credentials are available )rrrs rhas_credentialszCredentials.has_credentialss   "7doo&77rNc|j5|r|jng|_|jr|jr&|jj |jnU|j r|j j ntj}t|j||_nd|_dddy#1swYyxYw)zcUpdate session tokens list Args: sessions: New list of session tokens N) r:r3r-r6 update_itemsr7r/r r4r )rr-r/s rupdate_sessionszCredentials.update_sessionss YY -/7HMMORDM}}(())66t}}E?C?R?Rt22;;X`XlXlH,4T]]H,MD)(,% - - - B-CC c|j5|r|jng|_|jr|jr&|jj |jnU|j r|j j ntj}t|j||_nd|_dddy#1swYyxYw)zYUpdate API tokens list Args: tokens: New list of API tokens N) r:r3r.r7rPr6r/r r4r )rr.r/s r update_tokenszCredentials.update_tokenss YY ++1&++-rDK{{&&''44T[[AAEAVAVt44==\d\p\pH*24;;*ID'&*# + + +rRc|j5d|_d|_d|_|jr|jj |j r|j j dddy#1swYyxYw)zReset usage statisticsrN)r:r rrr6resetr7rs r reset_statszCredentials.reset_statssl YY ,"#D $%D !"#D $$%%++-""##))+ , , ,s A"A88Bc N|j5|jdkDrMt|j|jz dzd}t|j|jz dzd}nd}d}t |j|j|jt |jt |j|||jr|jjnd|jr|jjnd cdddS#1swYyxYw)zkGet usage statistics Returns: CredentialStats: Comprehensive usage statistics rdgN) r rrrrrrrr) r:r roundrrr lenr-r.r6 get_statsr7)rrrs rr]zCredentials.get_statss YY ""Q&%*D,A,ADDWDW,W[^+^`a%b"#($*=*=@S@S*SWZ)Z\]#^ %("#& "#22!%!6!6#22"4==1 -#5!1CGCXCXd33==?^b?C?R?RD//99;X\    s DDD$czdt|jdt|jd|jdS)zWString representation Returns: str: String representation Credentials(sessions= , tokens=z , requests=))r\r-r.r rs r__str__zCredentials.__str__s= 's4=='9&:)C DTCUU`aeatat`uuvwwrc<d|jd|jdS)zbDetailed string representation Returns: str: Detailed representation r_r`ra)r-r.rs r__repr__zCredentials.__repr__s! 't}}oYt{{m1MMr)r1)T)rN)r!r"r#r$rstrr<rr?rBr)rrHrJrrrNrQrTrWr r]rbrdr*rrr,r,6s8 c DI  4 /Xc] / -8C= -7475c?7: 6sCx 6#d#!D!88-S -d-"+DI+$+" ,?6xxN#Nrr,)r$r8 dataclassesrtypingrrrrbalancerr r r r,r*rrrisB !..( 77 7:ONONr