U b|e/@sdZddlmZddlmZmZmZddlmZddddgZ Gd ddeZ Gd ddeZ Gd ddeZ Gd ddeZ d S)zR Query subclasses which provide extra functionality beyond simple data retrieval. ) FieldError)CURSORGET_ITERATOR_CHUNK_SIZE NO_RESULTS)Query DeleteQuery UpdateQuery InsertQueryAggregateQueryc@s$eZdZdZdZddZddZdS)rzA DELETE SQL query.ZSQLDeleteCompilerc CsL||j|i|_||_||t}|rH||jW5QRSQRXdS)Nr) alias_mapwhere get_compiler execute_sqlrZrowcount)selftabler usingcursorr\/var/www/website-v5/atlas_env/lib/python3.8/site-packages/django/db/models/sql/subqueries.pydo_queryszDeleteQuery.do_querycCsld}|j}tdt|tD]H}|||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. rZ__in)r) get_metapkrangelenr clear_where add_filterattnamerdb_tabler )rpk_listrZ num_deletedfieldoffsetrrr delete_batchs   zDeleteQuery.delete_batchN)__name__ __module__ __qualname____doc__compilerrr!rrrrr s cs`eZdZdZdZfddZddZfddZd d Zd d Z d dZ ddZ ddZ Z S)rzAn UPDATE SQL query.ZSQLUpdateCompilercstj|||dSN)super__init__ _setup_query)rargskwargs __class__rrr)5szUpdateQuery.__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*9szUpdateQuery._setup_querycst}|j|_|Sr')r(cloner1copy)robjr-rrr2Bs  zUpdateQuery.clonecCsT||tdt|tD]4}||d|||t||tqdS)Nrpk__in) add_update_valuesrrrrrr rr)rrr/rr rrr update_batchGs zUpdateQuery.update_batchcCsg}|D]\}}||}|jo.|j p8|j }|jjj}|rT|jr`|j r`t d|||jk r~| |||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_createdconcretemodel_metaconcrete_model is_relation many_to_manyradd_related_updateappendadd_update_fields)rr/ values_seqnamevalrdirectr<rrrr6Ps" zUpdateQuery.add_update_valuescCs@|D]6\}}}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)hasattrrHr/rB)rrDrr<rFrrrrChs 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)r1 setdefaultrB)rr<rvaluerrrrAtszUpdateQuery.add_related_updatecCsZ|js gSg}|jD]<\}}t|}||_|jdk rJ|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. Nr5)r1r8rr/r0rrB)rresultr<r/queryrrrget_related_updates|s  zUpdateQuery.get_related_updates)r"r#r$r%r&r)r*r2r7r6rCrArP __classcell__rrr-rr0s    cs4eZdZdZddddfdd Zd ddZZS) r ZSQLInsertCompilerN) on_conflict update_fields unique_fieldscs8tj||g|_g|_||_|p&g|_|p0g|_dSr')r(r)fieldsobjsrRrSrT)rrRrSrTr+r,r-rrr)s  zInsertQuery.__init__FcCs||_||_||_dSr')rUrVraw)rrUrVrWrrr insert_valuesszInsertQuery.insert_values)F)r"r#r$r&r)rXrQrrr-rr s  cs$eZdZdZdZfddZZS)r zu Take another query as a parameter to the FROM clause and only select the elements in the provided list. ZSQLAggregateCompilercs||_t|dSr') inner_queryr(r))rr<rYr-rrr)szAggregateQuery.__init__)r"r#r$r%r&r)rQrrr-rr sN)r%django.core.exceptionsrdjango.db.models.sql.constantsrrrdjango.db.models.sql.queryr__all__rrr r rrrrs   $^