o @svdZddlmZddlmZddlmZddlZddlZdddZddd Zdd d Z dd d Z dddZ ddZ dS)zCA module for dealing with unknown string and environment encodings.)absolute_import)division)unicode_literalsNcCs:|durdStjs |St|tjr|S|pt}||S)zEncode the text string to a byte string. Args: string: str, The text string to encode. encoding: The suggested encoding if known. Returns: str, The binary string. N)sixPY2 isinstance binary_type _GetEncodingencode)stringencodingr >/tmp/google-cloud-sdk/lib/googlecloudsdk/core/util/encoding.pyEncodes    rc Cs|durdSt|tjst|tjr|}nzt|}Wnttfy+t|}Ynwt|tjr4|Sz|dWStyCYnw|rVz||WStyUYnwz|dWStyeYnwz|t WStywYnwz|t WStyYnw|dS)aReturns string with non-ascii characters decoded to UNICODE. UTF-8, the suggested encoding, and the usual suspects will be attempted in order. Args: data: A string or object that has str() and unicode() methods that may contain an encoding incompatible with the standard output encoding. encoding: The suggested encoding if known. Returns: A text string representing the decoded byte string. Nasciizutf-8z iso-8859-1) rr text_typer TypeError UnicodeErrorstrdecodesysgetfilesystemencodinggetdefaultencoding)datar r r r rDecode4sL           rcCs&t|}||}|dur|St|S)zReturns the decoded value of the env var name. Args: env: {str: str}, The env dict. name: str, The env var name. default: The value to return if name is not in env. Returns: The decoded value of the env var name. N)rgetr)envnamedefaultvaluer r rGetEncodedValues  r cCs8t||d}|dur||ddSt||d||<dS)a!Sets the value of name in env to an encoded value. Args: env: {str: str}, The env dict. name: str, The env var name. value: str or unicode, The value for name. If None then name is removed from env. encoding: str, The encoding to use or None to try to infer it. r N)rpop)rrrr r r rSetEncodedValues  r#cCs6|pt}i}t|D] \}}t||||dq |S)a%Encodes all the key value pairs in env in preparation for subprocess. Args: env: {str: str}, The environment you are going to pass to subprocess. encoding: str, The encoding to use or None to use the default. Returns: {bytes: bytes}, The environment to pass to subprocess. r!)r r iteritemsr#)rr encoded_envkvr r r EncodeEnvs r(cCstptS)z!Gets the default encoding to use.)rrrr r r rr sr )N) __doc__ __future__rrrrrrrr r#r(r r r r rs      [