|hZddlZddlmZmZmZmZddlmZmZm Z m Z edZ GddZ y)N) HUB_API_ROOT HUB_WEB_ROOTPREFIXrequest_with_credentials)IS_COLABLOGGERSETTINGSemojisz/settings?tab=api+keysc^eZdZdZdxZxZZd dedefdZ dde defdZ defd Z defd Z d Zy )Authaz Manages 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() Fapi_keyverbosecH|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) aP Initialize 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. _rr uAuthenticated ✅Nu!New authentication successful ✅zGet API key from z" and then run 'yolo login API_KEY') splitr getr rinfor authenticaterauth_with_cookiesrequest_api_keyupdate API_KEY_URL)selfr rsuccesss S/var/www/html/test/engine/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 d max_attemptsreturnc2ddl}t|D]k}tjtd|dzd||jdt d}|j ddd|_|jsky tttd ) z Prompt 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 ❌) getpassrangerrrrrr rConnectionErrorr )rr r$attempts input_keys rrzAuth.request_api_keyVs l+ H KK6(/(Q,tL>R S*=k]!(LMI$??3215DL  "   fx/I%JKLLrcD |jx}rJtjtd|}|j j dds t dyt d#t $r-dx|_|_tjtdYywxYw) z Attempt to authenticate with the server using either id_token or API key. Returns: (bool): True if authentication is successful, False otherwise. z/v1/auth)headersrFzUnable to authenticate.Tz#User has not authenticated locally.zInvalid API key) get_auth_headerrequestspostrjsonrr&id_tokenr rwarningr)rheaderrs rrzAuth.authenticatejs --//v/MM\N(";VLvvx||Iu5)*CDD!"GH H +0 0DMDL NNfX_5 6 sAA) A))3BBc tsy ttd}|jddr8|jdijdd|_|j yt d#t $r d|_YywxYw) z Attempt 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)z Get 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%>? ? \\. .rN)rF))__name__ __module__ __qualname____doc__r/r model_keystrboolrintrrrr+rrr r sb6&+*H*w+e+e4+eZMCMM(d&4, /rr ) r,ultralytics.hub.utilsrrrrultralytics.utilsrrr r rr rBrrrEs/^^@@45 R/R/r