o PaC@sddlZddlZddlZddlmZddlmZmZddlZddl m Z ddl m Z m Z ddlmZmZddlmZeeZdeefd d Zd ed edeeeffd dZd ede fddZdS)N)Path)OptionalTuple) constants) EXIT_CODE_OKExitCode)hazardstars) pipx_wrapreturncCsd}tt}tt}z||}Wn ty"d}Ynwd}|rI|ddttjddf}|D] }| rH|j}|Sq<|S)zmReturns None if pipx is not installed using `pip --user` Otherwise returns parent dir of pipx binary NFTbinpipxScriptszpipx.exe) r__file__resolvesite getuserbase relative_to ValueErrorgetusersitepackagesparentexists) pipx_bin_path script_path userbase_path_pip_user_installed test_paths test_pathr;/usr/lib/python3/dist-packages/pipx/commands/ensure_path.pyget_pipx_user_bin_paths(   r!locationforcecCst|}d}t|}t|}|s|s3|s3t|dttd|dddd}t|}||fS|sG|rGttd|d dd||fStt|d dd||fS) zhEnsure location is in user's PATH or add it to PATH. Returns True if location was added to PATH Fr zSuccess! Added z" to the PATH environment variable.z )subsequent_indentT z has been been added to PATH, but you need to open a new terminal or re-login for this PATH change to take effect. z is already in PATH.)struserpathneed_shell_restartin_current_pathappendprintr )r"r# location_str path_addedr(r)rrr ensure_path6s:      r.cCsttjg}t}|dur||d}d}|D]}t||d\}}||O}||O}qt|r9ttddn|sMtj t tdt dd|rWttddtd ttS) zReturns pipx exit code.NF)r#z Consider adding shell completions for pipx. Run 'pipx completions' for instructions.  r%z All pipx binary directories have been added to PATH. If you are sure you want to proceed, try again with the '--force' flag. z You will need to open a new terminal or re-login for the PATH changes to take effect. zOtherwise pipx is ready to go! )setr LOCAL_BIN_DIRr!addr.r+r sysstdoutflushloggerwarningrr r)r# bin_pathspipx_user_bin_pathr-r(bin_pathpath_added_currentneed_shell_restart_currentrrr ensure_pipx_paths\sT       r=)loggingrr3pathlibrtypingrrr'r rpipx.constantsrr pipx.emojisrr pipx.utilr getLogger__name__r6r!boolr.r=rrrr s    %&