B Khh @s dZddlZddlZddlZddlmZddlmZmZm Z m Z ddl m Z ddl mZddlmZeZeZe eefZeeefZedd d ZeZeed d d Zeed ddZeedddZeeee e ee efdddZeeeeddddZdS)z;Caching of formatted files with feature-based invalidation.N)Path)DictIterableSetTuple)user_cache_dir)version)Mode)returncCs"tdtd}ttjd|}|S)a[Get the cache directory used by black. Users can customize this directory on all systems using `BLACK_CACHE_DIR` environment variable. By default, the cache directory is the user cache directory under the black application. This result is immediately set to a constant `black.cache.CACHE_DIR` as to avoid repeated calls. Zblack)rZBLACK_CACHE_DIR)r __version__rosenvironget)Zdefault_cache_dir cache_dirr./tmp/pip-install-vv81h98y/black/black/cache.py get_cache_dirs r)moder c CsZt|}|siS|d2}yt|}Wntjttfk rJiSXWdQRX|S)zRead the cache if it exists and is well formed. If it is not well formed, the call to write_cache later should resolve the issue. rbN)get_cache_fileexistsopenpickleloadUnpicklingError ValueError IndexError)r cache_fileZfobjcacherrr read_cache(s rcCstd|dS)Nzcache.z.pickle) CACHE_DIRZ get_cache_key)rrrrr:sr)pathr cCs|}|j|jfS)zKReturn the information used to check if a file is already formatted or not.)statst_mtimest_size)r!r"rrrget_cache_info>sr%)rsourcesr cCsXtt}}x@|D]8}|}|t|t|krB||q||qW||fS)zSplit an iterable of paths in `sources` into two sets. The first contains paths of files that modified on disk or are not in the cache. The other contains paths to non-modified files. )setresolverstrr%add)rr&tododonesrcZres_srcrrr filter_cachedDs  r.)rr&rr c Cst|}ybtjddd|dd|D}tjt|jdd}tj||ddWd QRXt |j |Wnt k r~YnXd S) zUpdate the cache file.T)parentsexist_okcSsi|]}t|t|qSr)r%r)r().0r-rrr [szwrite_cache..F)dirdelete)protocolN) rr mkdirtempfileNamedTemporaryFiler)parentrdumpr replacenameOSError)rr&rrZ new_cachefrrr write_cacheTsr@) __doc__r rr8ZpathlibrtypingrrrrZ platformdirsrZ_black_versionrr Z black.moder floatZ TimestampintZFileSize CacheInfor)Cacherr rrr%r.r@rrrrs&      &