a yf@sTddlZddlmZmZmZmZddlmZmZm Z m Z edZ GdddZ dS)N) HUB_API_ROOT HUB_WEB_ROOTPREFIXrequest_with_credentials)IS_COLABLOGGERSETTINGSemojisz/settings?tab=api+keysc@sTeZdZdZdZZZdddZdddZe d d d Z e d d d Z ddZ dS)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 or bool): Token used for identity verification, initialized as False. api_key (str or bool): API key for authentication, initialized as False. model_key (bool): Placeholder for model key, initialized as False. FcCs|dd}|ptdd|_|jrX|jtdkrN|rJttddS|}ntrf| }n| }|rt d|ji|rttdn|rttdt d dS) 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. _rapi_keyr uAuthenticated ✅Nu!New authentication successful ✅zGet API key from z& and then run 'yolo hub login API_KEY') splitrgetr rinfor authenticaterauth_with_cookiesrequest_api_keyupdate API_KEY_URL)selfr verbosesuccessrP/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/hub/auth.py__init__s"   z Auth.__init__cCszddl}t|D]R}ttd|dd||dtd}|dd|_|rd Sqt t td dS) zX Prompt the user to input their API key. Returns the model ID. rNzLogin. Attempt z of zEnter API key from  r TuFailed to authenticate ❌) getpassrangerrrrrr rConnectionErrorr )rZ max_attemptsrZattemptsZ input_keyrrrrIs zAuth.request_api_key)returncCszJ|}r@tjtd|d}|dds:tdWdStdWn0tyzd|_|_t t dYdS0d S) 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.uInvalid API key ⚠️N) get_auth_headerrequestspostrjsonrr!id_tokenr rwarningr)rheaderrrrrrYs    zAuth.authenticatecCsrtsdSzJttd}|ddrH|didd|_|WdStdWntyld|_YdS0dS) 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/autordataZidTokenNTz/Unable to fetch browser authentication details.)rrrrr(rr!)rZauthnrrrrls   zAuth.auth_with_cookiescCs*|jrdd|jiS|jr&d|jiSdS)z Get the authentication header for making API requests. Returns: (dict): The authentication header if id_token or API key is set, None otherwise. authorizationzBearer z x-api-keyN)r(r )rrrrr$szAuth.get_auth_headerN)r F)r) __name__ __module__ __qualname____doc__r(r Z model_keyrrboolrrr$rrrrr s  - r ) r%Zultralytics.hub.utilsrrrrZultralytics.utilsrrrr rr rrrrs