U d. @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*|D] }|jrtd|jj|VqdS)Nz;Python-dotenv could not parse statement starting at line %s)errorloggerwarningoriginalline)rmappingr//tmp/pip-unpacked-wheel-j7y5vhg2/dotenv/main.pywith_warn_for_invalid_linessrc @seZdZdeeeeeeeeeeddddZe e eeddd Z e eeefdd d Z e eeeefdd d ZedddZeeedddZdS)DotEnvNFT) dotenv_pathstreamverboseencoding interpolateoverridercCs.||_||_d|_||_||_||_||_dSN)rr_dictrrr r!)selfrrrrr r!rrr__init__#s zDotEnv.__init__)rc cst|jr8tj|jr8t|j|jd }|VW5QRXn8|jdk rL|jVn$|jrdt d|jp`dt dVdS)Nrz3Python-dotenv could not find configuration file %s..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 csB|0}tt|D]}|jdk r|j|jfVqW5QRXdSr")r0rrkeyvalue)r$rrrrrr1Qs  z DotEnv.parsecCsJ|s dS|D],\}}|tjkr2|js2q|dk r|tj|<qdS)zI Load the current dotenv as system environment variable. FNT)r3itemsr)environr!)r$kvrrrset_as_environment_variablesWs z#DotEnv.set_as_environment_variables)r4rcCs2|}||kr||S|jr.td||jdS)z zKey %s not found in %s.N)r3rrrr)r$r4datarrrgetfs z DotEnv.get)NFNTT)__name__ __module__ __qualname__r StrPathrstrboolr%rrr0rr3r r1r:r<rrrrr"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)rr<)rrDrrrrget_keyts rE)r*rrc cstj|s0t|d|d}|dW5QRXtjd|ddP}z&t||d}||fVW5QRXWn"tk rt|j YnXW5QRXt |j |dS)Nw)moderr(F)rGrdeleter&) r)r*r+r,writetempfileNamedTemporaryFile BaseExceptionunlinknameshutilmove)r*rsourcedestrrrrewrites  rSalwaysF)r key_to_set value_to_set quote_modeexportrrc Cs|dkrtd||dkp.|dko.| }|rHd|dd}n|}|rdd|d |d }n|d |d }t||d x\} } d } d } tt| D]<} | j|kr| |d } q| | j j | j j d  } q| s| r| d | |W5QRXd ||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 )rTautoneverzUnknown quote_mode: rTrYz'{}''z\'zexport = r&FT) ValueErrorisalnumformatreplacerSrrr4rIrstringendswith)rrUrVrWrXrquoteZ value_outZline_outrQrRZreplacedZmissing_newlinerrrrset_keys2   re)r key_to_unsetrWrrc Cstj|s td|d|fSd}t||d:\}}tt|D]"}|j|krVd}qB| |j j qBW5QRX|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.NFr&Tz/Key %s not removed from %s - key doesn't exist.) r)r*existsrrrSrrr4rIrrb)rrfrWrremovedrQrRrrrr unset_keys   ri)valuesr!rcsi}|D]t\}}|dkrd}nVt|}i|rFtj|n|tjdfdd|D}|||<q|S)Nr(c3s|]}|VqdSr")resolve).0Zatomenvrr sz$resolve_variables..)rupdater)r7join)rjr!Z new_valuesrNr5resultZatomsrrmrr2s      r2)r*rccsttj|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)mainrrr_is_interactivesz$find_dotenv.._is_interactivefrozenFNzFile not foundr()getattrsysr)getcwd _getframer}f_code co_filenamef_backAssertionErrorr*rtrurxrqr+rs) ryrzr{rr*frameZ current_fileZframe_filenamert 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. Nrrrr r!r)rrr:)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. NTr)rrr3)rrrr rrrr dotenv_values]sr)rC)rTFrC)rTrC)r'FF)NNFFTrC)NNFTrC))r.loggingr)rOrrJ collectionsr contextlibrtypingrrrrrr r r parserr r variablesrrAr@ getLoggerr=rrrrErSrBrerir2rxrrrrrrrs  (    U  4   *  '