U |e@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)setrrQ/var/www/website-v5/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) FIELDSr get_models ignore_modelget_model_namemodel_has_filefields_meta get_fields isinstancer 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_FIELDSstorageFIELDS_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)r!excludefieldsr"rrr 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%)rget_deferred_fieldsr'getattr __class__rr)instanceusingr!deferred_fieldsr"Z fieldfilerrr fields_for_model_instanceGs r.cCstt||S)z$Restore a field from its dotted path)rrr!r"rrr get_field\sr0cCstt||S)z&Restore a storage from its dotted path)rr r/rrr get_field_storageasr1cCstj|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}optr4rrrrr rnsrcCsdj|jdS)zOreturns a mangled attribute name specific to the model for ignore functionalityz1_{opt.model_name}__{opt.app_label}_cleanup_ignorer5r7r8rrr get_mangled_ignoressr9cCsdj|jdS)zOreturns a mangled attribute name specific to the model for select functionalityz1_{opt.model_name}__{opt.app_label}_cleanup_selectr5r7r8rrr get_mangled_selectxsr:cCs|tkS)zCheck if a model has filefields)r r!rrr rsrcCs"|rt|t| St|t|S)z"Check if a model should be ignored)hasattrr:r9)rrrrr rsrcCst|rt|tdS)z!Remove the cache from an instanceN) has_cachedelattr CACHE_NAMEr+rrr remove_instance_cachesrAcCs(|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,)setattrr?dictr.)r+sourcerrr make_cleanup_caches  rEcCs t|tS)z&Check if an instance has a cache on it)r<r?r@rrr r=sr=cCst|t|S)z)Get a value from the cache on an instance)r)r?)r+r"rrr get_field_attrsrFccstD]}t|VqdS)z.Get all the models we have in the FIELDS cacheN)r r get_modelr;rrr cleanup_modelssrHcCstS)zGet a copy of the FIELDS cache)r copyrrrr cleanup_fieldssrJ)N)N)N)"__doc__ collectionsr django.appsr django.dbrdjango.utils.module_loadingrr?r r r rr r3rrr'r.r0r1rrr9r:rrrArEr=rFrHrJrrrr s: