v^p@sdZddlmZddlmZddlmZddlmZm Z m Z ddl m Z ddd d gZ Gd dde ZGd dde ZGd d d e ZGdd d e ZdS)zR Query subclasses which provide extra functionality beyond simple data retrieval. ) FieldError) connections)Q)CURSORGET_ITERATOR_CHUNK_SIZE NO_RESULTS)Query DeleteQuery UpdateQuery InsertQueryAggregateQueryc@s@eZdZdZdZddZddZddZd S) r zA DELETE SQL query.ZSQLDeleteCompilercCsH||j|i|_||_|j|jt}|rD|jSdS)Nr) alias_mapwhere get_compiler execute_sqlrrowcount)selftablerusingcursorrA/tmp/pip-build-8lau8j11/django/django/db/models/sql/subqueries.pydo_querys zDeleteQuery.do_querycCsd}|jj}xtdt|tD]k}|j|_|jt|j d|||ti||j |jj |jd|7}q.W|S)z Set up and execute delete queries for all the objects in pk_list. More than one physical query may be executed if there are a lot of values in pk_list. rZ__inr) get_metapkrangelenr where_classradd_qrZattnamerZdb_table)rpk_listrZ num_deletedfieldoffsetrrr delete_batchs %,zDeleteQuery.delete_batchcs7|jj|jtfddjD}| s[|t|jkrjj|_n|jjj}t|j j st |j ddd}|sdS|j ||Sj|j|jg_}|j|_|jtd||j|jt}|r3|jSdS)z Delete the queryset in one SQL query (if possible). For simple queries this is done by copying the query.query.where to self.query, for complex queries by using subquery. cs#g|]}j|r|qSr)Zalias_refcount).0t)innerqrr 7s z)DeleteQuery.delete_qs..rZflatTrpk__in)queryZget_initial_aliastupler rmodel_metarrfeaturesZupdate_can_self_selectlistZ values_listr"Zclear_select_clauseZget_colselectrrrrrrr)rr(rZinnerq_used_tablesrvaluesrr)r%r delete_qs,s&   " zDeleteQuery.delete_qsN)__name__ __module__ __qualname____doc__compilerrr"r0rrrrr s   cseZdZdZdZfddZddZfddZd d Zd d Z d dZ ddZ ddZ S)r zAn UPDATE SQL query.ZSQLUpdateCompilercs!tj|||jdS)N)super__init__ _setup_query)rargskwargs) __class__rrr7TszUpdateQuery.__init__cCsg|_d|_i|_dS)z Run on initialization and at the end of chaining. Any attributes that would normally be set in __init__() should go here instead. N)r/ related_idsrelated_updates)rrrrr8Xs  zUpdateQuery._setup_querycs%tj}|jj|_|S)N)r6cloner=copy)robj)r;rrr>aszUpdateQuery.clonecCs}|j|xitdt|tD]O}|j|_|jtd|||t|j|j t q&WdS)Nrr') add_update_valuesrrrrrrrrrr)rrr/rr!rrr update_batchfs  $zUpdateQuery.update_batchcCsg}x|jD]\}}|jj|}|joD|j pO|j }|jjj}| sz|jr|j rt d|||jjk r|j |||q|j |||fqW|j |S)z Convert a dictionary of field name to value mappings into an update query. This is the entry point for the public update() method on querysets. zMCannot update model field %r (only non-relations and foreign keys permitted).)itemsr get_fieldZ auto_createdZconcreter*r+Zconcrete_modelZ is_relationZ many_to_manyradd_related_updateappendadd_update_fields)rr/ values_seqnamevalr directr*rrrrAms zUpdateQuery.add_update_valuescCsaxZ|D]R\}}}t|dr@|j|dddd}|jj|||fqWdS)z Append a sequence of (field, model, value) triples to the internal list that will be used to generate the UPDATE query. Might be more usefully called add_update_targets() to hint at the extra information here. resolve_expressionZ allow_joinsFZfor_saveTN)hasattrrLr/rF)rrHr r*rJrrrrGszUpdateQuery.add_update_fieldscCs)|jj|gj|d|fdS)z Add (name, value) to an update query for an ancestor model. Update are coalesced so that only one update query per ancestor is run. N)r= setdefaultrF)rr*r valuerrrrEszUpdateQuery.add_related_updatecCs~|js gSg}xd|jjD]S\}}t|}||_|jdk ri|jd|jf|j|q#W|S)z Return a list of query objects: one for each update required to an ancestor model. Each query will have the same filtering conditions as the current query but will only update a single table. Nr')r=rCr r/r<Z add_filterrF)rresultr*r/r(rrrget_related_updatess   zUpdateQuery.get_related_updates) r1r2r3r4r5r7r8r>rBrArGrErQrr)r;rr Os     cs=eZdZdZddfddZdddZS)r ZSQLInsertCompilerignore_conflictsFcs2tj||g|_g|_||_dS)N)r6r7fieldsobjsrR)rrRr9r:)r;rrr7s  zInsertQuery.__init__cCs||_||_||_dS)N)rSrTraw)rrSrTrUrrr insert_valuess  zInsertQuery.insert_values)r1r2r3r5r7rVrr)r;rr s c@s(eZdZdZdZddZdS)r zu Take another query as a parameter to the FROM clause and only select the elements in the provided list. ZSQLAggregateCompilercCs4d|_|j|jdd\|_|_dS)NTZwith_col_aliases)ZsubqueryrZas_sqlZ sub_params)rr(rrrr add_subquerys zAggregateQuery.add_subqueryN)r1r2r3r4r5rWrrrrr s N)r4Zdjango.core.exceptionsrZ django.dbrZdjango.db.models.query_utilsrZdjango.db.models.sql.constantsrrrZdjango.db.models.sql.queryr__all__r r r r rrrrs?Z