U ,c@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:|dkr dStjs|St|tjr&|S|p.t}||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 9/tmp/pip-unpacked-wheel-1xt1w7un/fire/console/encoding.pyEncodes   rc Cs.|dkr dSt|tjs$t|tjr*|}n0zt|}Wn ttfk rXt|}YnXt|tjrj|Sz |dWStk rYnX|rz ||WStk rYnXz |dWStk rYnXz|t WStk rYnXz|t WStk r"YnX|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. Nasciiutf8z iso-8859-1) rr text_typer TypeError UnicodeErrorstrdecodesysgetfilesystemencodinggetdefaultencoding)datar r r r rDecode4s@    rcCs&t|}||}|dkr|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}|dkr$||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)rrr r r r rSetEncodedValues   r$cs"ptfddt|DS)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. cs&i|]\}}t|dt|dqS)r")r).0kvr"r r s zEncodeEnv..)r r iteritems)rr r r"r EncodeEnvs  r*cCstptS)z!Gets the default encoding to use.)rrrr r r rr sr )N)N)N)N)N) __doc__ __future__rrrrrrrr!r$r*r r r r rs     [