o T۷i[9@sddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZmZmZmZmZmZddlmZmZddlmZeedfZeeZd efd d Zd eed eefd dZ GdddZ! ddedededed edeeded eeeeeffd!d"Z$   d?ded#ededeeded eeeeff d$d%Z%d&eeeeefd'ed eeeeffd(d)Z&ded eefd*d+Z' ,  d@d-ed.ed/ed efd0d1Z(     2 dAdeed3ee ed4ed'ed5edeed efd6d7Z)    2 dBdeed3ee ed4ed5edeed e eeeff d8d9Z*ded efd:d;Z+dS)CN) OrderedDict)contextmanager)IODictIterableIteratorMappingOptionalTupleUnion)Binding parse_stream)parse_variableszos.PathLike[str]returncCs$dtjvrdStjd}|dvS)z8 Determine if dotenv loading has been disabled. PYTHON_DOTENV_DISABLEDF>1tyyestrue)osenvironcasefold)valuerA/home/ubuntu/vllm_env/lib/python3.10/site-packages/dotenv/main.py_load_dotenv_disableds rmappingsccs,|D]}|jrtd|jj|VqdS)Nz;python-dotenv could not parse statement starting at line %s)errorloggerwarningoriginalline)rmappingrrrwith_warn_for_invalid_lines sr%c@seZdZ     ddeedeeededeeded ed dfd d Ze d e eefd dZ d e eeeffddZ d e eeeeffddZd efddZded eefddZdS)DotEnvNFT dotenv_pathstreamverboseencoding interpolateoverridercCs.||_||_d|_||_||_||_||_dSN)r'r(_dictr)r*r+r,)selfr'r(r)r*r+r,rrr__init__+s  zDotEnv.__init__ccs|jr't|jr't|j|jd }|VWddS1s wYdS|jdur2|jVdS|jr>td|jpcCs2|}||vr ||S|jrtd||jdS) zKey %s not found in %s.N)r=r)r r!r')r/r>datarrrgetps z DotEnv.get)NFNTT)__name__ __module__ __qualname__r StrPathrstrboolr0rrr9rr=r r:rBrErrrrr&*s6  r&utf-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*)r&rE)r'rMr*rrrget_key}s rNFpathfollow_symlinksc cs|r tj|}z*t||d}zt|}t|jr"t|jnd}Wn t y1| wWnt yCt d}d}Ynwtjd|ddtjtj|d>}t|j}d}z| ||fVWdn1srwYWnt y} z| }WYd} ~ nd} ~ wwWdn1swY|durz|durt||t||WdSt y|jddw|jdd|d) Nr1r3wFz.tmp_)moder*deleteprefixdirT) missing_ok)rrOrealpathr5lstatstatS_ISREGst_modeS_IMODE BaseExceptioncloseFileNotFoundErrorr7r8tempfileNamedTemporaryFiledirnameabspathpathlibPathnamechmodreplaceunlink) rOr*rPsource path_stat original_modedest dest_pathrerrrrrrewritesh             rpalways 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 } qN| |j j |j j d  } qN| s{| rv| d | | Wdn1swYd ||fS)aS Adds or Updates a key/value to the given .env The target .env file is created if it doesn't exist. This function doesn't follow symlinks by default, to avoid accidentally modifying a file at a potentially untrusted path. If you don't need this protection and need symlinks to be followed, use `follow_symlinks`. )rqautoneverzUnknown quote_mode: rqrvz'{}''z\'zexport = r*rPFTN) ValueErrorisalnumformatrhrpr%rr>writer"stringendswith)r'rrrsrtrur*rPquote value_outline_outrjrmreplacedmissing_newliner$rrrset_keys:     r key_to_unsetc Cstj|std|d|fSd}t|||d"\}}tt|D]}|j|kr,d}q"| |j j q"Wdn1s>wY|sPtd||d|fS||fS)a 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. This function doesn't follow symlinks by default, to avoid accidentally modifying a file at a potentially untrusted path. If you don't need this protection and need symlinks to be followed, use `follow_symlinks`. z(Can't delete from %s - it doesn't exist.NFr{Tz/Key %s not removed from %s - key doesn't exist.) rrOexistsr r!rpr%rr>rr"r) r'rrtr*rPremovedrjrmr$rrr unset_keys&    rvaluesr,csi}|D]:\}}|durd}n+t|}i|r#tj|n |tjdfdd|D}|||<q|S)Nr3c3s|]}|VqdSr-)resolve).0atomenvrr 3sz$resolve_variables..)rupdaterrjoin)rr, new_valuesrfrresultatomsrrrr;!s      r;ccstj|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) rrOrIOErrorisfilerbrcrpardir)rOlast_dir current_dir parent_dirrrr _walk_to_root:s     rr2filenameraise_error_if_not_foundusecwdc Csdd}dd}|s|s|sttddrt}n:t}t}|jj|ks/tj |jjsG|j dus6J|j }|jj|ks/tj |jjr/|jj}tj tj |}t |D]} tj| |} t| rj| SqY|rqtdd S) z Search in increasingly higher folders for the given file Returns path to the file if found, or an empty string otherwise cSsPttds ttdr dSz tddddgd}Wn ty!YdSwt|d S) z._is_interactivecSs tduSr-)rgettracerrrr _is_debuggeras z!find_dotenv.._is_debuggerfrozenFNzFile not foundr3)getattrrrgetcwd _getframerf_code co_filenamerOrf_backrbrcrrr4r) rrrrrrOframe current_fileframe_filenamerb check_pathrrr find_dotenvLs0    rTr(r)r+cCsFtr tddS|dur|durt}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 with it's default parameters. If you need to change the default parameters of `find_dotenv()`, you can explicitly call `find_dotenv()` and pass the result to this function as `dotenv_path`. If the environment variable `PYTHON_DOTENV_DISABLED` is set to a truthy value, .env loading is disabled. zSpython-dotenv: .env loading disabled by PYTHON_DOTENV_DISABLED environment variableFNr'r(r)r+r,r*)rr debugrr&rB)r'r(r)r,r+r*dotenvrrr load_dotenvs rcCs.|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)rr&r=)r'r(r)r+r*rrr dotenv_valuessrc CsDtj|rdSzt|}Wn ttfyYdSwt|jS)zN Return True if `path` exists and is either a regular file or a FIFO. TF)rrOrrYr_OSErrorS_ISFIFOr[)rOstrrrr4s  r4)rL)F)rqFrLF)rqrLF)r2FF)NNFFTrL)NNFTrL),r7loggingrrdrYrr` collectionsr contextlibrtypingrrrrrr r r parserr r variablesrrJrI getLoggerrFr rKrr%r&rNrprrr;rrrrr4rrrrs  (     V   : : )  4  2  $