o ~j6h@ @sdZddlZddlZddlZddlmZddlZddlZddlZddl Z ddl m Z m Z m Z mZmZddlZddlmZdZdaejdd d Zhd Zd d ZddZddZddZdCddZddZdDddZddZddZ dd Z!d!d"Z"d#d$Z#d%ee$e%fd&ee$e%e&ffd'd(Z'd)e&d&ee&fd*d+Z(d,ej)d&e*fd-d.Z+d,ej)d&e*fd/d0Z,d,ej)d1e&d2e&d3ee-d4ee e&e&fd&df d5d6Z.dEd3ee-d8e&d&e fd9d:Z/d;e d&e fde d&dfd?d@Z1d,ej)d;e d&dfdAdBZ2dS)Fz-Helper functions for commonly used utilities.N)Message)AnyDictMappingOptionalUnion) exceptionsgoogleF-)minutesseconds>id_token client_id accessToken access_token client_secret refresh_tokencsfdd}|S)a0Decorator that copies a method's docstring from another class. Args: source_class (type): The class that has the documented method. Returns: Callable: A decorator that will copy the docstring of the same named method in the source class to the decorated method. cs(|jrtdt|j}|j|_|S)a?Decorator implementation. Args: method (Callable): The method to copy the docstring to. Returns: Callable: the same method passed in with an updated docstring. Raises: google.auth.exceptions.InvalidOperation: if the method already has a docstring. zMethod already has a docstring.)__doc__rInvalidOperationgetattr__name__)method source_method source_classc/var/www/html/chefvision.cloud.itp360.com/venv/lib/python3.10/site-packages/google/auth/_helpers.py decoratorAs   z!copy_docstring..decoratorr)rrrrrcopy_docstring6s rcCst}||d<|S)aParse a 'content-type' header value to get just the plain media-type (without parameters). This is done using the class Message from email.message as suggested in PEP 594 (because the cgi is now deprecated and will be removed in python 3.13, see https://peps.python.org/pep-0594/#cgi). Args: header_value (str): The value of a 'content-type' header as a string. Returns: str: A string with just the lowercase media-type from the parsed 'content-type' header. If the provided content-type is not parsable, returns 'text/plain', the default value for textual files. z content-type)rget_content_type) header_valuemrrrparse_content_typeXsr#cCs tjtjj}|jdd}|S)z_Returns the current UTC datetime. Returns: datetime: The current time in UTC. N)tzinfo)datetimenowtimezoneutcreplace)r&rrrutcnowns r*cCst|S)zConvert a datetime object to the number of seconds since the UNIX epoch. Args: value (datetime): The datetime to convert. Returns: int: The number of seconds since the UNIX epoch. )calendartimegm utctimetuplevaluerrrdatetime_to_secs~s r0utf-8cCs6t|tr ||n|}t|tr|Std|)aConverts a string value to bytes, if necessary. Args: value (Union[str, bytes]): The value to be converted. encoding (str): The encoding to use to convert unicode to bytes. Defaults to "utf-8". Returns: bytes: The original value converted to bytes (if unicode) or as passed in if it started out as bytes. Raises: google.auth.exceptions.InvalidValue: If the value could not be converted to bytes. z%{0!r} could not be converted to bytes) isinstancestrencodebytesr InvalidValueformat)r/encodingresultrrrto_bytess  r:cCs6t|tr |dn|}t|tr|Std|)aoConverts bytes to a string value, if necessary. Args: value (Union[str, bytes]): The value to be converted. Returns: str: The original value converted to unicode (if bytes) or as passed in if it started out as unicode. Raises: google.auth.exceptions.InvalidValue: If the value could not be converted to unicode. r1z'{0!r} could not be converted to unicode)r2r5decoder3rr6r7)r/r9rrr from_bytess  r<csndurgtj|}tj|j}||fdd|D}tjj|dd}|j|d}tj |S)aUpdates a URL's query parameters. Replaces any current values if they are already present in the URL. Args: url (str): The URL to update. params (Mapping[str, str]): A mapping of query parameter keys to values. remove (Sequence[str]): Parameters to remove from the query string. Returns: str: The URL with updated query parameters. Examples: >>> url = 'http://example.com?a=1' >>> update_query(url, {'a': '2'}) http://example.com?a=2 >>> update_query(url, {'b': '3'}) http://example.com?a=1&b=3 >> update_query(url, {'b': '3'}, remove=['a']) http://example.com?b=3 Ncsi|] \}}|vr||qSrr).0keyr/removerr sz update_query..T)doseq)query) urllibparseurlparseparse_qsrCupdateitems urlencode_replace urlunparse)urlparamsr@parts query_params new_query new_partsrr?r update_querys     rScCs d|S)zConverts scope value to a string suitable for sending to OAuth 2.0 authorization servers. Args: scopes (Sequence[str]): The sequence of scopes to convert. Returns: str: The scopes formatted as a single string.  )joinscopesrrrscopes_to_strings rXcCs|sgS|dS)zConverts stringifed scopes value to a list. Args: scopes (Union[Sequence, str]): The string of space-separated scopes to convert. Returns: Sequence(str): The separated scopes. rT)splitrVrrrstring_to_scopess  rZcCs(t|}|dt| d}t|S)zDecodes base64 strings lacking padding characters. Google infrastructure tends to omit the base64 padding characters. Args: value (Union[str, bytes]): The encoded value. Returns: bytes: The decoded value =)r:lenbase64urlsafe_b64decode)r/ b64stringpaddedrrrpadded_urlsafe_b64decodes  rbcCst|dS)auEncodes base64 strings removing any padding characters. `rfc 7515`_ defines Base64url to NOT include any padding characters, but the stdlib doesn't do that by default. _rfc7515: https://tools.ietf.org/html/rfc7515#page-6 Args: value (Union[str|bytes]): The bytes-like value to encode Returns: Union[str|bytes]: The encoded value r[)r^urlsafe_b64encoderstripr.rrrunpadded_urlsafe_b64encodesrecCs tjdkS)zCheck if the Python interpreter is Python 2 or 3. Returns: bool: True if the Python interpreter is Python 3 and False otherwise. )r r)sys version_inforrrr is_python_3"s rhdatareturncCst|tr7i}|D])\}}|tvr"t|ttfs"t||||<q t|ttfr0t|||<q |||<q |St|trLg}|D] }|t|q@|Stt |S)a Hashes sensitive information within a dictionary. Args: data: The dictionary containing data to be processed. Returns: A new dictionary with sensitive values replaced by their SHA512 hashes. If the input is a list, returns a list with each element recursively processed. If the input is neither a dict nor a list, returns the type of the input as a string. ) r2dictrI_SENSITIVE_FIELDSlist _hash_value_hash_sensitive_infoappendr3type)ri hashed_datar>r/ hashed_listvalrrrro+s    ro field_namecCsD|durdSt|d}t}|||}d|d|S)z3Hashes a value and returns a formatted hash string.Nr1hashed_-)r3r4hashlibsha512rH hexdigest)r/ru encoded_value hash_object hex_digestrrrrnNs rnloggercCs|jgkp|jtjkp|j S)zDetermines whether `logger` has non-default configuration Args: logger: The logger to check. Returns: bool: Whether the logger has any non-default configuration. )handlerslevelloggingNOTSET propagate)r~rrr_logger_configuredYs rcCs,tstt}t|sd|_da|tjS)z Checks if debug logging is enabled for the given logger. Args: logger: The logging.Logger instance to check. Returns: True if debug logging is enabled, False otherwise. FT)_LOGGING_INITIALIZEDr getLogger_BASE_LOGGER_NAMErr isEnabledForDEBUG)r~ base_loggerrrris_logging_enabledgs   rrrMbodyheaderscCsXt|r*|rd|vr|dnd}t||d}t|}|jdd||||diddSdS) aE Logs an HTTP request at the DEBUG level if logging is enabled. Args: logger: The logging.Logger instance to use. method: The HTTP method (e.g., "GET", "POST"). url: The URL of the request. body: The request body (can be None). headers: The request headers (can be None). z Content-Type) content_typezMaking request... httpRequest)rrMrrextraN)r_parse_request_bodyrodebug)r~rrMrrr json_body logged_bodyrrr request_logs   rrrc Cs|durdSz|d}Wn ttfyYdSw|}|r$d|vr9zt|WStjtfy8|YSwd|vrNtj |}dd| D}|Sd|vrT|SdS)a Parses a request body, handling bytes and string types, and different content types. Args: body (Optional[bytes]): The request body. content_type (str): The content type of the request body, e.g., "application/json", "application/x-www-form-urlencoded", or "text/plain". If empty, attempts to parse as JSON. Returns: Parsed body (dict, str, or None). - JSON: Decodes if content_type is "application/json" or None (fallback). - URL-encoded: Parses if content_type is "application/x-www-form-urlencoded". - Plain text: Returns string if content_type is "text/plain". - None: Returns if body is None, UTF-8 decode fails, or content_type is unknown. Nr1zapplication/jsonz!application/x-www-form-urlencodedcSsi|] \}}||dqS)rr)r=kvrrrrAsz'_parse_request_body..z text/plain) r;UnicodeDecodeErrorAttributeErrorlowerjsonloadsJSONDecodeError TypeErrorrDrErGrI)rrbody_str parsed_queryr9rrrrs*   rresponsecCs$z|}|WStyYdSw)a Parses a response, attempting to decode JSON. Args: response: The response object to parse. This can be any type, but it is expected to have a `json()` method if it contains JSON. Returns: The parsed response. If the response contains valid JSON, the decoded JSON object (e.g., a dictionary or list) is returned. If the response does not have a `json()` method or if the JSON decoding fails, None is returned. N)r Exception)r json_responserrr_parse_responses  rparsed_responsecCst|}|jdd|iddS)a Logs a parsed HTTP response at the DEBUG level. This internal helper function takes a parsed response and logs it using the provided logger. It also applies a hashing function to potentially sensitive information before logging. Args: logger: The logging.Logger instance to use for logging. parsed_response: The parsed HTTP response object (e.g., a dictionary, list, or the original response if parsing failed). zResponse received... httpResponserN)ror)r~rlogged_responserrr_response_log_basesrcCs"t|rt|}t||dSdS)z Logs an HTTP response at the DEBUG level if logging is enabled. Args: logger: The logging.Logger instance to use. response: The HTTP response object to log. N)rrr)r~rrrrr response_logsr)r1)N)r)3rr^r+r% email.messagerrxrrrftypingrrrrrrD google.authrrr timedeltaREFRESH_THRESHOLDrlrr#r*r0r:r<rSrXrZrbrerhrkrmr3rornLoggerboolrrr5rrrrrrrrrsb   "  - $ #  $&