o 5i'@sdZddlmZddlZddlZddlmZddlmZddl m Z er*ddl m Z Gd d d e Z dddZdddZdddZdddZdddZeddeZd gZdS)zWindows.) annotationsN) lru_cache) TYPE_CHECKING)PlatformDirsABC)Callablec@seZdZdZed+ddZddd,d d Zed+d dZed+ddZed+ddZ ed+ddZ ed+ddZ ed+ddZ ed+ddZ ed+ddZed+ddZed+dd Zed+d!d"Zed+d#d$Zed+d%d&Zed+d'd(Zed+d)d*ZdS)-Windowsa `MSDN on where to store app data files `_. Makes use of the `appname `, `appauthor `, `version `, `roaming `, `opinion `, `ensure_exists `. returnstrcCs(|jrdnd}tjt|}||S)z :return: data directory tied to the user, e.g. ``%USERPROFILE%\AppData\Local\$appauthor\$appname`` (not roaming) or ``%USERPROFILE%\AppData\Roaming\$appauthor\$appname`` (roaming) CSIDL_APPDATACSIDL_LOCAL_APPDATA)roamingospathnormpathget_win_folder _append_parts)selfconstrrf/root/parts/websockify/install/lib/python3.10/site-packages/setuptools/_vendor/platformdirs/windows.py user_data_dirs zWindows.user_data_dirN opinion_valuerr str | NonecCsg}|jr0|jdur|jp|j}||||j|dur'|jr'|||jr0||jtjj|g|R}|||S)NF) appname appauthorappendopinionversionrrjoin_optionally_create_directory)rrrparamsauthorrrrr&s       zWindows._append_partscCstjtd}||S)zT:return: data directory shared by users, e.g. ``C:\ProgramData\$appauthor\$appname``CSIDL_COMMON_APPDATArrrrrrrrrr site_data_dir5s zWindows.site_data_dircC|jS)zC:return: config directory tied to the user, same as `user_data_dir`rrrrruser_config_dir;zWindows.user_config_dircCr()zF:return: config directory shared by the users, same as `site_data_dir`)r'r*rrrsite_config_dir@r,zWindows.site_config_dircCtjtd}|j|ddS)z :return: cache directory tied to the user (if opinionated with ``Cache`` folder within ``$appname``) e.g. ``%USERPROFILE%\AppData\Local\$appauthor\$appname\Cache\$version`` r Cacherr%r&rrruser_cache_dirEszWindows.user_cache_dircCr.)zd:return: cache directory shared by users, e.g. ``C:\ProgramData\$appauthor\$appname\Cache\$version``r$r/rr%r&rrrsite_cache_dirNszWindows.site_cache_dircCr()zB:return: state directory tied to the user, same as `user_data_dir`r)r*rrruser_state_dirTr,zWindows.user_state_dircCs(|j}|jrtj|d}|||S)zg:return: log directory tied to the user, same as `user_data_dir` if not opinionated else ``Logs`` in itLogs)rrrrr r!r&rrr user_log_dirYs  zWindows.user_log_dircCtjtdS)zN:return: documents directory tied to the user e.g. ``%USERPROFILE%\Documents``CSIDL_PERSONALrrrrr*rrruser_documents_dirbzWindows.user_documents_dircCr5)zN:return: downloads directory tied to the user e.g. ``%USERPROFILE%\Downloads``CSIDL_DOWNLOADSr7r*rrruser_downloads_dirgr9zWindows.user_downloads_dircCr5)zL:return: pictures directory tied to the user e.g. ``%USERPROFILE%\Pictures``CSIDL_MYPICTURESr7r*rrruser_pictures_dirlr9zWindows.user_pictures_dircCr5)zH:return: videos directory tied to the user e.g. ``%USERPROFILE%\Videos`` CSIDL_MYVIDEOr7r*rrruser_videos_dirqr9zWindows.user_videos_dircCr5)zF:return: music directory tied to the user e.g. ``%USERPROFILE%\Music`` CSIDL_MYMUSICr7r*rrruser_music_dirvr9zWindows.user_music_dircCr5)zK:return: desktop directory tied to the user, e.g. ``%USERPROFILE%\Desktop``CSIDL_DESKTOPDIRECTORYr7r*rrruser_desktop_dir{r9zWindows.user_desktop_dircCs$tjtjtdd}||S)z :return: runtime directory tied to the user, e.g. ``%USERPROFILE%\AppData\Local\Temp\$appauthor\$appname`` r Temp)rrrr rrr&rrruser_runtime_dirs zWindows.user_runtime_dircCr()zF:return: runtime directory shared by users, same as `user_runtime_dir`)rEr*rrrsite_runtime_dirr,zWindows.site_runtime_dir)r r )rr rrr r )__name__ __module__ __qualname____doc__propertyrrr'r+r-r0r1r2r4r8r;r=r?rArCrErFrrrrrsF                r csidl_namer r cCsjt|}|dur |Sdddd|}|dur d|}t|tj|}|dur3d|}t||S)z&Get folder from environment variables.NAPPDATAALLUSERSPROFILE LOCALAPPDATA)r r$r Unknown CSIDL name: zUnset environment variable: )(get_win_folder_if_csidl_name_not_env_varget ValueErrorrenviron)rLresult env_var_namemsgrrrget_win_folder_from_env_varss"   rXrcCs|dkrtjtjtjddS|dkr$tjtjtjddS|dkr6tjtjtjddS|dkrHtjtjtjdd S|d krZtjtjtjdd Sd S) zMGet a folder for a CSIDL name that does not exist as an environment variable.r6 USERPROFILE Documentsr: Downloadsr<Picturesr>Videosr@MusicN)rrr rrT)rLrrrrQsrQc Csrddddddddd |}|d urd |}t|tjd kr"td d l}||jd}|||\}}t |S)z Get folder from the registry. This is a fallback technique at best. I'm not sure if using the registry for these guarantees us the correct answer for all CSIDL_* names. AppDatazCommon AppDataz Local AppDataPersonalz&{374DE290-123F-4565-9164-39C4925E467B}z My PictureszMy VideozMy Music)r r$r r6r:r<r>r@NrPwin32rz@Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders) rRrSsysplatformNotImplementedErrorwinregOpenKeyHKEY_CURRENT_USER QueryValueExr )rLshell_folder_namerWrekey directory_rrrget_win_folder_from_registrys(    rmc Csddl}ddddddd d d d |}|dur d |}t||d}t|d}|jd|dd|tdd|DrM|d}|j |j |drM|}|dkrYt j |j dS|j S)zGet folder with ctypes.rN#' () r r$r r6r<r>r@r:rBrPiwindllcss|] }t|dkVqdS)N)ord).0crrr sz,get_win_folder_via_ctypes..r:r[)ctypesrRrScreate_unicode_buffergetattrshell32SHGetFolderPathWanykernel32GetShortPathNameWvaluerrr )rLr} csidl_constrWbufrwbuf2rrrget_win_folder_via_ctypess4     rCallable[[str], str]cCsTzddl}Wn tyYnwt|drtSzddl}WtSty)tYSw)Nrrw)r} ImportErrorhasattrrrerXrm)r}rerrr_pick_get_win_folders     r)maxsize)rLr r r )rLr r r)r r)rJ __future__rrrb functoolsrtypingrapircollections.abcrrrXrQrmrrr__all__rrrrs$          '