a q+@svddlmZmZmZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZddlmZmZmZmZddlmZmZeeZerddlmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$e j%d krej&Z'ne"Z'e j%d kreZ(nee"Z(e)d ej*Z+d d Z,GdddZ-ddZ.eddZ/d&ddZ0d'ddZ1ddZ2ddZ3d(dd Z4d)d"d#Z5d*d$d%Z6dS)+)absolute_importprint_functionunicode_literalsN) OrderedDict)contextmanager)IS_TYPE_CHECKINGPY2StringIOto_env)Binding parse_stream) DictIterableIteratorMatchOptionalPatternUnionTextIOTuple))rrzh \$\{ (?P[^\}:]*) (?::- (?P[^\}]*) )? \} ccs*|D] }|jrtd|jj|VqdS)Nz;Python-dotenv could not parse statement starting at line %s)errorloggerwarningoriginalline)Zmappingsmappingr //usr/lib/python3.9/site-packages/dotenv/main.pywith_warn_for_invalid_lines.sr"c@sDeZdZdddZeddZdd Zd d Zdd d ZddZ dS)DotEnvFNTcCs"||_d|_||_||_||_dSN) dotenv_path_dictverboseencoding interpolate)selfr%r'r(r)r r r!__init__;s zDotEnv.__init__ccst|jtr|jVnjtj|jr^tj|j|jd}|VWdq1sR0Yn"|j rvt d|jprdtdVdS)N)r(z3Python-dotenv could not find configuration file %s..env) isinstancer%r ospathisfileioopenr(r'rinfo)r*streamr r r! _get_streamCs  &zDotEnv._get_streamcCs6|jr |jS|jr t|}n t|}||_|S)zReturn dotenv as dict)r&r)resolve_nested_variablesparser)r*valuesr r r!dictPs z DotEnv.dictccsV|:}tt|D]}|jdur|j|jfVqWdn1sH0YdSr$)r6r"r keyvalue)r*r5rr r r!r8^s  z DotEnv.parsecCsD|D]2\}}|tjvr$|s$q |dur t|tjt|<q dS)zH Load the current dotenv as system environemt variable. NT)r:itemsr/environr )r*overridekvr r r!set_as_environment_variableses z#DotEnv.set_as_environment_variablescCs2|}||vr||S|jr.td||jdS)z zKey %s not found in %s.N)r:r'rrr%)r*r;datar r r!getrs z DotEnv.get)FNT)F) __name__ __module__ __qualname__r+rr6r:r8rBrDr r r r!r#9s   r#cCst|dd|S)zl Gets the value of a given key from the given .env If the .env path given doesn't exist, fails T)r')r#rD)r%Z key_to_getr r r!get_keysrHc csztj|sDt|d}|dWdn1s:0YtjdddD}t|}||fVWdn1s~0YWdn1s0YWn.tytj|j rt |j Yn0t |j |dS)Nzw+r-F)modedelete) r/r0r1r2r3writetempfileZNamedTemporaryFile BaseExceptionnameunlinkshutilmove)r0sourcedestr r r!rewrites ( J  rTalwaysFc Cs|dd}d|vrd}|dkr8d|dd}n|}|rNd||}n d||}t|`\}}d } tt|D],} | j|kr||d } qx|| jj qx| s||Wd n1s0Yd ||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 '" rUz"{}"z\"z export {}={} z{}={} FTN) stripformatreplacerTr"r r;rKrstring) r%Z key_to_setZ value_to_set quote_modeZexportZ value_outZline_outrRrSZreplacedrr r r!set_keys&   (r^cCstj|s td|d|fSd}t|D\}}tt|D]"}|j|krRd}q>| |j j q>Wdn1sv0Y|std||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.NFTz/Key %s not removed from %s - key doesn't exist.) r/r0existsrrrTr"r r;rKrr\)r%Z key_to_unsetr]ZremovedrRrSrr r r! unset_keys   .r`csJfddfdd}i|D]$\}}|dur._replacementcs|}|d|ddS)zh From a match object gets the variable name and returns the correct replacement rNrb)rNrb) groupdict)matchmatches)rer r!_re_sub_callbacksz2resolve_nested_variables.._re_sub_callback)__posix_variablesub)r9rir@rAr )rerdr!r7s    r7ccsttj|stdtj|r,tj|}d}tj|}||krp|Vtjtj|tjj}||}}q Decide whether this is running in a REPL or IPython notebook __main__N__file__)fromlist) __import__hasattr)mainr r r!_is_interactivesz$find_dotenv.._is_interactivefrozenFz.py.rrNzFile not foundr-)getattrsysr/getcwd _getframer rsendswithrsplitf_code co_filenamef_backr0rmrnrqror1rl) filenameZraise_error_if_not_foundZusecwdrxr0frameZ current_fileZframe_filenamermZ check_pathr r r! find_dotenvs&     rTcKs,|p |p t}t|f||d|j|dS)aParse a .env file and then load all the variables found as environment variables. - *dotenv_path*: absolute or relative path to .env file. - *stream*: `StringIO` object with .env content. - *verbose*: whether to output the warnings related to missing .env file etc. Defaults to `False`. - *override*: where to override the system environment variables with the variables in `.env` file. Defaults to `False`. r'r))r?)rr#rB)r%r5r'r?r)kwargsfr r r! load_dotenv4s rcKs(|p |p t}t|f||d|S)Nr)rr#r:)r%r5r'r)rrr r r! dotenv_valuesBsr)rUF)rU)r,FF)NNFFT)NNFT)7Z __future__rrrr2Zloggingr/rerPr|rL collectionsr contextlibrcompatrr r r parserr r Z getLoggerrErtypingrrrrrrrrrr version_infoPathLikeZ _PathLikeZ _StringIOcompileVERBOSErjr"r#rHrTr^r`r7rqrrrr r r r!sH   0    H   $  ,