B id @sddlZddlZddlZddlZddlmZddlmZddlm Z m Z m Z m Z m Z y ddlZWn*ek rejdedYnXddlmZmZmZddlmZeejd d ejed ejd d ddejdddedddgddejdddej ddej!edej"ej#e e e ddddZ$eej%e e e&ddd Z'e$(ej"ejd!d"ed"d#d$d%gd&dej#e)dd'd(d)Z*e$(ej"ej+d*d d+ej+d,d d+ej#e e dd-d.d/Z,e$(ej"ej+d*d d+ej#e dd0d1d2Z-e$(ej"ej+d*d d+ej#e dd0d3d4Z.e$j(d5d id6ej"ejd7d d8d9ej+d:d;ej/d<ej#e)e e&dd=d>d?Z0e e&e e&e&fe1d@dAdBZ2dS)CN)contextmanager)Popen)AnyDictIOIteratorListzkIt seems python-dotenv is not installed with cli option. Run pip install "python-dotenv[cli]" to fix this.) dotenv_valuesset_key unset_key) __version__z-fz--filez.envT)Z file_okayzNLocation of the .env file, defaults to .env file in current working directory.)defaulttypehelpz-qz--quotealwaysneverautozbWhether 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.)version)ctxfilequoteexportreturncCs|||d|_dS)zAThis script is used to set, get or unset values from a .env file.)QUOTEEXPORTFILEN)obj)rrrrr./tmp/pip-unpacked-wheel-mbm0dzz5/dotenv/cli.pyclisr )pathrc csbyt| }|VWdQRXWn>tk r\}z td|tjdtdWdd}~XYnXdS)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)r!streamexcrrr stream_file%s  r+z--formatsimplejsonshellrzlThe format in which to display the list. Default format is simple, which displays name=value without quotes.)rformatrc Cs|jd}t|}t|d}WdQRX|dkrHttj|dddn\|dkrTd nd }xJt|D]>}||}|dk rb|d krt |}t||d |qbWdS) z!Display all the stored key/value.r)r)Nr-r"T)indent sort_keysrzexport )rr.=) rr+r clickechor-dumpssortedshlexr)rr/rr)valuesprefixkvrrrlist5s   r=key)requiredvalue)rr>r@rcCsZ|jd}|jd}|jd}t|||||\}}}|rNt|d|ntddS)zStore the given key/value.rrrr3r N)rr r4r5r()rr>r@rrrsuccessrrrsetNs   rB)rr>rc CsN|jd}t|}t|d}WdQRX||}|rBt|ntddS)z%Retrieve the value for the given key.r)r)Nr )rr+r getr4r5r()rr>rr)r9Z stored_valuerrrrC^s    rCcCsF|jd}|jd}t|||\}}|r:td|ntddS)zRemoves the given key.rrzSuccessfully removed r N)rr r4r5r()rr>rrrArrrunsetos   rDZignore_unknown_options)Zcontext_settingsz--override/--no-overridezKOverride variables from the environment file with those from the .env file.)rr commandline)nargsr)roverriderErcsr|jd}tj|s,tjd|d|dfddt|D}|s\tdt dt ||}t |d S) z/Run command with environment variables present.rzInvalid value for '-f' "z" does not exist.)rcs,i|]$\}}|dk rs"|tjkr||qS)N)osenviron).0r;r<)rHrr szrun..zNo command given.r N) rrIr!isfiler4Z BadParameterr itemsr5r( run_command)rrHrErZdotenv_as_dictretr)rHrrun}s      rQ)commandenvrcCs8tj}||t|ddd|d}|\}}|jS)awRun command in sub process. Runs the command in a sub process with 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 ------- int The return code of the command TrF)universal_newlinesbufsizer.rS)rIrJcopyupdater communicate returncode)rRrSZcmd_envp_rrrrOs   rO)3r-rIr8r& contextlibr subprocessrtypingrrrrrr4 ImportErrorr'writer(mainr r r rr groupoptionr!joingetcwdPathZChoiceZBOOLZversion_optionZ pass_contextContextr PathLikestrr+rRboolr=argumentrBrCrDZ UNPROCESSEDrQintrOrrrrsp         $      "