Fi^ddlmZddlmZmZmZmZddlmZm Z m Z m Z edZ GddZ y)) annotations) HUB_API_ROOT HUB_WEB_ROOTPREFIXrequest_with_credentials)IS_COLABLOGGERSETTINGSemojisz/settings?tab=api+keyscFeZdZdZdxZxZZd d dZd d dZd dZ d dZ dZ y)AuthauManages authentication processes including API key handling, cookie-based authentication, and header generation. The class supports different methods of authentication: 1. Directly using an API key. 2. Authenticating using browser cookies (specifically in Google Colab). 3. Prompting the user to enter an API key. Attributes: id_token (str | bool): Token used for identity verification, initialized as False. api_key (str | bool): API key for authentication, initialized as False. model_key (bool): Placeholder for model key, initialized as False. Methods: authenticate: Attempt to authenticate with the server using either id_token or API key. auth_with_cookies: Attempt to fetch authentication via cookies and set id_token. get_auth_header: Get the authentication header for making API requests. request_api_key: Prompt the user to input their API key. Examples: Initialize Auth with an API key >>> auth = Auth(api_key="your_api_key_here") Initialize Auth without API key (will prompt for input) >>> auth = Auth() FcH|jddd}|xstjdd|_|jrR|jtjdk(r|rt j t dy|j}n'tr|j}n|j}|rAtjd|ji|rt j t dyy|r$t j t d td yy) aGInitialize Auth class and authenticate user. Handles API key validation, Google Colab authentication, and new key requests. Updates SETTINGS upon successful authentication. Args: api_key (str): API key or combined key_id format. verbose (bool): Enable verbose logging. _rapi_keyuAuthenticated ✅Nu!New authentication successful ✅zGet API key from z" and then run 'yolo login API_KEY') splitr getrr infor authenticaterauth_with_cookiesrequest_api_keyupdate API_KEY_URL)selfrverbosesuccesss V/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/ultralytics/hub/auth.py__init__z Auth.__init__(s--Q'*=(,,y""=  <<||x||I66KK6(*; <=++- ,,.G**,G  OOY 5 6 vh&GHI  KK6("3K=@bc dc2ddl}t|D]k}tjtd|dzd||jdt d}|j ddd|_|jsky tttd ) zPrompt the user to input their API key. Args: max_attempts (int): Maximum number of authentication attempts. Returns: (bool): True if authentication is successful, False otherwise. rNzLogin. Attempt rz of zEnter API key from  rTuFailed to authenticate ❌) getpassranger rrrrrrConnectionErrorr )r max_attemptsr#attempts input_keys rrzAuth.request_api_keyTs l+ H KK6(/(Q,tL>R S*=k]!(LMI$??3215DL  "   fx/I%JKLLr cDddl} |jx}rF|jtd|}|j j dds t dyt d #t $r-dx|_|_tjtd YywxYw) zAttempt to authenticate with the server using either id_token or API key. Returns: (bool): True if authentication is successful, False otherwise. rNz/v1/auth)headersrFzUnable to authenticate.Tz#User has not authenticated locally.zInvalid API key) requestsget_auth_headerpostrjsonrr%id_tokenrr warningr)rr+headerrs rrzAuth.authenticategs  --//v/MM\N(";VMLvvx||Iu5)*CDD!"GH H +0 0DMDL NNfX_5 6 sAA) A))3BBc tsy ttd}|jddr8|jdijdd|_|j yt d#t $r d|_YywxYw) zAttempt to fetch authentication via cookies and set id_token. User must be logged in to HUB and running in a supported browser. Returns: (bool): True if authentication is successful, False otherwise. Fz /v1/auth/autordataidTokenNTz/Unable to fetch browser authentication details.)rrrrr/rr%)rauthns rrzAuth.auth_with_cookies{s , ~]-KLEyyE* % &" 5 9 9)T J !!#!"ST T !DM sAA0% A00BBcr|jrdd|jiS|jrd|jiSy)zGet the authentication header for making API requests. Returns: (dict | None): The authentication header if id_token or API key is set, None otherwise. authorizationzBearer z x-api-keyN)r/r)rs rr,zAuth.get_auth_headers< ==#wt}}o%>? ? \\. .r N)rF)rstrrbool))r&intreturnr:)r=r:) __name__ __module__ __qualname____doc__r/r model_keyrrrrr,r rr r s44&+*H*w*eXM&(* /r r N) __future__rultralytics.hub.utilsrrrrultralytics.utilsrr r r rr rCr rrGs/#^^@@45 N/N/r