B Ú‹dZ+ã@sŒdZddlmZddlZddlZddlZddddgZdd „Zdd d„Zdd d„Z Gd d„dƒZ Gdd„de ƒZ Gdd„de ƒZ ddd„Z dS)zx This module contains functions to determine where configuration and data/cache files used by Astropy should be placed. é)ÚwrapsNÚget_config_dirÚ get_cache_dirÚset_temp_configÚset_temp_cachec Csfytj d¡}WnNtk r`tjdkrPdtjkrDtjd}ntdƒ‚n tjdkr/.pkgname/`` rather than ``/.astropy`` (depending on platform). Returns ------- configdir : str The absolute path to the configuration directory. NÚXDG_CONFIG_HOMEÚconfig) rÚ _temp_pathrrrÚexistsÚmkdirÚabspathrrÚislinkÚ_find_or_create_root_dir)ÚrootnameÚlinktoÚxchZ config_pathÚxchpthrrrrLs        cCs®d}tjdk rDtj}tj ||¡}tj |¡s8t |¡tj |¡Stj  d¡}|dk rštj |¡rštj ||¡}tj  |¡sštj |¡r–tj |¡S|}tj t d||ƒ¡S)a© Determines the Astropy cache directory name and creates the directory if it doesn't exist. This directory is typically ``$HOME/.astropy/cache``, but if the XDG_CACHE_HOME environment variable is set and the ``$XDG_CACHE_HOME/astropy`` directory exists, it will be that directory. If neither exists, the former will be created and symlinked to the latter. Parameters ---------- rootname : str Name of the root cache directory. For example, if ``rootname = 'pkgname'``, the cache directory will be ``/.pkgname/``. Returns ------- cachedir : str The absolute path to the cache directory. NÚXDG_CACHE_HOMEÚcache) rr$rrrr%r&r'rrr(r))r*r+r,Ú cache_pathr-rrrr|s        c@s6eZdZdZdZd dd„Zdd„Zdd„Zd d „ZdS) Ú _SetTempPathNFcCs.|dk rtj |¡}||_||_|jj|_dS)N)rrr'Ú_pathÚ_deleteÚ __class__r$Ú _prev_path)ÚselfrÚdeleterrrÚ__init__±s  z_SetTempPath.__init__cCs:|j|j_y | d¡Stk r4|j|j_‚YnXdS)Nr!)r2r4r$Ú_default_path_getterrr5)r6rrrÚ __enter__¹s    z_SetTempPath.__enter__cGs*|j|j_|jr&|jdk r&t |j¡dS)N)r5r4r$r3r2ÚshutilÚrmtree)r6ÚargsrrrÚ__exit__Ás z_SetTempPath.__exit__cstˆƒ‡‡fdd„ƒ}|S)zImplements use as a decorator.c sˆˆ||ŽWdQRXdS)Nr)r=Úkwargs)Úfuncr6rrÚwrapperÊsz&_SetTempPath.__call__..wrapper)r)r6r@rAr)r@r6rÚ__call__Çsz_SetTempPath.__call__)NF) Ú__name__Ú __module__Ú __qualname__r$r9r8r:r>rBrrrrr1­s  r1cs4eZdZdZeeƒZ‡fdd„Z‡fdd„Z‡Z S)ra¢ Context manager to set a temporary path for the Astropy config, primarily for use with testing. If the path set by this context manager does not already exist it will be created, if possible. This may also be used as a decorator on a function to set the config path just within that function. Parameters ---------- path : str, optional The directory (which must exist) in which to find the Astropy config files, or create them if they do not already exist. If None, this restores the config path to the user's default config path as returned by `get_config_dir` as though this context manager were not in effect (this is useful for testing). In this case the ``delete`` argument is always ignored. delete : bool, optional If True, cleans up the temporary directory after exiting the temp context (default: False). cs(ddlm}| ¡|_| ¡tƒ ¡S)Né)Ú_cfgobjs)Ú configurationrGÚcopyÚ _cfgobjs_copyÚclearÚsuperr:)r6rG)r4rrr:ïs  zset_temp_config.__enter__cs4ddlm}| ¡| |j¡|`tƒj|ŽdS)NrF)rG)rHrGrKÚupdaterJrLr>)r6r=rG)r4rrr>ùs   zset_temp_config.__exit__) rCrDrEÚ__doc__Ú staticmethodrr9r:r>Ú __classcell__rr)r4rrÒs c@seZdZdZeeƒZdS)ra/ Context manager to set a temporary path for the Astropy download cache, primarily for use with testing (though there may be other applications for setting a different cache directory, for example to switch to a cache dedicated to large files). If the path set by this context manager does not already exist it will be created, if possible. This may also be used as a decorator on a function to set the cache path just within that function. Parameters ---------- path : str The directory (which must exist) in which to find the Astropy cache files, or create them if they do not already exist. If None, this restores the cache path to the user's default cache path as returned by `get_cache_dir` as though this context manager were not in effect (this is useful for testing). In this case the ``delete`` argument is always ignored. delete : bool, optional If True, cleans up the temporary directory after exiting the temp context (default: False). N)rCrDrErNrOrr9rrrrrscCs,tj tƒd|›¡}tj tƒd|›|¡}tj |¡sütj |¡szyt |¡Wqœtk rvtj |¡sr‚YqœXn"tj |¡sœd}t| |||¡ƒ‚yt |¡Wn"tk rÌtj |¡sÈ‚YnXt j   d¡sú|dk rútj |¡sút  ||¡n$tj |¡s d}t| |||¡ƒ‚tj  |¡S)NÚ.z2Intended {0} {1} directory {1} is actually a file.Úwin)rrrr r%r&rÚisdirÚformatÚsysÚplatformÚ startswithÚsymlinkr')Zdirnmr+ÚpkgnameZinnerdirZmaindirÚmsgrrrr)!s2        r))r!)r!)r!)rNÚ functoolsrrr;rUÚ__all__r rrr1rrr)rrrrÚs ; 0 1%/