o V h@sTddlZddlmZmZmZmZddlmZmZm Z m Z edZ GdddZ dS)N) HUB_API_ROOT HUB_WEB_ROOTPREFIXrequest_with_credentials)IS_COLABLOGGERSETTINGSemojisz/settings?tab=api+keysc@sheZdZdZdZZZddedefddZ dd e d efd d Z d efd dZ d efddZ ddZdS)Autha[ 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. Fapi_keyverbosecCs|dd}|ptdd|_|jr,|jtdkr'|r%ttddS|}n tr3| }n| }|rOt d|ji|rMttddSdS|r^ttdt d dSdS) 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 r uAuthenticated ✅Nu!New authentication successful ✅zGet API key from z" and then run 'yolo login API_KEY') splitrgetr rinfor authenticaterauth_with_cookiesrequest_api_keyupdate API_KEY_URL)selfr r successrH/var/www/vscode/kcb/lib/python3.10/site-packages/ultralytics/hub/auth.py__init__s&   z Auth.__init__ max_attemptsreturncCsvddl}t|D])}ttd|dd||dtd}|dd|_|r1d Sqt t td ) z'Prompt the user to input their API key.rNzLogin. Attempt z of zEnter API key from  rTuFailed to authenticate ❌) getpassrangerrrrrr rConnectionErrorr )rrr!attempts input_keyrrrrIs zAuth.request_api_keycCsxz#|}r tjtd|d}|ddstdWdStdty;d|_|_t t dYdSw) 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)rheaderrrrrrUs   zAuth.authenticatecCsjtsdSz#ttd}|ddr$|didd|_|WdStdty4d|_YdSw) 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#)rauthnrrrrhs   zAuth.auth_with_cookiescCs*|jr dd|jiS|jrd|jiSdS)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 )rrrrr'~s  zAuth.get_auth_headerN)r F)r)__name__ __module__ __qualname____doc__r+r model_keystrboolrintrrrr'rrrrr s -  r ) r(ultralytics.hub.utilsrrrrultralytics.utilsrrrr rr rrrrs