3 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.ZSQLDeleteCompilercCs4||j|i|_||_|j|jt}|r0|jSdS)Nr) alias_mapwhere get_compiler execute_sqlrrowcount)selftablerusingcursorrC/usr/lib/python3.6/site-packages/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_qrattnamerdb_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)alias_refcount).0t)innerqrr 7sz)DeleteQuery.delete_qs..rT)flatr)pk__in)queryget_initial_aliastupler rmodel_metarrfeaturesZupdate_can_self_selectlistZ values_listr$clear_select_clauseget_colselectrrrrrrr)rr,rZinnerq_used_tablesrvaluesrr)r(r delete_qs,s&     zDeleteQuery.delete_qsN)__name__ __module__ __qualname____doc__compilerrr$r7rrrrr 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__rrr>TszUpdateQuery.__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)r6 related_idsrelated_updates)rrrrr?XszUpdateQuery._setup_querycstj}|jj|_|S)N)r=clonerDcopy)robj)rBrrrEas  zUpdateQuery.clonecCs^|j|xNtdt|tD]:}|j|_|jt|||td|j|j t qWdS)Nr)r+) add_update_valuesrrrrrrrrrr)rr!r6rr#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_field auto_createdconcreter/r0concrete_model is_relation many_to_manyradd_related_updateappendadd_update_fields)rr6 values_seqnamevalr"directr/rrrrHms 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) allow_joinsfor_saveN)hasattrrXr6rR)rrTr"r/rVrrrrSs 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)rD setdefaultrR)rr/r"valuerrrrQszUpdateQuery.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+)rDrJr r6rC add_filterrR)rresultr/r6r,rrrget_related_updatess zUpdateQuery.get_related_updates)r8r9r:r;r<r>r?rErIrHrSrQr` __classcell__rr)rBrr Os   cs0eZdZdZddfdd ZdddZZS) r ZSQLInsertCompilerF)ignore_conflictscs$tj||g|_g|_||_dS)N)r=r>fieldsobjsrb)rrbr@rA)rBrrr>szInsertQuery.__init__cCs||_||_||_dS)N)rcrdraw)rrcrdrerrr insert_valuesszInsertQuery.insert_values)F)r8r9r:r<r>rfrarr)rBrr 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)subqueryras_sqlZ sub_params)rr,rrrr add_subqueryszAggregateQuery.add_subqueryN)r8r9r:r;r<rirrrrr sN)r;django.core.exceptionsr django.dbrdjango.db.models.query_utilsrdjango.db.models.sql.constantsrrrdjango.db.models.sql.queryr__all__r r r r rrrrs     ?Z