gX0LdZddlZddlmZddlZddlZddlZddlm Z ddl m Z ddl m Z ddl m Z ddl mZddl mZdd lmZej$eZej*e j,dZesej*e j0d Zd j3eZd j3ej*e j6d ZdZdZeeiZ e ej*ddZ!dZ#dZ$dZ%dZ&e!dfdZ'edddddfdZ(dZ)dZ*ddZ+ddZ,y#e"$rdZ!Y2wxYw)zProvides helper methods for talking to the Compute Engine metadata server. See https://cloud.google.com/compute/docs/metadata for more details. N)urljoin)_helpers)environment_vars) exceptions)metrics) transport)ExponentialBackoffzmetadata.google.internalzhttp://{}/computeMetadata/v1/z http://{}z169.254.169.254zmetadata-flavorGoogleGCE_METADATA_TIMEOUTz/sys/class/dmi/id/product_namecVt|rytjdk(rytS)aChecks to see if the code runs on Google Compute Engine Args: request (google.auth.transport.Request): A callable used to make HTTP requests. Returns: bool: True if the code runs on Google Compute Engine, False otherwise. TntF)pingosnamedetect_gce_residency_linuxrequests Y/var/www/openai/venv/lib/python3.12/site-packages/google/auth/compute_engine/_metadata.py is_on_gcerDs' G} ww$ & ''c ttd5}|jj}dddj t S#1swYxYw#t$rYywxYw)zDetect Google Compute Engine residency by smbios check on Linux Returns: bool: True if the GCE product name file is detected, False otherwise. rNF)open_GCE_PRODUCT_NAME_FILEreadstrip Exception startswith_GOOGLE)file_objcontents rrrYs\  (# .(mmo++-G/   g && / . s'AAAAA A&%A&ctj}tj|tj<t |}|D][} |t d||}|jjt}|jtjk(xr |tk(cSy#tj$r"}t j#d|||Yd}~d}~wwxYw)aChecks to see if the metadata server is available. Args: request (google.auth.transport.Request): A callable used to make HTTP requests. timeout (int): How long to wait for the metadata server to respond. retry_count (int): How many times to attempt connecting to metadata server using above timeout. Returns: bool: True if the metadata server is reachable, False otherwise. total_attemptsGET)urlmethodheaderstimeoutJCompute Engine Metadata server unavailable on attempt %s of %s. Reason: %sNF)_METADATA_HEADERScopyrmds_pingAPI_CLIENT_HEADERr _METADATA_IP_ROOTr)get_METADATA_FLAVOR_HEADERstatus http_clientOK_METADATA_FLAVOR_VALUErTransportError_LOGGERwarning) rr* retry_countr)backoffattemptresponsemetadata_flavores rrris& $$&G)0)9)9);GG % %& >1>#'== * ((  OO/    s AB''C:CCFc>t||}|in|} tj} |r| j||rd| d<t j || } t |} | D]N} || d| }|jtjvr#tjd| ||jNn%tjd j| t j|j }|jt"j$k(r|ry|jt"j&k(r=t j(|j*d d k(r t-j.|S|Stjd j| |j|j |#tj$r#}tjd| ||Yd}~zd}~wwxYw#t0$r,}tjd j|}||d}~wwxYw)aFetch a resource from the metadata server. Args: request (google.auth.transport.Request): A callable used to make HTTP requests. path (str): The resource to retrieve. For example, ``'instance/service-accounts/default'``. root (str): The full path to the metadata server root. params (Optional[Mapping[str, str]]): A mapping of query parameter keys to values. recursive (bool): Whether to do a recursive query of metadata. See https://cloud.google.com/compute/docs/metadata#aggcontents for more details. retry_count (int): How many times to attempt connecting to metadata server using above timeout. headers (Optional[Mapping[str, str]]): Headers for the request. return_none_for_not_found_error (Optional[bool]): If True, returns None for 404 error instead of throwing an exception. Returns: Union[Mapping, str]: If the metadata server returns JSON, a mapping of the decoded JSON is returned. Otherwise, the response content is returned as a string. Raises: google.auth.exceptions.TransportError: if an error occurred while retrieving metadata. Ntrue recursiver$r&)r'r(r)zSCompute Engine Metadata server unavailable on attempt %s of %s. Response status: %sr+zqFailed to retrieve {} from the Google Compute Engine metadata service. Compute Engine Metadata server unavailablez content-typezapplication/jsonzMReceived invalid JSON from the Google Compute Engine metadata service: {:.20}z^Failed to retrieve {} from the Google Compute Engine metadata service. Status: {} Response: {})rr,r-updater update_queryr r3rDEFAULT_RETRYABLE_STATUS_CODESr8r9rr7format from_bytesdatar4 NOT_FOUNDr5parse_content_typer)jsonloads ValueError)rpathrootparamsrCr:r)return_none_for_not_found_errorbase_url query_paramsheaders_to_user'r;r<r=r?r" caught_excnew_excs rr1r1sLtT"H2VL&++-Ng&$* [!   , 7C JJ??$x'9'9<((L n %| 33r)default)rrN)-__doc__rl http.clientclientr4rLloggingr urllib.parser google.authrrrrr google.auth._exponential_backoffr getLogger__name__r8getenvGCE_METADATA_HOST_GCE_METADATA_HOSTGCE_METADATA_ROOTrG_METADATA_ROOTGCE_METADATA_IPr0r2r6r,int_METADATA_DEFAULT_TIMEOUTrNr rrrrr1rZr]r`rqrrrs_ !  ("!? '  H % RYY/AA4H"**,F1778JK && BII..0AB,!,.DE" #IBII.Da$H I 9(* ' 4-f   $)l^.".<:"4c" !"sDD#"D#