o uvXjp@sdZddlmZddlmZddlmZddlmZm Z m Z ddl m Z gdZ Gdd d e ZGd d d e ZGd d d e ZGddde 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.SQLDeleteCompilercCs4||j|i|_||_||t}|r|jSdS)Nr) alias_mapwhere get_compiler execute_sqlrrowcount)selftablerusingcursorr\/var/www/html/myproject/venv/lib/python3.10/site-packages/django/db/models/sql/subqueries.pydo_queryszDeleteQuery.do_queryc Csxd}|j}tdt|tD]*}||_|tdi|j d|||ti||j |j |j|d7}q|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. r__in)rNr) get_metapkrangelenr where_classradd_qrattnamerdb_table)rpk_listr num_deletedfieldoffsetrrr delete_batchs   zDeleteQuery.delete_batchcs|j|tfddjD}|r |t|jkr%j|_n:|jjj}t|j j sCt |j ddd}|s=dS| ||S||g_}||_|t|d||t}|rl|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).0tinnerqrr 7sz)DeleteQuery.delete_qs..rT)flatrpk__in)queryget_initial_aliastuplerrmodel_metarrfeaturesupdate_can_self_selectlist values_listr'clear_select_clauseget_colselectrr rrrrr)rr1rinnerq_used_tablesrvaluesrrr+r delete_qs,s(      zDeleteQuery.delete_qsN)__name__ __module__ __qualname____doc__compilerrr'r?rrrrr s  r cs`eZdZdZdZfddZddZfddZd d Zd d Z d dZ ddZ ddZ Z S)r zAn UPDATE SQL query.SQLUpdateCompilercstj|i||dSN)super__init__ _setup_query)rargskwargs __class__rrrHTs zUpdateQuery.__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)rrrrrIXs zUpdateQuery._setup_querycst}|j|_|SrF)rGclonerOcopy)robjrLrrrPas  zUpdateQuery.clonecCsZ||tdt|tD]}||_|t|||td|| t q dS)Nrr/) add_update_valuesrrrrrr rrrr)rr#r>rr&rrr update_batchfs  zUpdateQuery.update_batchcCsg}|D]A\}}||}|jo|j p|j }|jjj}|r*|jr0|j r0t d|||jur?| |||q| |||fq| |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_createdconcreter4r5concrete_model is_relation many_to_manyradd_related_updateappendadd_update_fields)rr> values_seqnamevalr%directr4rrrrSms   zUpdateQuery.add_update_valuescCs@|D]\}}}t|dr|j|ddd}|j|||fqdS)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)hasattrrcr>r])rr_r%r4rarrrr^s  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)rO setdefaultr])rr4r%valuerrrr\szUpdateQuery.add_related_updatecCsX|jsgSg}|jD]\}}t|}||_|jdur$|d|jf||q |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. Nr0)rOrUr r>rN add_filterr])rresultr4r>r1rrrget_related_updatess  zUpdateQuery.get_related_updates)r@rArBrCrDrHrIrPrTrSr^r\rk __classcell__rrrLrr Os   r cs0eZdZdZddfdd ZdddZZS) r SQLInsertCompilerF)ignore_conflictscs(tj|i|g|_g|_||_dSrF)rGrHfieldsobjsrn)rrnrJrKrLrrrHs zInsertQuery.__init__cCs||_||_||_dSrF)rorpraw)rrorprqrrr insert_valuess zInsertQuery.insert_values)F)r@rArBrDrHrrrlrrrLrr sr c@seZdZdZdZddZdS)r zu Take another query as a parameter to the FROM clause and only select the elements in the provided list. SQLAggregateCompilercCs$d|_||jdd\|_|_dS)NT)with_col_aliases)subqueryras_sql sub_params)rr1rrrr add_subqueryszAggregateQuery.add_subqueryN)r@rArBrCrDrxrrrrr s r N)rCdjango.core.exceptionsr django.dbrdjango.db.models.query_utilsrdjango.db.models.sql.constantsrrrdjango.db.models.sql.queryr__all__r r r r rrrrs    ?Z