
    .eE                     .   d dl Z d dlZd dlZd dlZd dlZd dlmc mZ d dl	m
Z
 d dlmZmZ d dlmZ d dlmZ d dlmZmZmZmZmZmZmZ d dlmZ  ee          Zeed         ed	         f         Zeed
         ed         f         Zd0defdZ  e d          Z! e d	          Z"de#fdZ$d1dZ%d2dej&        de#dee#         fdZ' G d d          Z( G d d          Z)d3de j*        de+fdZ,de j*        defdZ-dedee         defd Z.d!e j*        d"e j*        de+fd#Z/ ed$%          de+fd&            Z0d'ee#         de1fd(Z2d2d"ee j*                 dee#         fd)Z3d*edefd+Z4	 	 	 d4d!ee j*                 d"ee j*                 d,efd-Z5d5dej6        fd/Z7dS )6    N)suppress)	lru_cachewraps)	getLogger)Path)CallableIterableLiteralMappingOptionalSequenceUnion)	normalizetargetbaseusersystemwhichc                    | dk    }d}dt           j        v r2t          t                    5  ddlm} ddd           n# 1 swxY w Y   |rVt          j                            d          }|r|S |r|j	        S t          j                            dt           j
                  S t          j                            d          }|r|S |r|j        S t          j                            d	t           j                  S )
a!  
    The prefixes in menuinst need to be handled with care.

    Conda installations that require superuser permissions need elevation for
    the creation of shortcuts. Constructor will leave a sentinel file to signal
    this. If a file `.nonadmin` is present in the 'base' environment (or root of
    the installation directory if conda is not present), superuser access is not
    needed.

    In order to check for this file, menuinst needs to track 'base_prefix'. For
    a regular 'conda' process, this should be `conda.base.context.context.root_prefix'.
    However, constructor also relies on a pyinstaller-frozen conda installation,
    'conda-standalone'. In these cases, 'sys.prefix' is set to temporary location
    of the extracted contents of the executable -- that's NOT the base installation!

    For those reasons, we handle the default prefix with 'sys.prefix' (or 'sys.base_prefix')
    as a last resort. The logic is:

    - If 'MENUINST_PREFIX' (or 'MENUINST_BASE_PREFIX') is an env var with a set value, use that.
    - If are already using conda, we get the context object and use those values.
    - If CONDA_PREFIX (or 'CONDA_ROOT_PREFIX') are available, use those
    - Last resort: use sys.prefix and sys.base_prefix

    This helps us pass a lot of CLI arguments back and forth.
    r   Ncondar   )contextMENUINST_BASE_PREFIXCONDA_ROOT_PREFIXMENUINST_PREFIXCONDA_PREFIX)sysmodulesr   ImportErrorconda.base.contextr   osenvirongetroot_prefixbase_prefixtarget_prefixprefix)r   r   r   r&   s       .lib/python3.11/site-packages/menuinst/utils.py_default_prefixr(      s4   4 F?DG#+k"" 	3 	3222222	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3 	3  D 677 	M 	'&&z~~13?CCCZ^^-..F  %$$:>>.#*555s   >AAtextc                    t          d|                               dd                              d          } t          j        dd|                                                                           } t          j        dd|           S )z
    Convert to ASCII, and spaces and underscores to hyphens.
    Remove characters that aren't alphanumerics, or hyphens.
    Convert to lowercase. Also strip leading and trailing whitespace.
    NFKDasciiignorez[^\w\s-] z[_\-\s]+-)r   encodedecoderesubstriplower)r)   s    r'   slugifyr6   G   sp    : VT""))'8<<CCGLLD6+r4((..006688D6+sD)))    c                    d}d||z  z   }t          |           r| j        r| j                                        s
||z   | _        | D ]?}t          ||dz              |j        r|j                                        s
||z   |_        @|j        r|j                                        s	||_        dS dS |r)| j        r| j                                        s|| _        dS dS dS )zR
    adds whitespace to the tree, so that it results in a pretty printed tree
    z    
   N)lenr)   r4   indent_xml_treetail)elemlevelindentationbase_indentationes        r'   r<   r<   i   s    Kek11
4yy )y 	7	 1 1 	7(;6DI 	8 	8AAuqy)))6 8 8)K7v 	&QV\\^^ 	&%AFFF	& 	&  	)$) 	)49??+<+< 	)(DIII	) 	) 	) 	)r7   parenttagc                 B    t          j        | |          }|||_        |S )ze
    Add a child element of specified tag type to parent.
    The new child element is returned.
    )XMLTree
SubElementr)   )rC   rD   r)   r>   s       r'   add_xml_childrH   }   s(    
 fc**D	Kr7   c                       e Zd Zedee         fd            Zedee         fd            Zed
dedefd            Zd	S )WinLexargsc           	          t          |          dk    rd|d                                         v sd|d                                         v r|d                                         dk    s|d                                         dk    rqt          d |dd          D                       rP                     |d         d	          |d         d
d                     fd|dd          D                       z  g}n fd|D             }|S )N   zCMD.EXEr   z	%COMSPEC%r:   z/Kz/Cc              3      K   | ]}d |v V  	dS ) N ).0args     r'   	<genexpr>z$WinLex.quote_args.<locals>.<genexpr>   s&      333C3J333333r7   ""%s"rO   c              3   D   K   | ]}                     |d           V  dS )rT   N)
ensure_padrQ   rR   clss     r'   rS   z$WinLex.quote_args.<locals>.<genexpr>   s1      "P"P3>>#s#;#;"P"P"P"P"P"Pr7   c                 :    g | ]}                     |          S rP   quote_stringrX   s     r'   
<listcomp>z%WinLex.quote_args.<locals>.<listcomp>   s'    :::cC$$S)):::r7   )r;   upperanyrW   joinrY   rK   s   ` r'   
quote_argszWinLex.quote_args   s    IIMMd1gmmoo--Q1O1OaD((DGMMOOt,C,C33$qrr(33333 -D tAw,,Q#(("P"P"P"PtABBx"P"P"PPPQDD ;:::T:::Dr7   sc                 b    |                     d          }|d         dv r|S d|v sd|v rd|z  S |S )z/
        quotes a string if necessary.
        rT   r   )r/   rO   rO   /rU   )r4   )rY   rc   s     r'   r\   zWinLex.quote_string   sG     GGCLLQ4:H!88saxxA:r7   _namepadc                 R    |r|d         |d         cxk    r|k    rn n|S |||S )zW

        Examples:
            >>> ensure_pad('conda')
            '_conda_'

        r   rP   )rY   rg   rh   s      r'   rW   zWinLex.ensure_pad   sM      	/tAw$r(1111c11111K"sDD##..r7   N)rf   )	__name__
__module____qualname__classmethodr   strrb   r\   rW   rP   r7   r'   rJ   rJ      s        hsm    [& Xc]    [ / /c / / / / [/ / /r7   rJ   c                   f    e Zd Zedee         dee         fd            Zededefd            ZdS )UnixLexrK   returnc                        fd|D             S )Nc                 :    g | ]}                     |          S rP   r[   )rQ   arY   s     r'   r]   z&UnixLex.quote_args.<locals>.<listcomp>   s'    222  ##222r7   rP   ra   s   ` r'   rb   zUnixLex.quote_args   s    2222T2222r7   rc   c                 |    t          j        |          }|                    d          rd|vrd|dd          d}|S )N'rT   r:   rj   )shlexquote
startswith)rY   rc   quoteds      r'   r\   zUnixLex.quote_string   sL    QS!! 	)c&7&7("(((Fr7   N)rk   rl   rm   rn   r   ro   rb   r\   rP   r7   r'   rq   rq      sv        3hsm 3 3 3 3 [3 S S    [  r7   rq   Fpath
missing_okc                 h    	 t          j        |            d S # t          $ r}|s|Y d }~d S d }~ww xY wN)r    unlinkFileNotFoundError)r|   r}   excs      r'   r   r      sb    
	$    	I	 	 	 	 	 	s    
1,1rr   c                 D    t          t                    j        }|dz  | z  S )Ndata)r   __file__rC   )r|   heres     r'   	data_pathr      s    >> D&=4r7   mappingupdating_mappingsc                    |                                  }|D ]o}|                                D ]X\  }}||v rJt          ||         t                    r/t          |t                    rt	          ||         |          ||<   S|||<   Yp|S r   )copyitems
isinstancedictdeep_update)r   r   updated_mappingupdating_mappingkvs         r'   r   r      s    , llnnO- 	' 	'$**,, 	' 	'DAq_$$q1488 %q$'' % &11CQ%G%G""%&""	' r7   r%   r$   c                 p   t                      rdS t          | d                                          rdS 	 t          | d                                           dS # t          $ r'}t
                              d| |           Y d}~nd}~ww xY w|| k    rdS t          |d                                          rdS t          j        dk    rdS t          j        dk    rZ	 t          |d                                           dS # t          $ r(}t
                              d| |           Y d}~dS d}~ww xY wdS )	zE
    Checks if the current installation needs admin permissions.
    Fz	.nonadminz%Attempt to write %s/.nonadmin failed.exc_infoNTntposix)	user_is_adminr   existstouch	Exceptionloggerdebugr    rg   )r%   r$   r   s      r'   needs_adminr      sy     uM;''..00  u[]K((..000u [ [ [<mVYZZZZZZZZ[ m##t
 K%%,,.. u	w$t	w'	k**00222
 5	  	 	 	LL@-Z]L^^^44444	 s/   "A 
B$BB"D 
D3D..D3r:   )maxsizec                      t           j        dk    rddlm}  t	           |                       S t           j        dk    rt          j                    dk    S t          dt           j                   )Nr   r:   isUserAdminr   r   Unsupported operating system: )r    rg   platforms.win_utils.win_elevater   boolgetuidRuntimeErrorr   s    r'   r   r   %  sj    	w$@@@@@@KKMM"""	G		y{{aEBGEEFFFr7   argvc                     t           j        dk    rddlm}  ||           S t           j        dk    rt	          j        dg|           S t          dt           j                   )zE
    Rerun this command in a new process with admin permissions.
    r   r:   )
runAsAdminr   sudor   )r    rg   r   r   
subprocesscallr   )r   r   s     r'   run_as_adminr   2  sm     
w$??????z$	G		///EBGEEFFFr7   c                    t          | pt                    } t          t          dd          rPt          j        dk    r| dz  }n| dz  dz  }|                                rt          |          fS t          j        dfS t          j        fS )NfrozenFr   z
python.exebinpython)	r   DEFAULT_BASE_PREFIXgetattrr   r    rg   is_filero   
executable)r$   base_prefix_pythons     r'   python_executabler   @  s    {9&9::K sHe$$ *7d??!,|!;!,u!4x!? %%'' 	.*++-- ))Nr7   funcc                 v     t                     ddddt          j        dt          j        f fd            }|S )a  
    Multiplatform decorator to run a function as a superuser, if needed.

    This depends on the presence of a `.nonadmin` file in the installation root.
    This is usually planted by the `constructor` installer if the installation
    process didn't need superuser permissions.

    In the absence of this file, we assume that we will need superuser
    permissions, so we try to run the decorated function as a superuser.
    If that fails (the user rejects the request or doesn't have permissions
    to accept it), we'll try to run it as a normal user.

    NOTE: Only functions that return None should be decorated. The function
    will run in a separate process, so we won't be able to capture the return
    value anyway.
    N)r%   r$   r%   r$   c                 ^   |                     dd            | pt          } |pt          }t          | |          rNt          j                            d          dk    r*	 	j        dk    rd	j         dt           d	j         d}nd		j         d
	j         d}d	                    d t          j        
                                D                       }g t                      dd| d| 	j         d|d| d|d|d}t                              d|           t          |          }t          j                             dd            |dk    rd S nM# t          $ r&}t                              d|           Y d }~n"d }~ww xY wt#                      r 	|| |dd|S  	|| |dd|S )N_mode_MENUINST_RECURSING1__main__zimport runpy;z = runpy.run_path('z')['z'];zfrom z import ;c                 P    g | ]#\  }}|                     d           d| d| d$S ))CONDA_CONSTRUCTOR_	MENUINST_zos.environ.setdefault('z', 'z'))rz   )rQ   r   r   s      r'   r]   z>elevate_as_needed.<locals>.wrapper_elevate.<locals>.<listcomp>  sR       "Q<<(OPP>!>>>>>  r7   z-cz<import os;os.environ.setdefault('_MENUINST_RECURSING', '1');z(*z,target_prefix=z,base_prefix=z,_mode='system',**)zElevating command: %sr   z,Elevation failed! Falling back to user mode.r   r   )r%   r$   r   r   )popr   r   r    r!   r"   rl   rk   r   r`   r   r   r   r   r   r   warnr   )
r%   r$   rK   kwargsimport_funcenv_varscmdreturn_coder   r   s
            r'   wrapper_elevatez*elevate_as_needed.<locals>.wrapper_elevatei  s    	

7D!!!%<)<!8%8{335	
455<<&?j000=0 0=E0 0!]0 0 0  K #T$/"S"S4="S"S"SK88 &(j&6&6&8&8   &((

 
 #
 }	
 

 
 
 &3
 
 $/
 
  
 
 
 4c:::*3// 
4d;;;!##F $	  Z Z ZJUXYYYYYYYYZ __ 	4  ,' 
    t  (#
 

 
 
 	
s    CE 
F FF)r   r    PathLike)r   r   s   ` r'   elevate_as_neededr   W  sn    $ 4[[ &*#'F
 F
 F
{F
 [F
 F
 F
 F
 F
 [F
P r7   r   c           	      v   t           j        dk    r6|r!t           j                            |d          }nd}t	          |d          }nt
          j        }t          dt                      dt           j	        
                    dd          d||	           t           j        dk    r|                                 d S d S )
Nr   z_test_output.txtru   zuser_is_admin():zenv_var:MENUINST_TESTzN/Az_mode:)file)r    rg   r|   r`   openr   stdoutprintr   r!   r"   close)r%   r$   r   outputouts        r'   _test_elevationr     s    
 
w$ 	(W\\+/ABBFF'F63j	

..    
w$		 r7   Tc                 ,   t          j        | fddd|}|rft                              d|j        |j                   t                              d|j                   t                              d|j                   |r|                                 |S )NT)capture_outputr)   z%s returned %dz
stdout:
%sz
stderr:
%s)	r   runr   r   rK   
returncoder   stderrcheck_returncode)rK   checklogr   processs        r'   
logged_runr     s    nTL$TLLVLLG
 4%w|W5GHHH]GN333]GN333 #  """Nr7   )r   )r   r   )F)NNr   )FT)8r    r2   rx   r   r   xml.etree.ElementTreeetreeElementTreerF   
contextlibr   	functoolsr   r   loggingr   pathlibr   typingr   r	   r
   r   r   r   r   unicodedatar   rk   r   _TargetOrBase_UserOrSystemr(   DEFAULT_PREFIXr   ro   r6   r<   ElementrH   rJ   rq   r   r   r   r   r   r   r   intr   r   r   r   CompletedProcessr   rP   r7   r'   <module>r      s    				 				      



 ' ' ' ' ' ' ' ' '       & & & & & & & &             R R R R R R R R R R R R R R R R R R ! ! ! ! ! !	8		gh'89gfowx'889-6 -6= -6 -6 -6 -6` !**%of-- *# * * * *D) ) ) )( '/  8C=    // // // // // // // //d
 
 
 
 
 
 
 
  $     BK  D        
! !hw6G !G ! ! ! !H*r{ * * * * * *Z 1	Gt 	G 	G 	G 	GGx} G G G G G 8BK#8 HSM    .[H [ [ [ [ [~ ,0)-! BK("+&    4 9T      r7   