3 g&M@sddlZddlmZmZddlmZmZddlmZddlm Z ddl m Z m Z m Z ddlmZmZmZGdd d e ZGd d d e Zd d ZddZddZddZddZddZddZddZGdddZdS)N)Counter defaultdict)partialreduce)chain) attrgetter)IntegrityError connections transaction) query_utilssignalssqlcseZdZfddZZS)ProtectedErrorcs||_tj||dS)N)protected_objectssuper__init__)selfmsgr) __class__W/var/www/tester-filtro-web/env/lib/python3.6/site-packages/django/db/models/deletion.pyr szProtectedError.__init__)__name__ __module__ __qualname__r __classcell__rr)rrr srcseZdZfddZZS)RestrictedErrorcs||_tj||dS)N)restricted_objectsrr)rrr)rrrrszRestrictedError.__init__)rrrrrrr)rrrsrcCsB|j||jj|j|jdd|jr>t|jj r>|j|d|dS)NF)source source_attrnullablefail_on_restricted) collect remote_fieldmodelnamenullr featuresZcan_defer_constraint_checksadd_field_update) collectorfieldsub_objsusingrrrCASCADEs   r,cCs(td|jjj|djj|jf|dS)NzoCannot delete some instances of model '%s' because they are referenced through a protected foreign key: '%s.%s'r)rr"r#rrr$)r(r)r*r+rrrPROTECT sr-cCs"|j|||j|jj|jdS)N)add_restricted_objectsadd_dependencyr"r#)r(r)r*r+rrrRESTRICT*s r0cs4trfdd}n fdd}fdd|_|S)Ncs|j||dS)N)r')r(r)r*r+)valuerr set_on_delete1szSET..set_on_deletecs|j||dS)N)r')r(r)r*r+)r1rrr24scs dfifS)Nzdjango.db.models.SETrr)r1rr6szSET..)callableZ deconstruct)r1r2r)r1rSET/s  r5cCs|j|d|dS)N)r')r(r)r*r+rrrSET_NULL:sr6cCs|j||j|dS)N)r' get_default)r(r)r*r+rrr SET_DEFAULT>sr8cCsdS)Nr)r(r)r*r+rrr DO_NOTHINGBsr9cCsdd|jddDS)Ncss,|]$}|jr|j r|js |jr|VqdS)N) auto_createdZconcreteZ one_to_oneZ one_to_many).0frrr Jsz4get_candidate_relations_to_delete..T)Zinclude_hidden)Z get_fields)optsrrr!get_candidate_relations_to_deleteFsr?c@seZdZddZd"ddZd#ddZd d Zd d Zd dZddZ ddZ d$ddZ ddZ d%ddZ ddZddZddZd d!ZdS)& CollectorcCsD||_tt|_tttt|_tttt|_g|_tt|_dS)N) r+rsetdatar field_updatesr fast_deletes dependencies)rr+rrrrPs  zCollector.__init__NFc Csl|sgSg}|dj}|j|}x|D]}||kr&|j|q&W|j||dk rh| rh|j|||d|S)a7 Add 'objs' to the collection of objects to be deleted. If the call is the result of a cascade, 'source' should be the model that caused it, and 'nullable' should be set to True if the relation can be null. Return a list of all objects that were not already collected. rN)reverse_dependency)rrBappendupdater/) robjsrrrFnew_objsr# instancesobjrrraddcs    z Collector.addcCs>|r||}}|j|jjj|jj|jj||jjdS)N)rE_metaconcrete_modelrMrB setdefaultdefault_factory)rr# dependencyrFrrrr/{s zCollector.add_dependencycCs.|sdS|dj}|j|||fj|dS)z Schedule a field update. 'objs' must be a homogeneous iterable collection of model instances (e.g. a QuerySet). Nr)rrCrH)rr)r1rIr#rrrr's zCollector.add_field_updatecCs&|r"|dj}|j||j|dS)Nr)rrrH)rr)rIr#rrrr.s z Collector.add_restricted_objectscs0||jkr,fdd|j|jD|j|<dS)Ncsi|]\}}||qSrr)r;r)items)rIrr sz?Collector.clear_restricted_objects_from_set..)rrS)rr#rIr)rIr!clear_restricted_objects_from_sets  z+Collector.clear_restricted_objects_from_setcCs>||jkr:t|jdd|j|jDd}|j||dS)NcSsg|]}|D] }|jq qSr)pk)r;rIrLrrr szDCollector.clear_restricted_objects_from_queryset..)Zpk__in)rrAfiltervaluesrU)rr#qsrIrrr&clear_restricted_objects_from_querysets  z0Collector.clear_restricted_objects_from_querysetcCstjj|ptjj|S)N)r pre_deleteZ has_listeners post_delete)rr#rrr_has_signal_listenerss zCollector._has_signal_listenerscsrjjtk rdSt|dr(|jj}n t|drDt|drD|j}ndS|j|rVdS|j}tfdd|jjj j Dotddt |Dot dd|j D S) a Determine if the objects in the given queryset-like or single object can be fast-deleted. This can be done if there are no cascades, no parents and no signal listeners for the object class. The 'from_field' tells where we are coming from - we need this to determine if the objects are in fact to be deleted. Allow also skipping parent -> child -> parent chain preventing fast delete of the child. FrNr# _raw_deletec3s|]}|kVqdS)Nr)r;link) from_fieldrrr=sz,Collector.can_fast_delete..css|]}|jjjtkVqdS)N)r)r" on_deleter9)r;relatedrrrr=scss|]}t|dVqdS)bulk_related_objectsN)hasattr)r;r)rrrr=s)r"rbr,rerNr#r^allrOparentsrYr?anyprivate_fields)rrIrar#r>r)rarcan_fast_deletes     zCollector.can_fast_deletecs^dd|D}tt|jjj|dtkrTfddtdtDSgSdS)zT Return the objs in suitably sized batches for the used connection. cSsg|] }|jqSr)r$)r;r)rrrrWsz-Collector.get_del_batches..csg|]}||qSrr)r;i)conn_batch_sizerIrrrWsrN)maxr r+opsZbulk_batch_sizelenrange)rrIfields field_namesr)rmrIrget_del_batchess  zCollector.get_del_batchesTc s|j|r|jj|dS|j||||d} | s4dS| dj} |s| jj} xD| jjjD]4rXfdd| D} |j | | j j ddddqXW|sdS|rt | jj } tt}tt}x6t| jD]&}|r|j| krq|j}|j jtkrq|j}|j||d r||j|q|j| |g}x|D]}|j||g|}|jjpR|j|st tjd d t|jD}|jt|}|r,y|j j||||jWnHt k r}z*d |jj!|j"f}|||j#7<WYdd}~XnXq,WqW|r"t d | j!dj$|ft tj|jxL|j%D]@\}}|j| |}x(|D] }|j|||}|jj|qFWq,Wx>| jj&D]2}t'|drz|j(| |j}|j || dddqzW|rx$|j)j%D]\}}|j*||qWx|jD]}|j+|j|qW|j,jrtt}xV|j,j%D]H\}}x<|j%D]0\}}|r.d |j!|j"f}|||7<q.WqW|rt-d| j!dj$|ft tj|jdS)a Add 'objs' to the collection of objects to be deleted as well as all parent instances. 'objs' must be a homogeneous iterable collection of model instances (e.g. a QuerySet). If 'collect_related' is True, related objects will be handled by their respective on_delete handler. If the call is the result of a cascade, 'source' should be the model that caused it and 'nullable' should be set to True, if the relation can be null. If 'reverse_dependency' is True, 'source' will be deleted before the current model, rather than after. (Needed for cascading to parent models, the one case in which the cascade follows the forwards direction of an FK rather than the reverse direction.) If 'keep_parents' is True, data of parent model's will be not deleted. If 'fail_on_restricted' is False, error won't be raised even if it's prohibited to delete such objects due to RESTRICT, that defers restricted object checking in recursive calls where the top-level call may need to collect more objects to determine whether restricted ones can be deleted. N)rFrcsg|]}t|jqSr)getattrr$)r;rL)ptrrrrWsz%Collector.collect..FT)rrcollect_relatedrFr )racss |]}dd|jjDVqdS)css|] }|jVqdS)N)attname)r;rfrrrr=#sz.Collector.collect...N)r)Zforeign_related_fields)r;relrrrr=#sz$Collector.collect..z'%s.%s'zhCannot delete some instances of model %r because they are referenced through protected foreign keys: %s.z, rd)rrr ziCannot delete some instances of model %r because they are referenced through restricted foreign keys: %s.).rjrDrGrMrrNrOrgrYr!r"Z related_namerAZget_parent_listrlistr?r#r)rbr9 related_modelrtrelated_objectsqueryZselect_relatedr^r from_iterableonlytupler+rrr$rjoinrSrirerdrBrUr[rr)rrIrrrwrrFZ keep_parentsr rJr#rOZ parent_objsrgZmodel_fast_deletesrrcr)r|Zbatchesbatchr*Zreferenced_fieldserrorkeyrelated_fieldsrKrZrrrr)rvrr!s       .       zCollector.collectcs.ttjfdd|D}|jj|jj|S)zQ Get a QuerySet of the related model to objs via related fields. c3s$|]}tjfd|jiVqdS)z%s__inN)r Qr$)r;Z related_field)rIrrr=_sz,Collector.related_objects..)roperatoror_Z _base_managerr+rX)rr|rrI predicater)rIrr}Zs  zCollector.related_objectsccs4x.|jjD] \}}x|D]}||fVqWq WdS)N)rBrS)rr#rKrLrrrinstances_with_modelds zCollector.instances_with_modelcsg}t}tj}xrt|t|krd}xP|D]H}||kr>q0jj|jj}|oZ|j|s0|j ||j |jjd}q0W|sdSqWfdd|D_dS)NFTcsi|]}j||qSr)rB)r;r#)rrrrTysz"Collector.sort..) rAr{rBrprEgetrNrO differencerGrM)rZ sorted_modelsZconcrete_modelsmodelsfoundr#rEr)rrsortis    zCollector.sortc Csx,|jjD]\}}t|tdd|j|<q W|jt}t|jdkrt|dkrt|d}|j|rt j |j t j |j|jg|j }WdQRXt||jjjd||jj|ifSt j|j ddzx0|jD]$\}}|jjstjj|||j dqWx:|jD]0}|j|j d}|r ||jjj|7<q Wx^|jjD]P\}}xD|jD]8\\} } }t j|} | jd d |D| j | i|j q^WqLWx|jj!D]}|j"qWx|jjD]x\}}t j |} d d |D} | j| |j }|r||jj|7<|jjsx"|D]}tj#j|||j dq"WqWWdQRXxN|jj!D]@}x8|jD],\\} } }x|D]}t|| j| q|WqjWq\Wx:|jjD],\}}x |D]}t||jjjdqWqWt$|j!t%|fS) NrV)rrkrF)r+Z savepoint)Zsenderinstancer+)r+cSsg|] }|jqSr)rV)r;rLrrrrWsz$Collector.delete..cSsg|] }|jqSr)rV)r;rLrrrrWs)&rBrSsortedrrrrpr{rjr Zmark_for_rollback_on_errorr+r Z DeleteQueryZ delete_batchrVsetattrrNrxlabelZatomicrr:r r\sendrDr_r#rCZ UpdateQueryZ update_batchr$rYreverser]sumdict) rr#rKZdeleted_counterrcountrLrZZinstances_for_fieldvaluesr)r1r~Zpk_listrrrdelete{sZ  "     $   zCollector.delete)NFF)F)N)NFTNFFT)rrrrrMr/r'r.rUr[r^rjrtr!r}rrrrrrrr@Os$    $   r@)r collectionsrr functoolsrr itertoolsrrZ django.dbrr r Zdjango.db.modelsr r r rrr,r-r0r5r6r8r9r?r@rrrrs"