o {,h@s|dZddlmZddlmZmZmZddlmZgdZ GdddeZ Gdd d eZ Gd d d eZ Gd d d eZ dS)zR Query subclasses which provide extra functionality beyond simple data retrieval. ) FieldError)GET_ITERATOR_CHUNK_SIZE NO_RESULTS ROW_COUNT)Query) DeleteQuery UpdateQuery InsertQueryAggregateQueryc@s$eZdZdZdZddZddZdS)rzA DELETE SQL query.SQLDeleteCompilercCs&||j|i|_||_||tSN) alias_mapwhere get_compiler execute_sqlr)selftablerusingrU/var/www/html/rh/venv/lib/python3.10/site-packages/django/db/models/sql/subqueries.pydo_queryszDeleteQuery.do_querycCsld}|j}tdt|tD]$}|||jd|||t||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)r) get_metapkrangelenr clear_where add_filterattnamerdb_tabler)rpk_listr num_deletedfieldoffsetrrr delete_batchs   zDeleteQuery.delete_batchN)__name__ __module__ __qualname____doc__compilerrr$rrrrrs  rcs`eZdZdZdZfddZddZfddZd d Zd d Z d dZ ddZ ddZ Z S)rzAn UPDATE SQL query.SQLUpdateCompilercstj|i||dSr )super__init__ _setup_query)rargskwargs __class__rrr,5s 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)values related_idsrelated_updates)rrrrr-9s zUpdateQuery._setup_querycst}|j|_|Sr )r+cloner4copy)robjr0rrr5Bs  zUpdateQuery.clonecCsT||tdt|tD]}||d|||t||tq dS)Nrpk__in) add_update_valuesrrrrrrrr)rr r2rr#rrr update_batchGs zUpdateQuery.update_batchcCsg}|D]N\}}||}|jo|j p|j }|jjj}|jdkr/|jj r/t d|r7|j r=|j r=t d|||jurL| |||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. rz:Composite primary key fields must be updated individually.zMCannot update model field %r (only non-relations and foreign keys permitted).)itemsr get_field auto_createdconcretemodel_metaconcrete_modelnameis_composite_pkr is_relation many_to_manyadd_related_updateappendadd_update_fields)rr2 values_seqrBvalr"directr?rrrr9Ps*  zUpdateQuery.add_update_valuescCsH|D]\}}}|jr qt|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) generatedhasattrrLr2rG)rrIr"r?rJrrrrHls 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)r4 setdefaultrG)rr?r"valuerrrrF{szUpdateQuery.add_related_updatecCsZ|jsgSg}|jD]\}}t|}||_|jdur%|d|j|||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. Nr8)r4r;rr2r3rrG)rresultr?r2queryrrrget_related_updatess  zUpdateQuery.get_related_updates)r%r&r'r(r)r,r-r5r:r9rHrFrU __classcell__rrr0rr0s   rcs4eZdZdZddddfdd Zd ddZZS) r SQLInsertCompilerN) on_conflict update_fields unique_fieldscs<tj|i|g|_g|_||_|pg|_|pg|_dSr )r+r,fieldsobjsrXrYrZ)rrXrYrZr.r/r0rrr,s  zInsertQuery.__init__FcCs||_||_||_dSr )r[r\raw)rr[r\r]rrr insert_valuess zInsertQuery.insert_values)F)r%r&r'r)r,r^rVrrr0rr s  r cs$eZdZdZdZfddZZS)r zu Take another query as a parameter to the FROM clause and only select the elements in the provided list. SQLAggregateCompilercs||_t|dSr ) inner_queryr+r,)rr?r`r0rrr,szAggregateQuery.__init__)r%r&r'r(r)r,rVrrr0rr sr N)r(django.core.exceptionsrdjango.db.models.sql.constantsrrrdjango.db.models.sql.queryr__all__rrr r rrrrs   e