B id. @s(ddlZddlZddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z m Z mZmZmZmZmZddlmZmZddlmZeedfZeeZeeeed d d ZGd d d Zd.eeeeeedddZe eeeeee ee efdddZ d/eeeee!eeeee!eefdddZ"d0eeeeeeee!efdddZ#e eeeefe!eeeefdddZ$eeed d!d"Z%d1ee!e!ed$d%d&Z&d2eeee ee!e!e!eee!d(d)d*Z'd3eeee ee!e!eee eeefd+d,d-Z(dS)4N) OrderedDict)contextmanager)IODictIterableIteratorMappingOptionalTupleUnion)Binding parse_stream)parse_variableszos.PathLike[str])mappingsreturnccs.x(|D] }|jr td|jj|VqWdS)Nz;Python-dotenv could not parse statement starting at line %s)errorloggerwarningoriginalline)rmappingr//tmp/pip-unpacked-wheel-mbm0dzz5/dotenv/main.pywith_warn_for_invalid_liness   rc @seZdZdeeeeeeeeeeddddZe e eeddd Z e eeefdd d Z e eeeefdd d ZedddZeeedddZdS)DotEnvNFT) dotenv_pathstreamverboseencoding interpolateoverridercCs.||_||_d|_||_||_||_||_dS)N)rr_dictrrr r!)selfrrrrr r!rrr__init__#s zDotEnv.__init__)rc cst|jr8tj|jr8t|j|jd }|VWdQRXn8|jdk rL|jVn$|jrdt d|jp`dt dVdS)N)rz3Python-dotenv could not find configuration file %s.z.env) rospathisfileopenrrrrinfoioStringIO)r#rrrr _get_stream4s   zDotEnv._get_streamcCs@|jr |jS|}|jr0tt||jd|_n t||_|jS)zReturn dotenv as dict)r!)r"parser rresolve_variablesr!)r#Z raw_valuesrrrdictCs z DotEnv.dictc csF|4}x,tt|D]}|jdk r|j|jfVqWWdQRXdS)N)r-rrkeyvalue)r#rrrrrr.Qs  z DotEnv.parsecCsN|s dSx<|D],\}}|tjkr4|js4q|dk r|tj|<qWdS)zI Load the current dotenv as system environment variable. FNT)r0itemsr&environr!)r#kvrrrset_as_environment_variablesWsz#DotEnv.set_as_environment_variables)r1rcCs2|}||kr||S|jr.td||jdS)z zKey %s not found in %s.N)r0rrrr)r#r1datarrrgetfs z DotEnv.get)NFNTT)__name__ __module__ __qualname__r StrPathrstrboolr$rrr-rr0r r.r7r9rrrrr"s rutf-8)r key_to_getrrcCst|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)rr9)rrArrrrget_keyts rB)r'rrc cstj|s0t|d|d}|dWdQRXtjd|ddP}y&t||d}||fVWdQRXWn"tk rt|j YnXWdQRXt |j |dS)Nw)moderr%F)rDrdelete)r) r&r'r(r)writetempfileNamedTemporaryFile BaseExceptionunlinknameshutilmove)r'rsourcedestrrrrewrites  rPalwaysF)r key_to_set value_to_set quote_modeexportrrc Cs |dkrtd||dkp.|dko.| }|rHd|dd}n|}|rdd|d |d }n|d |d }t||d |\} } d } d } xLtt| D]<} | j|kr| |d } q| | j j | j j d  } qW| s| r| d | |WdQRXd ||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 )rQautoneverzUnknown quote_mode: rQrVz'{}''z\'zexport = )rFTN) ValueErrorisalnumformatreplacerPrrr1rFrstringendswith)rrRrSrTrUrquoteZ value_outZline_outrNrOZreplacedZmissing_newlinerrrrset_keys0   rb)r key_to_unsetrTrrc Cstj|s td|d|fSd}t||d>\}}x2tt|D]"}|j|krXd}qD| |j j qDWWdQRX|std||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.NF)rTz/Key %s not removed from %s - key doesn't exist.) r&r'existsrrrPrrr1rFrr_)rrcrTrremovedrNrOrrrr unset_keys   rf)valuesr!rcsi}x||D]t\}}|dkr d}nVt|}i|rHtj|n|tjdfdd|D}|||<q W|S)Nr%c3s|]}|VqdS)N)resolve).0Zatom)envrr sz$resolve_variables..)rupdater&r4join)rgr!Z new_valuesrKr2resultZatomsr)rjrr/s     r/)r'rccsxtj|stdtj|r,tj|}d}tj|}x6||krr|Vtjtj|tjj}||}}q>WdS)zL Yield directories starting from the given directory up to the root zStarting path not foundN) r&r'rdIOErrorr(dirnameabspathrmpardir)r'Zlast_dirZ current_dir parent_dirrrr _walk_to_roots     rt.env)filenameraise_error_if_not_foundusecwdrc Csdd}|s|sttddr(t}nNt}t}x$|jj|krX|jdk sPt |j}q6W|jj}tj tj |}x.t |D]"}tj ||} tj | r| SqW|rtddS)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_interactivesz$find_dotenv.._is_interactivefrozenFNzFile not foundr%)getattrsysr&getcwd _getframerzf_code co_filenamef_backAssertionErrorr'rprqrtrmr(ro) rvrwrxrr'frameZ current_fileZframe_filenamerp check_pathrrr find_dotenvs"    rT)rrrr!r rrcCs2|dkr|dkrt}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. N)rrrr r!r)rrr7)rrrr!r rZdotenvrrr load_dotenv7sr)rrrr rrcCs.|dkr|dkrt}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. NT)rrrr r!r)rrr0)rrrr rrrr dotenv_values]sr)r@)rQFr@)rQr@)ruFF)NNFFTr@)NNFTr@))r+loggingr&rLrrG collectionsr contextlibrtypingrrrrrr r r parserr r variablesrr>r= getLoggerr:rrrrBrPr?rbrfr/rtrrrrrrrs`  (    U  $-&