o T۷i@sddlZddlZddlZddlZddlmZddlmZmZm Z m Z m Z m Z ej dkr1ddlmZzddlZWneyKejdedYnwddlmZmZmZdd lmZd e efd d Zeejd deejddddejddde gdddejdddej!ddej"edej#dej$deded ed df d!d"Z%ed#ej&d e eefd$d%Z'e%j(d&d'ej#ejd(d)d*e gd+d,ddej$d)ed dfd-d.Z)e%j(d/d'ej#ej*d0dd1ej*d2dd1dej$d0ed2ed dfd3d4Z+e%(ej#ej*d0dd1dej$d0ed dfd5d6Z,e%(ej#ej*d0dd1dej$d0ed dfd7d8Z-e%j(dddd9d:ej#ejd;ddd?ej.d@dej$dAe/d>e0edBfd dfdCdDZ1dEe edFe eefd dfdGdHZ2dS)IN)contextmanager)IOAnyDictIteratorListOptionalwin32)PopenzkIt seems python-dotenv is not installed with cli option. Run pip install "python-dotenv[cli]" to fix this.) dotenv_valuesset_key unset_key) __version__returncCs4zt}Wn tyYdSwtj|d}|S)zZ Return a path for the ${pwd}/.env file. If pwd does not exist, return None. Nz.env)osgetcwdFileNotFoundErrorpathjoin)cwdrr@/home/ubuntu/vllm_env/lib/python3.10/site-packages/dotenv/cli.py enumerate_envs  rz-fz--fileT) file_okayzNLocation of the .env file, defaults to .env file in current working directory.)defaulttypehelpz-qz--quotealways)rneverautozbWhether to quote or not the variable values. Default mode is always. This does not affect parsing.z-ez--exportFz;Whether to write the dot file as an executable bash script.)versionctxfilequoteexportcCs|||d|_dS)zAThis script is used to set, get or unset values from a .env file.)QUOTEEXPORTFILEN)obj)r"r#r$r%rrrcli&sr*rc cszt| }|VWdWdS1swYWdSty?}ztd|tjdtdWYd}~dSd}~ww)z~ Open a file and yield the corresponding (decoded) stream. Exits with error code 2 if the file cannot be opened. NzError opening env file: )r#)openOSErrorprintsysstderrexit)rstreamexcrrr stream_fileCs &r4list)namez--format output_formatsimple)r8jsonshellr%zlThe format in which to display the list. Default format is simple, which displays name=value without quotes.cCs|jd}t| }t|d}Wdn1swY|dkr/ttj|ddddS|dkr5d nd }t|D]}||}|durZ|d vrNt |}t||d |q;dS) z!Display all the stored key/value.r(r2Nr9r+T)indent sort_keysr%zexport )r%r:=) r)r4r clickechor9dumpssortedshlexr$)r"r7r#r2valuesprefixkvrrr list_valuesSs    rIsetkey)requiredvaluecCs^|jd}|jd}|jd}t|||||\}}}|r(t|d|dStddS)z Store the given key/value. This doesn't follow symlinks, to avoid accidentally modifying a file at a potentially untrusted path. r(r&r'r?r N)r)r r@rAr/r1)r"rKrMr#r$r%successrrr set_valueps  rOcCsf|jd}t| }t|d}Wdn1swY||}|r,t|dStddS)z%Retrieve the value for the given key.r(r;Nr )r)r4r getr@rAr/r1)r"rKr#r2rE stored_valuerrrrPs    rPcCsJ|jd}|jd}t|||\}}|rtd|dStddS)z Removes the given key. This doesn't follow symlinks, to avoid accidentally modifying a file at a potentially untrusted path. r(r&zSuccessfully removed r N)r)rr@rAr/r1)r"rKr#r$rNrrrunsets rR)allow_extra_argsallow_interspersed_argsignore_unknown_options)context_settingsz--override/--no-overridezKOverride variables from the environment file with those from the .env file.)rr commandline)nargsroverride.csv|jd}tj|stjd|d|dfddt|D}|s/tdt dt g||j |d S) z/Run command with environment variables present.r(zInvalid value for '-f' "z" does not exist.)r"cs,i|]\}}|durs|tjvr||qS)N)renviron).0rGrHrZrr s zrun..zNo command given.r N) r)rrisfiler@ BadParameterr itemsrAr/r1 run_commandargs)r"rZrWr#dotenv_as_dictrr]rruns       recommandenvcCsdtj}||tjdkr&t|ddd|d}|\}}t|j dStj |d||ddS)aReplace the current process with the specified command. Replaces the current process with the specified command and the variables from `env` added in the current environment variables. Parameters ---------- command: List[str] The command and it's parameters env: Dict The additional environment variables Returns ------- None This function does not return any value. It replaces the current process with the new one. r TrF)universal_newlinesbufsizer:rg)rcrgN) rr[copyupdater/platformr communicater1 returncodeexecvpe)rfrgcmd_envp_rrrrbs    rb)3r9rrDr/ contextlibrtypingrrrrrrrl subprocessr r@ ImportErrorr0writer1mainr r rr!rstrrgroupoptionPathChoiceBOOLversion_option pass_contextContextr*PathLiker4rfrIargumentrOrPrR UNPROCESSEDbooltuplererbrrrrs          ,     $  ,&