B Wvg@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_cachecCstS)N)setrr@/tmp/pip-install-4y76y3dr/django-cleanup/django_cleanup/cache.pyfields_default sr cCsiS)Nrrrrrfields_dict_defaultsr z'{klass.__module__}.{klass.__qualname__}cCsntrdSx`tD]T}t||r"qt|}t|r4q|j}x*|D]}t|t j rDt ||j |qDWqWdS)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_modemodelroptsfieldrrrprepares  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_namerrrrr,srNccsHt|rDt|}|dk r0t|ts&t||}x|D] }|Vq6WdS)z-Get the filefields for a model if it has themN)rr rrAssertionError difference)rexcludefieldsrrrrget_fields_for_model:s  r"ccsZ|dkr |}t|}|}x8t||dD](}t||d}||||j|jfVq*WdS)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 )r Zget_deferred_fieldsr"getattr __class__rr)instanceusingrZdeferred_fieldsrZ fieldfilerrrfields_for_model_instanceGs r'cCstt||S)z$Restore a field from its dotted path)rr)rrrrr get_field\sr(cCstt||S)z&Restore a storage from its dotted path)rr)rrrrrget_field_storageasr)cCstj|jdS)z!get the dotted path for an object)klass) DOTTED_PATHformatr$)Zobject_rrrrisrcCsdj|jdS)zreturns a unique model namez {opt.app_label}.{opt.model_name})opt)r,r)rrrrr nsr cCsdj|jdS)zOreturns a mangled attribute name specific to the model for ignore functionalityz1_{opt.model_name}__{opt.app_label}_cleanup_ignore)r-)r,r)rrrrget_mangled_ignoressr.cCsdj|jdS)zOreturns a mangled attribute name specific to the model for select functionalityz1_{opt.model_name}__{opt.app_label}_cleanup_select)r-)r,r)rrrrget_mangled_selectxsr/cCs|tkS)zCheck if a model has filefields)r )rrrrrsrcCs"|rt|t| St|t|S)z"Check if a model should be ignored)hasattrr/r.)rrrrrr sr cCst|rt|tdS)z!Remove the cache from an instanceN) has_cachedelattr CACHE_NAME)r%rrrremove_instance_cachesr4cCs(|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&)setattrr3dictr')r%sourcerrrmake_cleanup_cachesr8cCs t|tS)z&Check if an instance has a cache on it)r0r3)r%rrrr1sr1cCst|t|S)z)Get a value from the cache on an instance)r#r3)r%rrrrget_field_attrsr9ccsxtD]}t|VqWdS)z.Get all the models we have in the FIELDS cacheN)r rZ get_model)rrrrcleanup_modelss r:cCstS)zGet a copy of the FIELDS cache)r copyrrrrcleanup_fieldssr<)N)N)N)"__doc__ collectionsrZ django.appsrZ django.dbrZdjango.utils.module_loadingrr3r r r rrr+rrr"r'r(r)rr r.r/rr r4r8r1r9r:r<rrrrs: