o h0@sddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZmZmZmZmZmZddlmZmZddlmZeedfZeeZd eed eefd d ZGd ddZ d7dededeed eefddZ e dedeed eee ee effddZ!   d8dedededede"deed eee"eeffddZ#  d9ded ededeed eee"eff d!d"Z$d#eeeeefd$e"d eeeeffd%d&Z%ded eefd'd(Z& )  d:d*ed+e"d,e"d efd-d.Z'     / d;deed0ee ed1e"d$e"d2e"deed e"fd3d4Z(    / ddt dVdS)Nrz3python-dotenv could not find configuration file %s..env) rospathisfileopenrrrrinfoioStringIO)r$rrrr _get_stream4s"  zDotEnv._get_streamcCsD|jr|jS|}|jrtt||jd|_|jSt||_|jS)zReturn dotenv as dict)r!)r#parser rresolve_variablesr!)r$ raw_valuesrrrdictCs  z DotEnv.dictccsZ|}tt|D]}|jdur|j|jfVq WddS1s&wYdSr")r0rrkeyvalue)r$rrrrrr1Ss  "z DotEnv.parsecCsJ|sdS|D]\}}|tjvr|jsq |dur"|tj|<q dS)zI Load the current dotenv as system environment variable. FNT)r4itemsr)environr!)r$kvrrrset_as_environment_variablesYs z#DotEnv.set_as_environment_variablesr5cCs2|}||vr ||S|jrtd||jdS) zKey %s not found in %s.N)r4rrrr)r$r5datarrrgeths z DotEnv.get)NFNTT)__name__ __module__ __qualname__r StrPathrstrboolr%rrr0rr4r r1r;r>rrrrr"s6  rutf-8r key_to_getrcCst|d|d|S)z Get the value of a given key from the given .env. Returns `None` if the key isn't found or doesn't have a value. T)rr)rr>)rrFrrrrget_keyus rGr*c cst|tjd|dd<}d}zt||d }||fVWdn1s*wYWntyD}z|}WYd}~nd}~wwWdn1sOwY|durat|j |dSt |j |d)NwF)moderdeleter&) pathlibPathtouchtempfileNamedTemporaryFiler, BaseExceptionshutilmovenamer)unlink)r*rdestrsourceerrrrrrewrites&  rXalwaysF key_to_set value_to_set quote_modeexportcCs|dvr td||dkp|dko| }|r$d|dd}n|}|r2d|d |d }n|d |d }t||d A\} } d } d } tt| D]} | j|kr\| |d } qM| | j j | j j d  } qM| sz| ru| d | |Wdn1swYd ||fS)z Adds or Updates a key/value to the given .env If the .env path given doesn't exist, fails instead of risking creating an orphan .env somewhere in the filesystem )rYautoneverzUnknown quote_mode: rYr^z'{}''z\'zexport = r&FTN) ValueErrorisalnumformatreplacerXrrr5writerstringendswith)rrZr[r\r]rquote value_outline_outrVrUreplacedmissing_newlinerrrrset_keys6     ro key_to_unsetcCstj|std|d|fSd}t||d"\}}tt|D]}|j|kr+d}q!| |j j q!Wdn1s=wY|sOtd||d|fS||fS)z Removes a given key from the given `.env` file. If the .env path given doesn't exist, fails. If the given key doesn't exist in the .env, fails. z(Can't delete from %s - it doesn't exist.NFr&Tz/Key %s not removed from %s - key doesn't exist.) r)r*existsrrrXrrr5rgrrh)rrpr\rremovedrVrUrrrr unset_keys"  rsvaluesr!csi}|D]:\}}|durd}n+t|}i|r#tj|n |tjdfdd|D}|||<q|S)Nr(c3s|]}|VqdSr")resolve).0atomenvrr sz$resolve_variables..)rupdater)r8join)rtr! new_valuesrSr6resultatomsrrxrr2s      r2ccstj|s tdtj|rtj|}d}tj|}||kr>|Vtjtj|tjj}||}}||ks#dSdS)zL Yield directories starting from the given directory up to the root zStarting path not foundN) r)r*rqIOErrorr+dirnameabspathr|pardir)r*last_dir current_dir parent_dirrrr _walk_to_roots     rr'filenameraise_error_if_not_foundusecwdc Csdd}dd}|s|s|sttddrt}n:t}t}|jj|ks/tj |jjsG|j dus6J|j }|jj|ks/tj |jjr/|jj}tj tj |}t |D]} tj| |} tj| rl| SqY|rstdd S) z Search in increasingly higher folders for the given file Returns path to the file if found, or an empty string otherwise cSsPttds ttdr dSz tddddgd}Wn ty!YdSwt|d S) z._is_interactivecSs tduSr")rgettracerrrr _is_debugger)s z!find_dotenv.._is_debuggerfrozenFNzFile not foundr()getattrrr)getcwd _getframerf_code co_filenamer*rqf_backrrrr|r+r) rrrrrr*frame current_fileframe_filenamer check_pathrrr find_dotenvs0     rTrrr cCs2|dur |dur t}t||||||d}|S)aParse a .env file and then load all the variables found as environment variables. Parameters: dotenv_path: Absolute or relative path to .env file. stream: Text stream (such as `io.StringIO`) with .env content, used if `dotenv_path` is `None`. verbose: Whether to output a warning the .env file is missing. override: Whether to override the system environment variables with the variables from the `.env` file. encoding: Encoding to be used to read the file. Returns: Bool: True if at least one environment variable is set else False If both `dotenv_path` and `stream` are `None`, `find_dotenv()` is used to find the .env file with it's default parameters. If you need to change the default parameters of `find_dotenv()`, you can explicitly call `find_dotenv()` and pass the result to this function as `dotenv_path`. Nrrrr r!r)rrr;)rrrr!r rdotenvrrr load_dotenvGsrcCs.|dur |dur t}t||||d|dS)a Parse a .env file and return its content as a dict. The returned dict will have `None` values for keys without values in the .env file. For example, `foo=bar` results in `{"foo": "bar"}` whereas `foo` alone results in `{"foo": None}` Parameters: dotenv_path: Absolute or relative path to the .env file. stream: `StringIO` object with .env content, used if `dotenv_path` is `None`. verbose: Whether to output a warning if the .env file is missing. encoding: Encoding to be used to read the file. If both `dotenv_path` and `stream` are `None`, `find_dotenv()` is used to find the .env file. NTr)rrr4)rrrr rrrr dotenv_valuesosr)rE)rYFrE)rYrE)r'FF)NNFFTrE)NNFTrE)*r.loggingr)rKrQrrN collectionsr contextlibrtypingrrrrrr r r parserr r variablesrrCrB getLoggerr?rrrrGrXrDrorsr2rrrrrrrrs  (    V    3 !  4  )