3 7^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.ZSQLDeleteCompilercCs4||j|i|_||_|j|jt}|r0|jSdS)Nr) alias_mapwhere get_compiler execute_sqlrrowcount)selftablerusingcursorrA/tmp/pip-build-8app2_gc/Django/django/db/models/sql/subqueries.pydo_queryszDeleteQuery.do_queryc Csxd}|jj}xdtdt|tD]P}|j|_|jtf|j d|||ti||j |jj |j|d7}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__in)r) get_metapkrangelenr where_classradd_qrZattnamerZdb_table)rpk_listrZ num_deletedfieldoffsetrrr delete_batchs   zDeleteQuery.delete_batchcs|jj|jtfddjD}| sB|t|jkrLj|_nt|jjj}t|j j st |j ddd}|s|dS|j ||Sj|j|jg_}|j|_|jt|d|j|jt}|r|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. csg|]}j|r|qSr)Zalias_refcount).0t)innerqrr 7sz)DeleteQuery.delete_qs..rT)Zflatr)pk__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 cs`eZdZdZdZfddZddZfddZd d Zd d Z d dZ ddZ ddZ Z S)r zAn UPDATE SQL query.ZSQLUpdateCompilercstj|||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)rrrrr8XszUpdateQuery._setup_querycstj}|jj|_|S)N)r6cloner=copy)robj)r;rrr>as  zUpdateQuery.clonecCs^|j|xNtdt|tD]:}|j|_|jt|||td|j|j t qWdS)Nr)r') add_update_valuesrrrrrrrrrr)rrr/rr!rrr update_batchfs   zUpdateQuery.update_batchcCsg}x|jD]\}}|jj|}|jo0|j p:|j }|jjj}| sX|jrd|j rdt 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_valuescCsDx>|D]6\}}}t|dr*|j|ddd}|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_expressionFT)Z allow_joinsZfor_saveN)hasattrrLr/rF)rrHr r*rJrrrrGs zUpdateQuery.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}xH|jjD]:\}}t|}||_|jdk rJ|jd|jf|j|qW|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>rBrArGrErQ __classcell__rr)r;rr Os   cs0eZdZdZddfdd ZdddZZS) r ZSQLInsertCompilerF)ignore_conflictscs$tj||g|_g|_||_dS)N)r6r7fieldsobjsrS)rrSr9r:)r;rrr7szInsertQuery.__init__cCs||_||_||_dS)N)rTrUraw)rrTrUrVrrr insert_valuesszInsertQuery.insert_values)F)r1r2r3r5r7rWrRrr)r;rr sc@seZdZdZdZddZdS)r zu Take another query as a parameter to the FROM clause and only select the elements in the provided list. ZSQLAggregateCompilercCs$d|_|j|jdd\|_|_dS)NT)Zwith_col_aliases)ZsubqueryrZas_sqlZ sub_params)rr(rrrr add_subqueryszAggregateQuery.add_subqueryN)r1r2r3r4r5rXrrrrr sN)r4Zdjango.core.exceptionsrZ django.dbrZdjango.db.models.query_utilsrZdjango.db.models.sql.constantsrrrZdjango.db.models.sql.queryr__all__r r r r rrrrs     ?Z