U Z4d@sdZddlmZddlmZddlmZddlmZdZ ddZ ee Z d d Z ee Z ee Zd Zd d ZddZd1ddZd2ddZddZddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd3d'd(Zd)d*Zd+d,Zd-d.Z d/d0Z!dS)4zF Our local cache of filefields, everything is private to this package.) defaultdict)apps)models) import_stringZ_django_cleanup_original_cachecCstSN)setrrM/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/django_cleanup/cache.pyfields_default sr cCsiSrrrrrr fields_dict_defaultsr z'{klass.__module__}.{klass.__qualname__}cCsftrdStD]P}t||r qt|}t|r2q|j}|D]}t|t j r@t ||j |q@qdS)z/Prepare the cache for all models, non-reentrantN) FIELDSrZ get_models ignore_modelget_model_namemodel_has_filefields_metaZ get_fields isinstancerZ FileFieldadd_field_for_modelname) select_modemodelroptsfieldrrr prepares    rcCs4t||t|t||<t|jt||<dS)z2Centralized function to make all our local caches.N)r addget_dotted_path FIELDS_FIELDSZstorageFIELDS_STORAGE) model_name field_namerrrr r,srNccsDt|r@t|}|dk r0t|ts&t||}|D] }|Vq4dS)z-Get the filefields for a model if it has themN)rr rrAssertionError difference)rexcludefieldsrrrr get_fields_for_model:s r#ccsV|dkr |}t|}|}t||dD](}t||d}||||j|jfVq(dS)z Yields (name, descriptor) for each file field given an instance Can use the `using` kwarg to change the instance that the `FieldFile` will receive. N)r!)rZget_deferred_fieldsr#getattr __class__rr)instanceusingrZdeferred_fieldsrZ fieldfilerrr fields_for_model_instanceGs r(cCstt||S)z$Restore a field from its dotted path)rrrrrrr get_field\sr*cCstt||S)z&Restore a storage from its dotted path)rrr)rrr get_field_storageasr+cCstj|jdS)z!get the dotted path for an object)klass) DOTTED_PATHformatr%)Zobject_rrr risrcCsdj|jdS)zreturns a unique model namez {opt.app_label}.{opt.model_name}optr.rrrrr rnsrcCsdj|jdS)zOreturns a mangled attribute name specific to the model for ignore functionalityz1_{opt.model_name}__{opt.app_label}_cleanup_ignorer/r1r2rrr get_mangled_ignoressr3cCsdj|jdS)zOreturns a mangled attribute name specific to the model for select functionalityz1_{opt.model_name}__{opt.app_label}_cleanup_selectr/r1r2rrr get_mangled_selectxsr4cCs|tkS)zCheck if a model has filefields)r rrrr rsrcCs"|rt|t| St|t|S)z"Check if a model should be ignored)hasattrr4r3)rrrrr r sr cCst|rt|tdS)z!Remove the cache from an instanceN) has_cachedelattr CACHE_NAMEr&rrr remove_instance_cachesr;cCs(|dkr |}t|ttt||ddS)z~ Make the cleanup cache for an instance. Can also change the source of the data with the `source` kwarg. N)r')setattrr9dictr()r&sourcerrr make_cleanup_caches  r?cCs t|tS)z&Check if an instance has a cache on it)r6r9r:rrr r7sr7cCst|t|S)z)Get a value from the cache on an instance)r$r9)r&rrrr get_field_attrsr@ccstD]}t|VqdS)z.Get all the models we have in the FIELDS cacheN)r rZ get_modelr5rrr cleanup_modelssrAcCstS)zGet a copy of the FIELDS cache)r copyrrrr cleanup_fieldssrC)N)N)N)"__doc__ collectionsrZ django.appsrZ django.dbrZdjango.utils.module_loadingrr9r r r rrr-rrr#r(r*r+rrr3r4rr r;r?r7r@rArCrrrr s: