jdZddlmZddlmZddlmZddlZddlZd dZd dZd dZ d d Z d d Z d Z y) zCA module for dealing with unknown string and environment encodings.)absolute_import)division)unicode_literalsNc|ytjs|St|tjr|S|xs t }|j |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)stringencodings (lib/googlecloudsdk/core/util/encoding.pyEncodersH ^   M( M  '( x  c|yt|tjst|tjr|}n tj|}t|tjr|S |jdS#tt f$rt |}YJwxYw#t $rYnwxYw|r! |j|S#t $rYnwxYw |jdS#t $rYnwxYw |jtjS#t $rYnwxYw |jtjS#t $rYnwxYw|jdS)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) r r text_typer TypeError UnicodeErrorstrdecodesysgetfilesystemencodinggetdefaultencoding)datarr s rDecoder4s^ \  cmm$ 4(I F}}T"f  & M == !! | $4yf    ]]8 $$     == !!    ==224 55    ==//1 22   & | $$skA?.B?BB B+*B+1C C CC## C/.C/3"D D"!D"&"E EEcXt|}|j|}||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. )rgetr)envnamedefaultvalues rGetEncodedValuer$s0 $ ''$-% ] N rcht||}||j|dyt||||<y)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. rN)rpop)r r!r#rs rSetEncodedValuer(s6& x ($ ]GGD$ UX.#d)rc~|xs t}i}tj|D]\}}t|||||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()r r encoded_envkvs r EncodeEnvr.s@ '(+mmC daKA9! rcVtjxstjS)z!Gets the default encoding to use.)rrrrrr r s " " $ @(>(>(@@r)N) __doc__ __future__rrrrrrrr$r(r.r r0rrr3s;"J&' !0X%v(/4$Ar