B CVvgp@sdZddlmZddlmZddlmZmZmZddl m Z dddd gZ Gd dde Z Gd dde Z Gd dde ZGd d d e ZdS)zR Query subclasses which provide extra functionality beyond simple data retrieval. ) FieldError)Q)CURSORGET_ITERATOR_CHUNK_SIZE NO_RESULTS)Query DeleteQuery UpdateQuery InsertQueryAggregateQueryc@s$eZdZdZdZddZddZdS)rzA DELETE SQL query.ZSQLDeleteCompilerc Cs@||j|i|_||_||t}|r<||jSQRXdS)Nr)Z alias_mapwhere get_compiler execute_sqlrZrowcount)selftabler usingcursorrC/tmp/pip-install-o3oxmrkh/Django/django/db/models/sql/subqueries.pydo_querys zDeleteQuery.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_classr add_qrZattnamerZdb_table)rpk_listrZ num_deletedfieldoffsetrrr delete_batchs   zDeleteQuery.delete_batchN)__name__ __module__ __qualname____doc__compilerrrrrrrrs 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__rrr&4szUpdateQuery.__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'8szUpdateQuery._setup_querycst}|j|_|S)N)r%cloner-copy)robj)r*rrr.As  zUpdateQuery.clonecCs^||xNtdt|tD]:}||_|t|||td|| t qWdS)Nr)pk__in) add_update_valuesrrrrr rrr rr)rrr+rrrrr 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_createdZconcretemodelZ_metaZconcrete_modelZ is_relationZ many_to_manyradd_related_updateappendadd_update_fields)rr+ values_seqnamevalrdirectr6rrrr2Ms 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)hasattrr>r+r8)rr:rr6r<rrrr9cs 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- setdefaultr8)rr6rvaluerrrr7oszUpdateQuery.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. Nr1)r-r4r r+r,Z add_filterr8)rresultr6r+queryrrrget_related_updatesws zUpdateQuery.get_related_updates)r r!r"r#r$r&r'r.r3r2r9r7rD __classcell__rr)r*rr /s   cs0eZdZdZddfdd ZdddZZS) r ZSQLInsertCompilerF)ignore_conflictscs$tj||g|_g|_||_dS)N)r%r&fieldsobjsrF)rrFr(r))r*rrr&szInsertQuery.__init__cCs||_||_||_dS)N)rGrHraw)rrGrHrIrrr insert_valuesszInsertQuery.insert_values)F)r r!r"r$r&rJrErr)r*rr scs$eZdZdZdZfddZZS)r zu Take another query as a parameter to the FROM clause and only select the elements in the provided list. ZSQLAggregateCompilercs||_t|dS)N) inner_queryr%r&)rr6rK)r*rrr&szAggregateQuery.__init__)r r!r"r#r$r&rErr)r*rr sN)r#Zdjango.core.exceptionsrZdjango.db.models.query_utilsrZdjango.db.models.sql.constantsrrrZdjango.db.models.sql.queryr__all__rr r r rrrrs     Z