B 0H[@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|_||_||t}|r0|jSdS)Nr) alias_mapwhere get_compiler execute_sqlrrowcount)selftablerusingcursorrcC:\Users\HIRONO~1\AppData\Local\Temp\pip-install-6bm3nxem\django\django\db\models\sql\subqueries.pydo_queryszDeleteQuery.do_queryc Csxd}|j}xdtdt|tD]P}||_|tf|j d|||ti||j |j |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|j|tfddjD}|r@|t|jkrJj|_nt|jjj}t|j j st |j ddd}|szdS| ||S||g_}||_|t|d||t}|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|||dS)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_querycst}|j|_|S)N)r6cloner=copy)robj)r;rrr>as  zUpdateQuery.clonecCs^||xNtdt|tD]:}||_|t|||td|| t qWdS)Nr)r') add_update_valuesrrrrrrrrrr)rrr/rr!rrr update_batchfs   zUpdateQuery.update_batchcCsg}x|D]\}}||}|jo0|j p:|j }|jjj}|rV|jrb|j rbt d|||jk r| |||q| |||fqW| |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}|j|||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|j|g|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|jD]:\}}t|}||_|jdk rJ|d|jf||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   cs*eZdZdZfddZdddZZS)r ZSQLInsertCompilercstj||g|_g|_dS)N)r6r7fieldsobjs)rr9r:)r;rrr7szInsertQuery.__init__FcCs||_||_||_dS)N)rSrTraw)rrSrTrUrrr insert_valuesszInsertQuery.insert_values)F)r1r2r3r5r7rVrRrr)r;rr s c@seZdZdZdZddZdS)r zu Take another query as a parameter to the FROM clause and only select the elements in the provided list. ZSQLAggregateCompilercCs$d|_||jdd\|_|_dS)NT)Zwith_col_aliases)ZsubqueryrZas_sqlZ sub_params)rr(rrrr add_subqueryszAggregateQuery.add_subqueryN)r1r2r3r4r5rWrrrrr 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