͈2gqNdZddlZddlZddlZddlZddlmZddlmZddl m Z ddl m Z ddl mZmZmZmZmZdd lmZmZmZd d lmZmZd d lmZd d lmZd dlmZm Z m!Z!m"Z"m#Z#ee$Z%GddeZ&GddeZ'y) N) getLogger)Path)CompletedProcess)NamedTemporaryFile)AnyDictListOptionalTuple)WinLex logged_rununlink)MenuMenuItem) folder_path)windows_terminal_settings_files)notify_shell_changesregister_file_extensionregister_url_protocolunregister_file_extensionunregister_url_protocolc 8eZdZdeej fdZdeej fdZede fdZ ede fdZ ede fdZ ede e fdZedeeefffd Zd Z dd ed ed eeeefdeffd Zdde fdZdee fdZxZS) WindowsMenureturnc:tjd|j|jjdd|jr|jjdd|j r|j jdd|jfS)Nz Creating %sTparentsexist_ok)logdebugstart_menu_locationmkdirquick_launch_locationdesktop_locationselfs 6lib/python3.12/site-packages/menuinst/platforms/win.pycreatezWindowsMenu.creates -!9!9:   &&td&C  % %  & & , ,TD , I   ! ! ' 't ' D((**clt|j}|jr' t|j |jfS|jfS#t $rPt jd|jtj|jdY|jfSwxYw)N Removing %sT) ignore_errors) rr$existsnextiterdir StopIterationr"r#shutilrmtree)r) menu_locations r*removezWindowsMenu.remove'sT556    ! L]**,-((**((**! L -)A)AB d66dK((** LsAA B32B3cztt|jdd|j|jz S)a  On Windows we can create shortcuts in several places: - Start Menu - Desktop - Quick launch (only for user installs) In this property we only report the path to the Start menu. For other menus, check their respective properties. Fstart)rwindows_folder_pathmoderendernamer(s r*r$zWindowsMenu.start_menu_location4s2' 5'BCdkkRVR[R[F\\\r,c|jdk(rtjdytt |jddS)Nsystemz>Quick launch menus are not available for system level installsF quicklaunch)r;warningswarnrr:r(s r*r&z!WindowsMenu.quick_launch_locationBs7 99 MMZ [ ' 5-HIIr,cBtt|jddS)NFdesktop)rr:r;r(s r*r'zWindowsMenu.desktop_locationJs' 5)DEEr,cx|jdk(rtjdgSt|jS)zLocation of the Windows terminal profiles. The parent directory is used to check if Terminal is installed because the settings file is generated when Terminal is opened, not when it is installed. r?z=Terminal profiles are not available for system level installs)r;r"warningrr(s r*terminal_profile_locationsz&WindowsMenu.terminal_profile_locationsNs1 99 KKW XI.tyy99r,c t|}|jt|jdz t|jdz t|jdz t|j dz t|j dz t|jdz dz t|j dd|S)NScriptsz python.exez pythonw.exeLibrarybinico) SCRIPTS_DIRPYTHONPYTHONW BASE_PYTHON BASE_PYTHONWBIN_DIRSP_DIRICON_EXT)super placeholdersupdatestrprefix base_prefix_site_packages)r)rV __class__s r*rVzWindowsMenu.placeholders[sw+ "4;;#:;dkkL89t{{]:;"4#3#3l#BC #D$4$4}$D Et{{Y6>?d1134!  r,c J|jdz ttjj dd|jdz dz |jdz dz ttjj dd|jdz dz |jdz dz fS) Nz _conda.exe CONDA_EXEz"C:\oops\a_file_that_does_not_existcondabinz conda.batrK MAMBA_EXEzmicromamba.exe)rZrosenvirongetr(s r*_conda_exe_path_candidatesz&WindowsMenu._conda_exe_path_candidatesls   | +  -RS T   z )K 7   u ${ 2  -RS T   z ),< <   u $'7 7  r,valueslugextract||||}t|drd|vr|ddk7r|jdd}|S)a? We extend the render method here to replace forward slashes with backslashes. We ONLY do it if the string does not start with /, because it might be just a windows-style command-line flag (e.g. cmd /K something). This will not escape strings such as `/flag:something`, common in compiler stuff but we can assume such windows specific constructs will have their platform override, which is always an option. This is just a convenience for things like icon paths or Unix-like paths in the command key. )rfrgreplace/r\)rUr<hasattrri)r)rerfrgr\s r*r<zWindowsMenu.renderwsHu4u= 5) $%(c/MM#t,E r,cB| |j}|jdz dz S)NLibz site-packages)rY)r)rYs r*r[zWindowsMenu._site_packagess$ >[[F{{U"_44r,c|jfSN)r$r(s r*_pathszWindowsMenu._pathss((**r,)FNrp)__name__ __module__ __qualname__r raPathLiker+r7propertyrr$r&r'r rGr rXrVrdrboolr r<r[rq __classcell__)r\s@r*rrs#+bkk*+ +bkk* + ]T ] ]JtJJF$FF :DJ : :d38n  QU $5=d38n5M (5T5 +d +r,rcZeZdZedefdZdeedffdZdeedffdZdeedffdZ dde fdZ defd Z d Z de fd Zdd eej"defdZddee fdZddedefdZdZdZdZed defdZed defdZdefdZdefdZdefdZdefdZ dZ!y )!WindowsMenuItemrcP|jj|jz S)z Path to the .lnk file placed in the Start Menu On Windows, menuinst can create up to three shortcuts (start menu, desktop, quick launch) This property only lists the one for start menu )menur$_shortcut_filenamer(s r*locationzWindowsMenuItem.locations"yy,,t/F/F/HHHr,.c :ddlm}|j|j}|D]}|jdk(s|j ^}}|j d}|r:ttjj|jddndtjvrd}nd }|j d xsd }t|jrtjd ||||j!d t#|dj%|||d|j'|j(j*D]}|j-|d|j/} |j1} | s| r t3|S)Nr)create_shortcutz.lnk working_dirTr USERPROFILEz %USERPROFILE%z%HOMEDRIVE%%HOMEPATH%iconz Overwriting existing link at %s.)ext rFr7)win_utils.winshortcutr _precreaterqsuffix_process_command render_keyrrapath expandvarsr%rbr0r"rFr}rXjoin_app_user_model_idr|rG$_add_remove_windows_terminal_profile_register_file_extensions_register_url_protocolsr) r)rpathsr target_path argumentsrrr~changed_extensionschanged_protocolss r*r+zWindowsMenuItem.createsh:  D;;&(&*&;&;&= #K)//-8KRWW'' 45;;DSW;X "**,- 5 ??6*0bD Dz  " >E ''B'/D #'') 3H < ?   dii== >  ZXZ i$"9"9";;ZX XE == $ LL..0 1U| Ys*DrcB|rd|nd}|jdi|S)N.rr=)rgr)r)rs r*r}z"WindowsMenuItem._shortcut_filenames.#iB//&/34SE::r,cjt|jjd|jdz S)NMENU_DIRbat)rr|rVr}r(s r*rz WindowsMenuItem._path_for_scripts-DII**:67$:Q:QRW:XXXr,c|jd}|sytdd5}|j|dddttj j dddz }t|dz d z d z d jd g}t|d t j|jy#1swYxYw)N precreateFw)deleter; SystemRoot C:\Windowssystem32WindowsPowerShellv1.0powershell.exe"start 'z' -WindowStyle hidden"Tcheck) rrwriterrarbrcrXr=rr)r)precreate_codetmprcmds r*rzWindowsMenuItem._precreates5  u3 73 IIn %8 |]CDzQ ..7:JJ Kz!8 9  3d# #((8 7s B99Cc ddg}|jd}|r|j||jdrv|jj}|jj |rd}nd}|jjd|d|jj d }|d d |d d dgz }|jdjtj|jddj|S)Nz @ECHO OFFz%:: Script generated by conda/menuinst precommandrzshell activatezshell.cmd.exe activaterz ""z @SETLOCAL ENABLEDELAYEDEXPANSIONz%@FOR /F "usebackq tokens=*" %%i IN (`z`) do set "ACTIVATOR=%%i"z@CALL %ACTIVATOR%z!:: This below is the user commandcommandz ) rrrr| conda_exe_is_micromambarYrr quote_args)r)linesrrr activators r*_commandzWindowsMenuItem._commands  3 __\2  LL $ == $ ++Iyy'' 2+399../q "TYY=M=Mz3WindowsMenuItem._process_command..Ms@T_s) rrrrrarbrcrallrr)r) with_arg1scriptrrarg1s r*rz WindowsMenuItem._process_command1s! == $'')F}}Z( $&m<NN4(&N#  |] KLzY )ur 9,-Q/#66?BRRSSTU"xr$/DE N//), @@@ NN4   ))r,r~r7c|jjdr|jjsy|j d}|jr#t j |jni}d}t|jdijdgD]\}}|jd|k(s|}n|r|dkry|dd|=ndjtj|j d |d }|jjd r|j d |d <|jjd r|j d |d <|dkr0d|vri|d<d|dvrg|dd<|ddj|n$tjd|d||dd|<|jt j |dy)aAdd/remove the Windows Terminal profile. Windows Terminal is using the name of the profile to create a GUID, so the name will be used as the unique identifier to find existing profiles. If the Terminal app has never been opened, the settings file may not exist yet. Writing a minimal profile file will not break the application - Terminal will automatically generate the missing options and profiles without overwriting the profiles menuinst has created. terminal_profileNprofileslistr=rrr) commandliner=rrstartingDirectoryz!Overwriting terminal profile for r)indent)rrcrr0rjsonloads read_text enumeraterrrrr"rF write_textdumps) r)r~r7r=settingsindexpprofile profile_datas r*rz4WindowsMenuItem._add_remove_windows_terminal_profileQs}}  !34HOO V$}}  /48OOM4R 01qyX-+-HZ(*!5535HZ(0$V,33LA ?vQGH6B$V,U3DJJx:;r,c6|jdd}|d|S)Nr=Trfz .AssocFiler)r) extension identifiers r*_ftype_identifierz!WindowsMenuItem._ftype_identifiers%__V$_7 Z {33r,c ^|jd}|sydj|j}ttj |Dcgc]}|j c}}|D]8}|j|}|j|||j||:ycc}w)H This function uses CMD's `assoc` and `ftype` commands. file_extensionsNr) associate_to) rrrrdictfromkeyslowerr _cmd_ftype _cmd_assoc)r) extensionsrrextsrs r*_register_file_extensions_cmdz-WindowsMenuItem._register_file_extensions_cmds]]#45  ((40023DMM*"E*3399;*"EFGC//4J OOJ 0 OOCjO 9#Fs B* c |jd}|syttj|Dcgc]}|j c}}|D]$}|j |}|j |&ycc}w)rrN)rrrrrrrr)rrrrs r*_unregister_file_extensions_cmdz/WindowsMenuItem._unregister_file_extensions_cmdsm]]#45  DMM*"E*3399;*"EFGC//4J OOJ '#FsA7 ct|||gdk7r td|jds td|r|d|}n |r|}n|r|d}tdddd gd S) zVhttps://learn.microsoft.com/en-us/windows-server/administration/windows-commands/assocrz6Only one of {associate_to, query, remove} must be set.rzextension must startwith '.'=rrrassoc Tr)sum ValueError startswithr)rrqueryr7rs r*rzWindowsMenuItem._cmd_assocs  eV, - 2UV V##C(;< < Kq/C C Kq/C5$sen=TJJr,ct|||gdk7r td|r|d|}n |r|}n|r|d}tddddgd S) zVhttps://learn.microsoft.com/en-us/windows-server/administration/windows-commands/ftyperz1Only one of {command, query, remove} must be set.rrrrrTr)rrr)rrr r7rs r*rzWindowsMenuItem._cmd_ftypesj ' (A -PQ Q L'+C C L"C5$sen=TJJr,c |jd}|sydj|jd}|jd}t t j |Dcgc]}|jc}}|D]V}|j|}t|||||jd|j|jjXycc}w) WIPrFrTrrr=rapp_nameapp_user_model_idr;) rrrrrrrrrrrr|r;)r)rrrrrrs r*rz)WindowsMenuItem._register_file_extensionss]]#45 ((40040@Av&DMM*"E*3399;*"EFGC//4J #0"&"9"9";YY^^ #FsC c |jd}|syttj|Dcgc]}|j c}}|D]5}|j |}t |||jj7ycc}w)NrFr;T) rrrrrrrr|r;rs r*rz+WindowsMenuItem._unregister_file_extensionssx]]#45 DMM*"E*3399;*"EFGC//4J %c:DIINN K #FsB c B|jd}|sydj|jd}|jd}|D]V}|j |}t |||||jd|j |jjXy) z~See https://learn.microsoft.com/en-us/previous-versions/windows/internet-explorer/ie-developer/platform-apis/aa767914(v=vs.85) url_protocolsFrTrrr=r) rrrrrrrr|r;)r) protocolsrrprotocolrs r*rz'WindowsMenuItem._register_url_protocolssMM/2 ((40040@Av&!H//9J !0"&"9"9";YY^^ "r,c|jd}|sy|D]5}|j|}t|||jj7y)NrFrT)rrrr|r;)r)rrrs r*rz)WindowsMenuItem._unregister_url_protocolssIMM/2 !H//9J #Hjtyy~~ N"r,c~|jd}|s)d|jddjddddS|S) Nrz Menuinst.r=Trrr)rri)r)aumis r*rz"WindowsMenuItem._app_user_model_idsJ23tvDAII#rRSTUYVYZ Z r,)lnkrp)F)NFF)"rrrsrtrvrr~r r+r7rqrXr}rrrr rarurrrwrrrr staticmethodrrrrrrrrrr,r*rzrzsL I$II1dCi(1fdCi( dCi($;c;Y$Y "#"4 "++)> $ *5:*@-r(sl ''33.. FCt+$t+nihir,