o ;h.@sddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZddlmZmZddlmZeedfZeeZd eed eefd d ZGd ddZ d7dededeed eefddZe dedeed eee ee effddZ   d8dedededede!deed eee!eeffddZ"  d9ded ededeed eee!eff d!d"Z#d#e eeeefd$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$rrrrrr1Qs  "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_variablesWs z#DotEnv.set_as_environment_variablesr5cCs2|}||vr ||S|jrtd||jdS)z zKey %s not found in %s.N)r4rrrr)r$r5datarrrgetfs 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=)rrErrrrget_keyts rFr*c cstj|s#t|d|d }|dWdn1swYtjd|dd4}zt||d }||fVWdn1sCwYWntyWt|j wWdn1sbwYt |j |dS)Nw)moderr(F)rHrdeleter&) r)r*r+r,writetempfileNamedTemporaryFile BaseExceptionunlinknameshutilmove)r*rsourcedestrrrrewrites$     rTalwaysF 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 )rUautoneverzUnknown quote_mode: rUrZz'{}''z\'zexport = r&FTN) ValueErrorisalnumformatreplacerTrrr5rJrstringendswith)rrVrWrXrYrquote value_outline_outrRrSreplacedmissing_newlinerrrrset_keys6     rj 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*existsrrrTrrr5rJrrc)rrkrXrremovedrRrSrrrr unset_keys  rnvaluesr!csi}|D]:\}}|durd}n+t|}i|r#tj|n |tjdfdd|D}|||<q|S)Nr(c3s|]}|VqdSr")resolve).0atomenvrr sz$resolve_variables..)rupdater)r8join)ror! new_valuesrOr6resultatomsrrsrr2s      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*rlIOErrorr+dirnameabspathrwpardir)r*last_dir current_dir parent_dirrrr _walk_to_roots     rr'filenameraise_error_if_not_foundusecwdc Csdd}|s|sttddrt}n*t}t}|jj|kr0|jdus'J|j}|jj|ks |jj}tj tj |}t |D]}tj ||} tj | rU| SqB|r\tddS)z Search in increasingly higher folders for the given file Returns path to the file if found, or an empty string otherwise cSstddddgd}t|d S)z> Decide whether this is running in a REPL or IPython notebook __main__N__file__)fromlist) __import__hasattr)mainrrr_is_interactives z$find_dotenv.._is_interactivefrozenFNzFile not foundr()getattrsysr)getcwd _getframerf_code co_filenamef_backr*r|r}rrwr+r{) rrrrr*frame current_fileframe_filenamer| check_pathrrr find_dotenvs&      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. Nrrrr r!r)rrr;)rrrr!r rdotenvrrr load_dotenv7srcCs.|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_values]sr)rD)rUFrD)rUrD)r'FF)NNFFTrD)NNFTrD))r.loggingr)rPrrK collectionsr contextlibrtypingrrrrrr r r parserr r variablesrrBrA getLoggerr>rrrrFrTrCrjrnr2rrrrrrrrs  (    U    4   *  '