ó ø¢TQc@sSdZddlmZddlmZddlmZddlmZm Z ddl Tddl m Z ddl mZdd lmZmZdd lmZdd lmZdd lmZdd lmZdddddgZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdS(sR Query subclasses which provide extra functionality beyond simple data retrieval. iÿÿÿÿ(t FieldError(t connections(t LOOKUP_SEP(t DateFieldtFieldDoesNotExist(t*(tDate(tQuery(tANDt Constraint(t SortedDict(tPromise(t force_text(tsixt DeleteQueryt UpdateQueryt InsertQueryt DateQuerytAggregateQuerycBs2eZdZdZd„Zdd„Zd„ZRS(so Delete queries are done through this class, since they are more constrained than general queries. tSQLDeleteCompilercCs/|g|_||_|j|ƒjdƒdS(N(ttablestwheret get_compilert execute_sqltNone(tselfttableRtusing((s8../Django//lib/python/django/db/models/sql/subqueries.pytdo_querys  cCs|s|jjj}nx~tdt|ƒtƒD]d}|jƒ}|jtd|j |ƒd|||t!ft ƒ|j |jjj |d|ƒq1WdS(sº 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. itinRN(tmodelt_metatpktrangetlentGET_ITERATOR_CHUNK_SIZEt where_classtaddR RtcolumnRRtdb_table(Rtpk_listRtfieldtoffsetR((s8../Django//lib/python/django/db/models/sql/subqueries.pyt delete_batch#s c Csb|j}|jƒ|jƒg|jD]}|j|r'|^q'}| s\||jkr{t|jƒ r{|j|_nÍ|jjj }t |j j sÓt |jddtƒƒ}|s¿dS|j||ƒdS|jƒ|jƒ|jfgdg|_|_|}|jƒ}|jtd|j|ƒd|ftƒ||_|j|ƒjdƒdS(sÌ 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. R tflatNR(tquerytget_initial_aliasRtalias_refcountR"thavingRRRR Rtfeaturestupdate_can_self_selecttlistt values_listtTrueR+tclear_select_clauseR&Rtselectt select_fieldsR$R%R RRR( RR-Rtinnerqtttinnerq_used_tablesR tvaluesR((s8../Django//lib/python/django/db/models/sql/subqueries.pyt delete_qs2s,    ( ( N(t__name__t __module__t__doc__tcompilerRRR+R=(((s8../Django//lib/python/django/db/models/sql/subqueries.pyRs   cBs_eZdZdZd„Zd„Zd d„Zd„Zd„Z d„Z d„Z d „Z RS( s+ Represents an "update" SQL query. tSQLUpdateCompilercOs'tt|ƒj||Ž|jƒdS(N(tsuperRt__init__t _setup_query(Rtargstkwargs((s8../Django//lib/python/django/db/models/sql/subqueries.pyRD^scCs1g|_d|_t|dƒs-i|_ndS(sÐ Runs on initialization and after cloning. Any attributes that would normally be set in __init__ should go in here, instead, so that they are also set up after a clone() call. trelated_updatesN(R<Rt related_idsthasattrRH(R((s8../Django//lib/python/django/db/models/sql/subqueries.pyREbs  cKs(tt|ƒj|d|jjƒ|S(NRH(RCRtcloneRHtcopy(RtklassRG((s8../Django//lib/python/django/db/models/sql/subqueries.pyRKmscCsž|jjj}|j|ƒx{tdt|ƒtƒD]a}|jƒ|_|jj t d|j |ƒd|||t!ft ƒ|j|ƒjdƒq5WdS(NiR(RRR tadd_update_valuesR!R"R#R$RR%R RR&RRR(RR(R<Rtpk_fieldR*((s8../Django//lib/python/django/db/models/sql/subqueries.pyt update_batchqs c Cs©g}x“tj|ƒD]‚\}}|jjj|ƒ\}}}}| sP|rctd|ƒ‚n|r‚|j|||ƒqn|j|||fƒqW|j|ƒS(s± 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. sMCannot update model field %r (only non-relations and foreign keys permitted).( R t iteritemsRRtget_field_by_nameRtadd_related_updatetappendtadd_update_fields( RR<t values_seqtnametvalR)Rtdirecttm2m((s8../Django//lib/python/django/db/models/sql/subqueries.pyRN{s! cCsag|D]@}t|dtƒrA|d|dt|dƒfn|^q}|jj|ƒdS(sº Turn a sequence of (field, model, value) triples into an update query. Used by add_update_values() as well as the "fast" update path when saving models. iiiN(t isinstanceR R R<textend(RRVtvalue((s8../Django//lib/python/django/db/models/sql/subqueries.pyRUŒsJcCsRy!|j|j|d|fƒWn*tk rM|d|fg|j|R?R@RARDRERRKRPRNRURSRb(((s8../Django//lib/python/django/db/models/sql/subqueries.pyRWs    cBs/eZdZd„Zdd„Zed„ZRS(tSQLInsertCompilercOs/tt|ƒj||Žg|_g|_dS(N(RCRRDtfieldstobjs(RRFRG((s8../Django//lib/python/django/db/models/sql/subqueries.pyRD·s cKsLi|jd6|jd6|jd6}|j|ƒtt|ƒj||S(NRdRetraw(RdReRftupdateRCRRK(RRMRGtextras((s8../Django//lib/python/django/db/models/sql/subqueries.pyRK¼s     cCs~||_x\|D]T}xK|D]C}t||jƒ}t|tƒrt||jt|ƒƒqqWqW||_||_dS(s¢ Set up the insert query from the 'insert_values' dictionary. The dictionary gives the model field names and their target values. If 'raw_values' is True, the values in the 'insert_values' dictionary are inserted directly into the query, rather than passed as SQL parameters. This provides a way to insert NULL and DEFAULT keywords into the query, for example. N( RdtgetattrtattnameR[R tsetattrR ReRf(RRdReRfR)tobjR]((s8../Django//lib/python/django/db/models/sql/subqueries.pyt insert_valuesÅs   $ N(R>R?RARDRRKtFalseRm(((s8../Django//lib/python/django/db/models/sql/subqueries.pyR´s  cBs eZdZdZdd„ZRS(sã A DateQuery is a normal query, except that it specifically selects a single date field. This requires some special handling when converting the results back to Python objects, so we put it in a separate class. tSQLDateCompilertASCcCs*y1|j|jtƒ|jƒ|jƒtƒ}Wn0tk rctd|jj j |fƒ‚nX|d}t |t ƒst d|jƒ‚|dd}t||jf|ƒ}|jƒ|gd g|_|_t|_|dkr÷dgpýdg|_|jr&|jd|tfƒnd S( sB Converts the query into a date extraction query. s%s has no field named '%s'is%r isn't a DateField.iiÿÿÿÿRpis %s__isnullN(t setup_joinstsplitRtget_metaR.RnRRRRt object_nameR[RtAssertionErrorRWRR&R6RR7R8R5tdistincttorder_bytnullR`(Rt field_namet lookup_typetorderRaR)taliasR7((s8../Django//lib/python/django/db/models/sql/subqueries.pytadd_date_selectâs(         ! (R>R?R@RAR}(((s8../Django//lib/python/django/db/models/sql/subqueries.pyRÙscBseZdZdZd„ZRS(s‰ An AggregateQuery takes another query as a parameter to the FROM clause and only selects the elements in the provided list. tSQLAggregateCompilercCs+|j|ƒjdtƒ\|_|_dS(Ntwith_col_aliases(Rtas_sqlR5tsubqueryt sub_params(RR-R((s8../Django//lib/python/django/db/models/sql/subqueries.pyt add_subquerys(R>R?R@RARƒ(((s8../Django//lib/python/django/db/models/sql/subqueries.pyRþsN( R@tdjango.core.exceptionsRt django.dbRtdjango.db.models.constantsRtdjango.db.models.fieldsRRtdjango.db.models.sql.constantst#django.db.models.sql.datastructuresRtdjango.db.models.sql.queryRtdjango.db.models.sql.whereRR tdjango.utils.datastructuresR tdjango.utils.functionalR tdjango.utils.encodingR t django.utilsR t__all__RRRRR(((s8../Django//lib/python/django/db/models/sql/subqueries.pyts&   A]%%