o b-@sddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZddlmZmZddlmZeeZejdkrTejZneZd eed eefd d ZGd ddZ d7deeefdedeed eefddZ!e deeefdeed eee ee effddZ"   d8deeefdededede#deed eee#eeffddZ$  d9deeefd 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;deeedfd0ee ed1e#d$e#d2e#deed e#fd3d4Z)    / d|D]\}}|tjvr|jsq|dur|tj|<qdS)zI Load the current dotenv as system environment variable. NT)r5itemsr+environr#)r&kvrrrset_as_environment_variablesVs z#DotEnv.set_as_environment_variablesr6cCs2|}||vr ||S|jrtd||jdS)z zKey %s not found in %s.N)r5r rrr)r&r6datarrrgetbs z DotEnv.get)NFNTT)__name__ __module__ __qualname__r r str _PathLikerboolr'rrr2rr5r r3r<r>rrrrr!s6    rutf-8r key_to_getr!cCst|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)r r!)rr>)rrFr!rrrget_keyps rGr,c cszYtj|s%tj|d|d }|dWdn1s wYtjdd|d$}tj||d }||fVWdn1sEwYWdn1sTwYWntyotj|j rnt |j wt |j |dS)Nzw+r(r*F)modedeleter!) r+r,r-r.r/writetempfileNamedTemporaryFile BaseExceptionnameunlinkshutilmove)r,r!sourcedestrrrrewrite}s&     rTalwaysF key_to_set value_to_set quote_modeexportcCs|dvr td||dkp|dko| }|r$d|dd}n|}|r/d||}nd ||}t||d A\} } d } d } tt| D]} | j|krW| |d } qH| | j j | j j d  } qH| su| rp| 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\'z export {}={} z{}={} r(FT N) ValueErrorformatisalnumreplacerTrrr6rJrstringendswith)rrVrWrXrYr!quoteZ value_outZline_outrRrSZreplacedZmissing_newlinerrrrset_keys6      re 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 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,existsrrrTrrr6rJrrb)rrfrXr!removedrRrSrrrr unset_keys  rivaluesr#csi}|D]:\}}|durd}n+t|}i|r#tj|n |tjdfdd|D}|||<q|S)Nr*c3s|]}|VqdSr$)resolve).0Zatomenvrr sz$resolve_variables..)rupdater+r9join)rjr# new_valuesrNr7resultZatomsrrmrr4s      r4ccstj|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,rgIOErrorr-dirnameabspathrqpardir)r,Zlast_dirZ current_dir parent_dirrrr _walk_to_roots     ryr)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 _getframer~f_code co_filenamef_backr,rurvryrqr-rt) rzr{r|rr,frameZ current_fileframe_filenameru check_pathrrr find_dotenv s&      rTrr r"cCs2|dur |dur t}t||||||d}|S)azParse a .env file and then load all the variables found as environment variables. - *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. Defaults to `False`. - *override*: whether to override the system environment variables with the variables in `.env` file. Defaults to `False`. - *encoding*: encoding to be used to read the file. If both `dotenv_path` and `stream`, `find_dotenv()` is used to find the .env file. Nrrr r"r#r!)rrr<)rrr r#r"r!Zdotenvrrr load_dotenv5srcCs.|dur |dur t}t||||d|dS)a Parse a .env file and return its content as a dict. - *dotenv_path*: absolute or relative path to .env file. - *stream*: `StringIO` object with .env content, used if `dotenv_path` is `None`. - *verbose*: whether to output a warning the .env file is missing. Defaults to `False`. in `.env` file. Defaults to `False`. - *encoding*: encoding to be used to read the file. If both `dotenv_path` and `stream`, `find_dotenv()` is used to find the .env file. NTr)rrr5)rrr r"r!rrr dotenv_valuesXsr)rE)rUFrE)rUrE)r)FF)NNFFTrE)NNFTrE)+r.loggingr+rPrrK collectionsr contextlibrtypingrrrrrr r r parserr r variablesr getLoggerr?r version_infoPathLikerCrBrrrGrTrDrerir4ryrrrrrrrs  (    R       4    *   $