U ku f'@sdZddlmZddlZddlZddlZddlmZddlm Z ddl m Z e r\ddl m Z Gd d d e Zd d d d dZd dd ddZd d d ddZd d d ddZddddZeddeZd gZdS)zWindows.) annotationsN) lru_cache) TYPE_CHECKING)PlatformDirsABC)Callablec@sHeZdZdZeddddZdddddd d d Zeddd d ZeddddZeddddZ eddddZ eddddZ eddddZ eddddZ eddddZeddddZeddddZeddd d!Zeddd"d#Zeddd$d%Zeddd&d'Zeddd(d)ZdS)*Windowsa `MSDN on where to store app data files `_. Makes use of the `appname `, `appauthor `, `version `, `roaming `, `opinion `, `ensure_exists `. strreturncCs(|jr dnd}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)selfconstrr8/tmp/pip-unpacked-wheel-kvdkdwa_/platformdirs/windows.py user_data_dirszWindows.user_data_dirN opinion_value str | None)rrr cCsg}|jr`|jdk r*|jp|j}||||j|dk rN|jrN|||jr`||jtjj|f|}|||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_dir8szWindows.site_data_dircCs|jS)zC:return: config directory tied to the user, same as `user_data_dir`rrrrruser_config_dir>szWindows.user_config_dircCs|jS)zF:return: config directory shared by the users, same as `site_data_dir`)r(r*rrrsite_config_dirCszWindows.site_config_dircCstjtd}|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_dirHszWindows.user_cache_dircCstjtd}|j|ddS)zd:return: cache directory shared by users, e.g. ``C:\ProgramData\$appauthor\$appname\Cache\$version``r%r-rr&r'rrrsite_cache_dirQszWindows.site_cache_dircCs|jS)zB:return: state directory tied to the user, same as `user_data_dir`r)r*rrruser_state_dirWszWindows.user_state_dircCs(|j}|jr$tj|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_dir\s  zWindows.user_log_dircCstjtdS)zN:return: documents directory tied to the user e.g. ``%USERPROFILE%\Documents``CSIDL_PERSONALrrrrr*rrruser_documents_direszWindows.user_documents_dircCstjtdS)zN:return: downloads directory tied to the user e.g. ``%USERPROFILE%\Downloads``CSIDL_DOWNLOADSr4r*rrruser_downloads_dirjszWindows.user_downloads_dircCstjtdS)zL:return: pictures directory tied to the user e.g. ``%USERPROFILE%\Pictures``CSIDL_MYPICTURESr4r*rrruser_pictures_diroszWindows.user_pictures_dircCstjtdS)zH:return: videos directory tied to the user e.g. ``%USERPROFILE%\Videos`` CSIDL_MYVIDEOr4r*rrruser_videos_dirtszWindows.user_videos_dircCstjtdS)zF:return: music directory tied to the user e.g. ``%USERPROFILE%\Music`` CSIDL_MYMUSICr4r*rrruser_music_diryszWindows.user_music_dircCstjtdS)zK:return: desktop directory tied to the user, e.g. ``%USERPROFILE%\Desktop``CSIDL_DESKTOPDIRECTORYr4r*rrruser_desktop_dir~szWindows.user_desktop_dircCs$tjtjtdd}||S)z :return: runtime directory tied to the user, e.g. ``%USERPROFILE%\AppData\Local\Temp\$appauthor\$appname`` r ZTemp)rrrr!rrr'rrruser_runtime_dirszWindows.user_runtime_dircCs|jS)zF:return: runtime directory shared by users, same as `user_runtime_dir`)r@r*rrrsite_runtime_dirszWindows.site_runtime_dir)__name__ __module__ __qualname____doc__propertyrrr(r+r,r.r/r0r2r5r7r9r;r=r?r@rArrrrrsD  rr ) csidl_namer cCsjt|}|dk r|Sdddd|}|dkr@d|}t|tj|}|dkrfd|}t||S)z&Get folder from environment variables.NAPPDATAZALLUSERSPROFILE LOCALAPPDATA)r r%r Unknown CSIDL name: zUnset environment variable: )(get_win_folder_if_csidl_name_not_env_varget ValueErrorrenviron)rGresultZ env_var_namemsgrrrget_win_folder_from_env_varss"   rQrcCs|dkr$tjtjtjddS|dkrHtjtjtjddS|dkrltjtjtjddS|dkrtjtjtjdd S|d krtjtjtjdd Sd S) zKGet folder for a CSIDL name that does not exist as an environment variable.r3 USERPROFILEZ Documentsr6 Downloadsr8ZPicturesr:ZVideosr<ZMusicN)rrr!rrN)rGrrrrKsrKc Csrddddddddd |}|d kr6d |}t|tjd krDtd 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 AppDataZPersonalz&{374DE290-123F-4565-9164-39C4925E467B}z My PictureszMy VideozMy Music)r r%r r3r6r8r:r<NrJwin32rz@Software\Microsoft\Windows\CurrentVersion\Explorer\Shell Folders) rLrMsysplatformNotImplementedErrorwinregOpenKeyHKEY_CURRENT_USER QueryValueExr )rGshell_folder_namerPrYkey directory_rrrget_win_folder_from_registrys(    rac Csddddddddd d |}|d kr8d |}t|td }ttd}|jd |d d|tdd|Drtd }|j |j |d r|}|dkrt j |j dS|j S)zGet folder with ctypes.#' () r r%r r3r8r:r<r6r>NrJiwindllrcss|]}t|dkVqdS)N)ord).0crrr sz,get_win_folder_via_ctypes..r6rS)rLrMctypescreate_unicode_buffergetattrshell32SHGetFolderPathWanykernel32GetShortPathNameWvaluerrr!)rG csidl_constrPbufrkbuf2rrrget_win_folder_via_ctypess2      r}zCallable[[str], str]r cCs<ttdrtSz ddl}Wntk r2tYSXtSdS)Nrkr)hasattrrqr}rY ImportErrorrQra)rYrrr_pick_get_win_folders   r)maxsize)rE __future__rrqrrV functoolsrtypingrapircollections.abcrrrQrKrar}rr__all__rrrrs&     %