o
    ^"<fp  ã                   @   sˆ   d Z ddlmZ ddlmZ ddlmZmZmZ ddl	m
Z
 g d¢ZG dd„ de
ƒZG d	d
„ d
e
ƒZG dd„ de
ƒZG dd„ de
ƒZdS )zR
Query subclasses which provide extra functionality beyond simple data retrieval.
é    )Ú
FieldError)ÚQ)ÚCURSORÚGET_ITERATOR_CHUNK_SIZEÚ
NO_RESULTS)ÚQuery)ÚDeleteQueryÚUpdateQueryÚInsertQueryÚAggregateQueryc                   @   s$   e Zd ZdZdZdd„ Zdd„ ZdS )r   zA DELETE SQL query.ÚSQLDeleteCompilerc                 C   sX   || j | i| _ || _|  |¡ t¡}|r*| |jW  d   ƒ S 1 s%w   Y  dS )Nr   )Ú	alias_mapÚwhereÚget_compilerÚexecute_sqlr   Úrowcount)ÚselfÚtabler   ÚusingÚcursor© r   úV/var/www/html/kck/venv/lib/python3.10/site-packages/django/db/models/sql/subqueries.pyÚdo_query   s    ÿzDeleteQuery.do_queryc                 C   sx   d}|   ¡ j}tdt|ƒtƒD ]*}|  ¡ | _|  tdi |j	d |||t … i¤Ž¡ || j
|   ¡ j| j|d7 }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   Nr   )Úget_metaÚpkÚrangeÚlenr   Úwhere_classr   Úadd_qr   Úattnamer   Údb_table)r   Úpk_listr   Únum_deletedÚfieldÚoffsetr   r   r   Údelete_batch   s   


ÿzDeleteQuery.delete_batchN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__Úcompilerr   r&   r   r   r   r   r      s
    	r   c                       s`   e Zd ZdZdZ‡ fdd„Zdd„ Z‡ fdd„Zd	d
„ Zdd„ Z	dd„ Z
dd„ Zdd„ Z‡  ZS )r	   zAn UPDATE SQL query.ÚSQLUpdateCompilerc                    s   t ƒ j|i |¤Ž |  ¡  d S ©N)ÚsuperÚ__init__Ú_setup_query)r   ÚargsÚkwargs©Ú	__class__r   r   r/   4   s   zUpdateQuery.__init__c                 C   s   g | _ d| _i | _dS )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_idsÚrelated_updates)r   r   r   r   r0   8   s   
zUpdateQuery._setup_queryc                    s   t ƒ  ¡ }| j ¡ |_|S r-   )r.   Úcloner7   Úcopy)r   Úobjr3   r   r   r8   A   s   
zUpdateQuery.clonec                 C   sZ   |   |¡ tdt|ƒtƒD ]}|  ¡ | _|  t|||t … d¡ |  |¡ 	t
¡ qd S )Nr   )Úpk__in)Úadd_update_valuesr   r   r   r   r   r   r   r   r   r   )r   r"   r5   r   r%   r   r   r   Úupdate_batchF   s   

ýzUpdateQuery.update_batchc                 C   sš   g }|  ¡ D ]A\}}|  ¡  |¡}|jo|j  p|j }|jjj}|r*|jr0|j	r0t
d| ƒ‚||  ¡ jur?|  |||¡ q| |||f¡ q|  |¡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_createdÚconcreteÚmodelÚ_metaÚconcrete_modelÚis_relationÚmany_to_manyr   Úadd_related_updateÚappendÚadd_update_fields)r   r5   Ú
values_seqÚnameÚvalr$   ÚdirectrB   r   r   r   r<   M   s    
ÿÿ
zUpdateQuery.add_update_valuesc                 C   s@   |D ]\}}}t |dƒr|j| ddd}| j |||f¡ qdS )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_joinsÚfor_saveN)ÚhasattrrN   r5   rH   )r   rJ   r$   rB   rL   r   r   r   rI   c   s
   
üzUpdateQuery.add_update_fieldsc                 C   s   | j  |g ¡ |d|f¡ dS )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)r7   Ú
setdefaultrH   )r   rB   r$   Úvaluer   r   r   rG   o   s   zUpdateQuery.add_related_updatec                 C   sX   | j sg S g }| j  ¡ D ]\}}t|ƒ}||_| jdur$| d| jf¡ | |¡ 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.
        Nr;   )r7   r>   r	   r5   r6   Ú
add_filterrH   )r   ÚresultrB   r5   Úqueryr   r   r   Úget_related_updatesw   s   
zUpdateQuery.get_related_updates)r'   r(   r)   r*   r+   r/   r0   r8   r=   r<   rI   rG   rW   Ú__classcell__r   r   r3   r   r	   /   s    	r	   c                       s0   e Zd ZdZddœ‡ fdd„
Zddd„Z‡  ZS )	r
   ÚSQLInsertCompilerF)Úignore_conflictsc                   s(   t ƒ j|i |¤Ž g | _g | _|| _d S r-   )r.   r/   ÚfieldsÚobjsrZ   )r   rZ   r1   r2   r3   r   r   r/   Œ   s   
zInsertQuery.__init__c                 C   s   || _ || _|| _d S r-   )r[   r\   Úraw)r   r[   r\   r]   r   r   r   Úinsert_values’   s   
zInsertQuery.insert_values)F)r'   r(   r)   r+   r/   r^   rX   r   r   r3   r   r
   ‰   s    r
   c                       s$   e Zd ZdZdZ‡ fdd„Z‡  ZS )r   zu
    Take another query as a parameter to the FROM clause and only select the
    elements in the provided list.
    ÚSQLAggregateCompilerc                    s   || _ tƒ  |¡ d S r-   )Úinner_queryr.   r/   )r   rB   r`   r3   r   r   r/       s   zAggregateQuery.__init__)r'   r(   r)   r*   r+   r/   rX   r   r   r3   r   r   ˜   s    r   N)r*   Údjango.core.exceptionsr   Údjango.db.models.query_utilsr   Údjango.db.models.sql.constantsr   r   r   Údjango.db.models.sql.queryr   Ú__all__r   r	   r
   r   r   r   r   r   Ú<module>   s     Z