X5@sddlmZmZddlmZddlmZmZmZddl m Z m Z ddl m Z GdddeZdd Zd d Zd d ZddZddZddZddZGdddeZdS))Counter OrderedDict) attrgetter)IntegrityError connections transaction)signalssql)sixcs"eZdZfddZS)ProtectedErrorcs&||_tt|j||dS)N)protected_objectssuperr __init__)selfmsgr ) __class__G/home/ubuntu/projects/ifolica/build/django/django/db/models/deletion.pyr s zProtectedError.__init__)__name__ __module__ __qualname__rrr)rrr s r cCs_|j|d|jjd|jd|j|jr[t|jj r[|j|d|ndS)Nsource source_attrnullable) collect remote_fieldmodelnamenullrfeaturesZcan_defer_constraint_checksadd_field_update) collectorfieldsub_objsusingrrrCASCADEsr%cCs6td|jjj|djj|jf|dS)NzoCannot delete some instances of model '%s' because they are referenced through a protected foreign key: '%s.%s'r)r rrrrr)r!r"r#r$rrrPROTECTs#r&csLtr!fdd}nfdd}fdd|_|S)Ncs|j||dS)N)r )r!r"r#r$)valuerr set_on_delete"szSET..set_on_deletecs|j||dS)N)r )r!r"r#r$)r'rrr(%scsdfifS)Nzdjango.db.models.SETrr)r'rr'szSET..)callableZ deconstruct)r'r(r)r'rSET s  r+cCs|j|d|dS)N)r )r!r"r#r$rrrSET_NULL+sr,cCs|j||j|dS)N)r Z get_default)r!r"r#r$rrr SET_DEFAULT/sr-cCsdS)Nr)r!r"r#r$rrr DO_NOTHING3sr.cCsdd|jddDS)Ncss:|]0}|jr|j r|js.|jr|VqdS)N) auto_createdZconcreteZ one_to_oneZ one_to_many).0frrr ;sz4get_candidate_relations_to_delete..Zinclude_hiddenT)Z get_fields)optsrrr!get_candidate_relations_to_delete7sr4c@seZdZddZdddddZddZdd d Zd d Zddd dddddZddZ ddZ ddZ ddZ dS) CollectorcCs4||_t|_i|_g|_i|_dS)N)r$rdata field_updates fast_deletes dependencies)rr$rrrrAs     zCollector.__init__NFc Cs|s gSg}|dj}|jj|t}x*|D]"}||kr<|j|q<q<W|j||dk r| r|r||}}n|jj|jjtj |jjn|S)a9 Adds '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. Returns a list of all objects that were not already collected. rN) rr6 setdefaultsetappendupdater9_metaconcrete_modeladd) robjsrrreverse_dependencynew_objsr instancesobjrrrr@Qs     %z Collector.addcCsL|s dS|dj}|jj|ij||ftj|dS)z Schedules a field update. 'objs' must be a homogeneous iterable collection of model instances (e.g. a QuerySet). Nr)rr7r:r;r=)rr"r'rArrrrr ls    zCollector.add_field_updatecsrjjtk rdSt|do7t|ds>dS|j}tjj|s}tjj|s}tj j|rdS|j }t fdd|j j j jDrdSx-t|D]}|jjjtk rdSqWx'|j jD]}t|drdSqWdS)a Determines if the objects in the given queryset-like 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. Allows also skipping parent -> child -> parent chain preventing fast delete of the child. Fr _raw_deletec3s|]}|kVqdS)Nr)r0link) from_fieldrrr2sz,Collector.can_fast_delete..bulk_related_objectsT)r on_deleter%hasattrrr pre_deleteZ has_listeners post_deleteZ m2m_changedr>anyr?parentsvaluesr4r"r.private_fields)rrArHrr3relatedr"r)rHrcan_fast_deletexs&   .zCollector.can_fast_deletecsttt|jjj|jgdtkrifddtdtDSgSdS)zU Returns the objs in suitably sized batches for the used connection. cs$g|]}||qSrr)r0i)conn_batch_sizerArr s z-Collector.get_del_batches..rN)maxrr$opsZbulk_batch_sizerlenrange)rrAr"r)rVrArget_del_batchess (zCollector.get_del_batchesTc s|j|r#|jj|dS|j|||d|}|sHdS|dj} |s| jj} xmtj| jj D]Sr}fdd|D} |j | d| dj j dd dd q}q}Wn|rxt | jD]} | j} | j jtkrqn|j|| }xm|D]e}|j| |}|j|d | rm|jj|q-|r-| j j|| ||jq-q-WqWxW| jjD]F} t| d r| j||j}|j |d| d d qqWndS)a] Adds '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. NrBrcs"g|]}t|jqSr)getattrr)r0rE)ptrrrrWs z%Collector.collect..rrcollect_relatedFTrHrIr)rSr8r<r@rr>r?r itervaluesrOrrZ related_namer4r"rJr.r\related_objectsr$rQrKrI)rrArrr_rrBZ keep_parentsrCrr?Z parent_objsrRr"Zbatchesbatchr#r)r^rrsB      'zCollector.collectcCs0|jjj|jji|d|jj6S)z^ Gets a QuerySet of objects related to ``objs`` via the relation ``related``. z%s__in)Z related_modelZ _base_managerr$filterr"r)rrRrArrrraszCollector.related_objectsccsCx<tj|jD](\}}x|D]}||fVq&WqWdS)N)r iteritemsr6)rrrDrErrrinstances_with_models zCollector.instances_with_modelcsg}t}tj}xt|t|krd}xv|D]n}||kr^qFnjj|jj}|o|j|sF|j ||j |jjd}qFqFW|s!dSq!Wt fdd|D_dS)NFTc3s"|]}|j|fVqdS)N)r6)r0r)rrrr2sz!Collector.sort..) r;listr6rZr9getr>r? differencer<r@r)rZ sorted_modelsZconcrete_modelsmodelsfoundrr9r)rrsorts"     zCollector.sortc Cs4x<|jjD]+\}}t|dtd|j|.cSsg|]}|jqSr)rm)r0rErrrrW-s )%r6itemssortedrrkrrZatomicr$rer>r/rrLsendr8rFrlabelr rdr7r Z UpdateQueryZ update_batchrr`reverseZ DeleteQueryZ delete_batchrMsetattrZattnamermsumrPdict) rrrDZdeleted_counterrEqscountZinstances_for_fieldvaluesqueryr"r'Zpk_listrnrrrdeletesH#    "   '"  !zCollector.delete) rrrrr@r rSr\rrarerkrzrrrrr5@s   $ =   r5N) collectionsrroperatorrZ django.dbrrrZdjango.db.modelsrr Z django.utilsr r r%r&r+r,r-r.r4objectr5rrrrs