o
    ޭh%3                    @  s  U d dl mZ d dlZd dlmZ d dlmZmZmZm	Z	m
Z
 d dlmZ d dlmZmZmZmZmZmZmZ d dlmZmZmZ d dlmZ d d	lmZ d d
lmZ d dlmZm Z m!Z!m"Z"m#Z# d dl$m%Z%m&Z& d dl'm(Z(m)Z) d dl*m+Z+m,Z,m-Z-m.Z.m/Z/ d dl0m1Z1m2Z2m3Z3m4Z4m5Z5 d dl6m7Z7m8Z8m9Z9 d dl:m;Z; d dl<m=Z=m>Z>m?Z?m@Z@mAZA d dlBmCZC d dlDmEZE e ZFdeGd< erd dlHmIZI edddZJedddZKedeLdZMG dd  d e&eK ZNG d!d" d"eeJ ZOG d#d$ d$eOeJ ZPG d%d& d&eOZQG d'd( d(eOZRG d)d* d*eOZSG d+d, d,eOZTG d-d. d.eOZUG d/d0 d0eUeeM ZVG d1d2 d2eUeeM ZWG d3d4 d4eOZXG d5d6 d6eQeeJ ZYG d7d8 d8eOeeJ ZZdS )9    )annotationsN)defaultdict)AsyncIteratorCallable
Collection	GeneratorIterable)copy)TYPE_CHECKINGAnyGenericOptionalTypeVarcastoverload)JoinTypeOrderTable)Count)Cast)QueryBuilder)CaseFieldStarTermValueWrapper)LiteralProtocol)BaseDBAsyncClientCapabilities)DoesNotExist
FieldErrorIntegrityErrorMultipleObjectsReturnedParamsError)
ExpressionQRawSQLResolveContextResolveResult)ForeignKeyFieldInstanceOneToOneFieldInstanceRelationalField)FilterInfoDict)PrefetchQueryModifierTableCriterionTupleexpand_lookup_expressionget_joins_for_related_field)router)chunkr   QUERY)ModelMODELr6   )boundT_coT)	covariantSINGLEc                   @  s\   e Zd ZdZd!ddZd"d	d
Zd#ddZd$ddZd%ddZddd&ddZ	d'ddZ
d S )(QuerySetSinglezb
    Awaiting on this will resolve a single instance of the Model object, and not a sequence.
    returnGenerator[Any, None, T_co]c                 C     d S N selfrA   rA   V/var/www/html/stock_analysis/be/venv/lib/python3.10/site-packages/tortoise/queryset.py	__await__;       zQuerySetSingle.__await__argsstr | PrefetchQuerySetSingle[T_co]c                 G  r?   r@   rA   rC   rG   rA   rA   rD   prefetch_related=      zQuerySetSingle.prefetch_relatedstrc                 G  r?   r@   rA   rJ   rA   rA   rD   select_relatedA   rF   zQuerySetSingle.select_relatedkwargsExpression | Termc                 K  r?   r@   rA   rC   rO   rA   rA   rD   annotateC   rL   zQuerySetSingle.annotatefields_for_selectc                 G  r?   r@   rA   )rC   rS   rA   rA   rD   onlyG   rF   zQuerySetSingle.onlyFflatfields_rV   boolValuesListQuery[Literal[True]]c                G  r?   r@   rA   )rC   rV   rW   rA   rA   rD   values_listI   rL   zQuerySetSingle.values_listValuesQuery[Literal[True]]c                 O  r?   r@   rA   rC   rG   rO   rA   rA   rD   valuesM   rL   zQuerySetSingle.valuesN)r=   r>   )rG   rH   r=   rI   )rG   rM   r=   rI   )rO   rP   r=   rI   )rS   rM   r=   rI   )rW   rM   rV   rX   r=   rY   )rG   rM   rO   rM   r=   r[   )__name__
__module____qualname____doc__rE   rK   rN   rR   rT   rZ   r]   rA   rA   rA   rD   r<   5   s    




r<   c                   @  s   e Zd ZdZd6ddZd7d8ddZd7d9ddZd:ddZd;ddZd<ddZ	e
d7d=d!d"Z	#d>d?d+d,Zd@d-d.Zd7dAd/d0Zd:d1d2ZdBd4d5Zd#S )CAwaitableQuery)querymodel_joined_tables_dbcapabilities_annotations_custom_filters
_q_objectsrd   type[MODEL]r=   Nonec                 C  s:   g | _ || _t| _d | _|jjj| _i | _i | _	g | _
d S r@   )re   rd   r5   rc   rf   _metadbrg   rh   ri   rj   rC   rd   rA   rA   rD   __init__^   s   
zAwaitableQuery.__init__F	for_writerX   r   c                 C  s8   | j r| j S |rt| j}nt| j}|p| jjjS )z}
        Return the connection that will be used if this query is executed now.

        :return: BaseDBAsyncClient:
        )rf   r3   db_for_writerd   db_for_readrm   rn   )rC   rq   rn   rA   rA   rD   
_choose_dbh   s   zAwaitableQuery._choose_dbc                 C  s   | j d u r| || _ d S d S r@   )rf   rt   )rC   rq   rA   rA   rD   _choose_db_if_not_chosenv   s   
z'AwaitableQuery._choose_db_if_not_chosenc              
     s      }t } jD ]}||t j jjj j j	dM }q
|j
D ]"}|d  jvrE jj|d tjd|d  _ j|d  q#|j j_|j j_|rp js] jjs] jjrr jj fdd jjjD   _dS dS dS )z)Builds the common filters for a QuerySet.rd   tabler   custom_filtersr   how   c                   s   g | ]	} j jj| qS rA   )rd   rm   	basetable.0fieldrB   rA   rD   
<listcomp>       z2AwaitableQuery.resolve_filters.<locals>.<listcomp>N)_resolve_annotater/   rj   resolver(   rd   rm   r|   rh   ri   joinsre   rc   joinr   
left_outeronappendhaving_criterion_havingswhere_criterion_wheres	_orderbysgroupby	db_fields)rC   has_aggregatemodifiernoder   rA   rB   rD   resolve_filtersz   s.   

	"


zAwaitableQuery.resolve_filtersrw   r   related_field_namerM   related_fieldr,   c                 C  s,   t |||}|D ]}| | q|d d S )Nr   )r2   _join_table)rC   rw   r   r   r   r   rA   rA   rD   _join_table_by_field   s   z#AwaitableQuery._join_table_by_fieldtable_criterio_tupler0   c                 C  sH   |d | j vr"| jj|d tjd|d | _| j |d  d S d S )Nr   ry   r{   )re   rc   r   r   r   r   r   )rC   r   rA   rA   rD   r      s   zAwaitableQuery._join_tableorderingreversetuple[str, Order]c                 C  sL   t j}| d dkr| dd  }t j}n| }|r"|t jkrt jnt j}||fS )Nr   -r{   r   ascdesc)r   r   
order_type
field_namerA   rA   rD   _resolve_ordering_string   s   z'AwaitableQuery._resolve_ordering_stringNtype[Model]	orderings!Iterable[tuple[str, str | Order]]r   dict[str, Term | Expression]rS   Collection[str] | Nonec              	   C  sz  |s| j jjr|s| j jj}|D ]}|d }||jjv r td|d\}}	}
||jjv rMtt|jj| }| 	|||}| 
|j||
|d fgi  q||v r|rW||v r\t|}n|| }t|trh|}n|t| j ||i dj}| jj||d d| _q|jj|}|std| d|j |jp|}|| }||jjjjd	}|r|||}| jj||d d| _qd
S )aS  
        Applies standard ordering to QuerySet.

        :param model: The Model this queryset is based on.
        :param table: ``pypika_tortoise.Table`` to keep track of the virtual SQL table
            (to allow self referential joins)
        :param orderings: What columns/order to order by
        :param annotations:  Annotations that may be ordered on
        :param fields_for_select: Contains fields that are selected in the SELECT clause if
            .only(), .values() or .values_list() are used.

        :raises FieldError: If a field provided does not exist in model.
        r   zNFiltering by relation is not possible. Filter by nested field of related model__r{   rv   )orderUnknown field  for model function_castN)rd   rm   r   fetch_fieldsr!   	partitionr   r,   
fields_mapr   resolve_orderingrelated_modelr   
isinstancer   r   r(   termrc   orderbygetr^   source_fieldget_for_dialectrn   rg   dialect)rC   rd   rw   r   r   rS   r   r   r   r   	forwardedr   related_tabler   
annotationfield_objectr   funcrA   rA   rD   r      s^   




zAwaitableQuery.resolve_orderingc              	   C  s   | j sdS i }| j  D ]$\}}t|trt|d||< q|t| j| jjj	| j | j
d||< q| D ]\}}|jD ]}| | q<|| j v rS| j|j| q5tdd | D S )NF)r   rv   c                 s  s    | ]}|j jV  qd S r@   )r   is_aggregate)r~   inforA   rA   rD   	<genexpr>      z3AwaitableQuery._resolve_annotate.<locals>.<genexpr>)rh   itemsr   r   r)   r   r(   rd   rm   r|   ri   r   r   rc   _select_otherr   as_anyr]   )rC   annotation_infokeyr   r   r   rA   rA   rD   r     s*   

	

z AwaitableQuery._resolve_annotatec                 C  s4   |    |   |r| j }|S | j \}}|S )a  
        Returns the SQL query that will be executed. By default, it will return the query with
        placeholders, but if you set `params_inline=True`, it will inline the parameters.

        :param params_inline: Whether to inline the parameters
        )ru   _make_queryrc   get_sqlget_parameterized_sql)rC   params_inlinesql_rA   rA   rD   r     s   
zAwaitableQuery.sqlc                 C  s   t  r@   NotImplementedErrorrB   rA   rA   rD   r   .  s   zAwaitableQuery._make_queryr   c                   s   t  r@   r   rB   rA   rA   rD   _execute1  s   zAwaitableQuery._executerd   rk   r=   rl   F)rq   rX   r=   r   )rq   rX   r=   rl   r=   rl   )rw   r   r   rM   r   r,   r=   r   )r   r0   r=   rl   )r   rM   r   rX   r=   r   r@   )rd   r   rw   r   r   r   r   r   rS   r   r=   rl   r=   rX   r=   rM   r=   r   )r^   r_   r`   	__slots__rp   rt   ru   r   r   r   staticmethodr   r   r   r   r   r   rA   rA   rA   rD   rb   R   s     





Q
rb   c                      s  e Zd ZdZd fddZdd	d
ZdddZdddZdddZ	ddddZ	dddZ
dd!d"Zdd#d$Zdd%d&Zdd)d*Zdd,d-Zdd0d1Zdd2d3Z			4	ddd9d:Zdd<d=Zdd?d@ZddAddEdFZddHdIZddKdLZddNdOZddQdRZddTdUZddVdWZddZd[Zdd\d]Zdd^d_ZddadbZddgdhZ 	i		i	idddsdtZ!	idddwdxZ"ddydzZ#dd|d}Z$dd~dZ%dddZ&dddZ'dddZ(dddZ)dddZ*dddZ+dddZ,dddZ-dddZ.dddZ/dddZ0  Z1S )QuerySet)fields_prefetch_map_prefetch_queries_single_raise_does_not_existrf   _limit_offset_fields_for_select_filter_kwargs
_orderings	_distinct_having
_group_bys_select_for_update_select_for_update_nowait_select_for_update_skip_locked_select_for_update_of_select_for_update_no_key_select_related_select_related_idx_use_indexes_force_indexesrd   rk   r=   rl   c                   s   t  | |jj| _i | _i | _d| _d| _d | _	d | _
i | _g | _d| _i | _d| _d| _d| _d| _d| _t | _d| _t | _g | _t | _t | _d S )NFrA   )superrp   rm   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   setr   r   r   r   r   r   ro   	__class__rA   rD   rp   P  s0   
zQuerySet.__init__QuerySet[MODEL]c                 C  s*  | j | j }| j|_| j|_| j|_| j|_t| j|_t| j|_| j	|_	| j
|_
| j|_| j|_| j|_| j|_t| j|_t| j|_t| j|_t| j|_| j|_t| j|_t| j|_t| j|_t| j|_| j|_| j|_| j|_| j|_| j|_| j|_| j|_| j|_| j |_ |S r@   )!r   __new__r   rd   rc   rg   r	   r   r   r   r   rf   r   r   r   r   r   re   rj   r   rh   r   ri   r   r   r   r   r   r   r   r   r   r   rC   querysetrA   rA   rD   _clonek  s@   zQuerySet._clonerG   r&   negaterX   rO   r   c                O  s   |   }|D ]}t|tstd|r|j|  q|j| q| D ]"\}}|r;|jtdi ||i  q&|jtdi ||i q&|S )Nzexpected Q objects as argsrA   )r   r   r&   	TypeErrorrj   r   r   )rC   r   rG   rO   r   argr   valuerA   rA   rD   _filter_or_exclude  s   
zQuerySet._filter_or_excludec                 O     | j |ddi|S )z
        Filters QuerySet by given kwargs. You can filter by related objects like this:

        .. code-block:: python3

            Team.filter(events__tournament__name='Test')

        You can also pass Q objects to filters as args.
        r   Fr   r\   rA   rA   rD   filter  s   
zQuerySet.filterc                 O  r   )zG
        Same as .filter(), but with appends all args with NOT
        r   Tr   r\   rA   rA   rD   exclude  s   zQuerySet.excludeFr   tuple[str, ...]list[tuple[str, Order]]c                 C  sl   g }|D ]/}| j ||d\}}|dd | jjjv s,|| jv s,td| d| jj |||f q|S )z
        Convert ordering from strings to standard items for queryset.

        :param orderings: What columns/order to order by
        :param reverse:  Whether reverse order
        :return: standard ordering for QuerySet.
        r   r   r   r   r   )	r   splitrd   rm   r   rh   r!   r^   r   )rC   r   r   new_orderingr   r   r   rA   rA   rD   _parse_orderings  s   
zQuerySet._parse_orderingsrM   c                 G  s   |   }| ||_|S )aa  
        Accept args to filter by in format like this:

        .. code-block:: python3

            .order_by('name', '-tournament__name')

        Supports ordering by related models too.
        A '-' before the name will result in descending sort order, default is ascending.

        :raises FieldError: If unknown field has been provided.
        )r   r  r   rC   r   r   rA   rA   rD   order_by  s   zQuerySet.order_byQuerySetSingle[MODEL | None]c                 C  s   d| _ d| _tttt  | S )NTr{   )r   r   r   r<   r   r7   rB   rA   rA   rD   
_as_single  s   zQuerySet._as_singlec                 G  s,   |st d|  }| j|dd|_| S )z
        Returns the most recent object by ordering descending on the providers fields.

        :params orderings: Fields to order by.

        :raises FieldError: If unknown or no fields has been provided.
        No fields passedTr  r!   r   r  r   r  r	  rA   rA   rD   latest  s
   zQuerySet.latestc                 G  s(   |st d|  }| ||_| S )z
        Returns the earliest object by ordering ascending on the specified field.

        :params orderings: Fields to order by.

        :raises FieldError: If unknown or no fields has been provided.
        r  r  r	  rA   rA   rD   earliest  s
   zQuerySet.earliestlimitintc                 C  s"   |dk rt d|  }||_|S )zu
        Limits QuerySet to given length.

        :raises ParamsError: Limit should be non-negative number.
        r   z#Limit should be non-negative number)r$   r   r   )rC   r  r   rA   rA   rD   r    s
   zQuerySet.limitoffsetc                 C  s:   |dk rt d|  }||_| jjr|jdu rd|_|S )zp
        Query offset for QuerySet.

        :raises ParamsError: Offset should be non-negative number.
        r   z$Offset should be non-negative numberNi@B )r$   r   r   rg   requires_limitr   )rC   r  r   rA   rA   rD   r    s   zQuerySet.offsetr   slicec                 C  s   t |ts	td|jdu st |jtr|jdkstd|jdur%|jnd}t |tr0|dk r4td|jdurHt |jtrD|j|krHtd| |}|jrX||j| }|S )aw  
        Query offset and limit for Queryset.

        :raises ParamsError: QuerySet indices must be slices.

        :raises ParamsError: Slice steps should be 1 or None.

        :raises ParamsError: Slice start should be non-negative number or None.

        :raises ParamsError: Slice stop should be non-negative number greater that slice start,
        or None.
        z QuerySet indices must be slices.Nr{   z Slice steps should be 1 or None.r   z2Slice start should be non-negative number or None.zKSlice stop should be non-negative number greater that slice start, or None.)	r   r  r$   stepr  startstopr  r  )rC   r   r  r   rA   rA   rD   __getitem__  s   
  
zQuerySet.__getitem__c                 C  s   |   }d|_|S )z
        Make QuerySet distinct.

        Only makes sense in combination with a ``.values()`` or ``.values_list()`` as it
        precedes all the fetched fields with a distinct.
        T)r   r   r   rA   rA   rD   distinct6  s   zQuerySet.distinctrA   nowaitskip_lockedofno_keyc                 C  sB   | j jr|  }d|_||_||_t||_|o| j j|_	|S | S )a  
        Make QuerySet select for update.

        Returns a queryset that will lock rows until the end of the transaction,
        generating a SELECT ... FOR UPDATE SQL statement on supported databases.

        :param nowait:
            If `True`, raise an error if the lock cannot be obtained immediately.
        :param skip_locked:
            If `True`, skip rows that are already locked by other transactions instead of waiting.
        :param of:
            Specify the tables to lock when dealing with multiple related tables, e.g. when using `select_related`.
            Provide a tuple of table names to indicate which tables' rows should be locked. By default, all fetched
            rows are locked.
        :param no_key:
            If `True`, use the lower SELECT ... FOR NO KEY UPDATE lock strength on PostgreSQL to allow creating or
            deleting rows in other tables that reference the locked rows via foreign keys. The parameter is ignored
            on other backends.
        T)
rg   support_for_updater   r   r   r   r   r   support_for_no_key_updater   )rC   r  r  r  r  r   rA   rA   rD   select_for_updateA  s   

zQuerySet.select_for_updaterP   c                 K  sH   ddl m} |  }| D ]\}}||j|< |j||d| q|S )z
        Annotate result with aggregation or function result.

        :raises TypeError: Value of kwarg is expected to be a ``Function`` instance.
        r   )get_filters_for_fieldN)tortoise.modelsr"  r   r   rh   ri   update)rC   rO   r"  r   r   r   rA   rA   rD   rR   g  s   
zQuerySet.annotater   c                 G  s   |   }||_|S )z
        Make QuerySet returns list of dict or tuple with group by.

        Must call before .values() or .values_list()
        )r   r   )rC   r   r   rA   rA   rD   group_byw  s   zQuerySet.group_byrU   rW   rV   ValuesListQuery[Literal[False]]c                  s    j rtd|p fdd jjjD t j  }tdi d j	d jd j
d jd jd	|d
|d jd jd jd jd jd jd jd jd jS )a  
        Make QuerySet returns list of tuples for given args instead of objects.

        If call after `.get()`, `.get_or_none()` or `.first()` return tuples for given args instead of object.

        If ```flat=True`` and only one arg is passed can return flat list or just scalar.

        If no arguments are passed it will default to a tuple containing all fields
        in order of declaration.
        z*.values_list() cannot be used with .only()c                   s   g | ]}| j jjv r|qS rA   )rd   rm   r   r}   rB   rA   rD   r     s
    z(QuerySet.values_list.<locals>.<listcomp>rn   rd   	q_objectssingleraise_does_not_existrV   fields_for_select_listr  r  r  r   r   rx   	group_bysforce_indexesuse_indexesNrA   )r   
ValueErrorrd   rm   r   listrh   keysValuesListQueryrf   rj   r   r   r   r   r   r   ri   r   r   r   )rC   rV   rW   r*  rA   rB   rD   rZ     sP   	
zQuerySet.values_listValuesQuery[Literal[False]]c                   s    j rtd|s|r:i }|D ]}||v rtd| |||< q| D ]\}}||v r4td| |||< q%n fdd jjj D t j	  }dd |D }t
 j j j j j| j j j j j	 j j j jdS )a  
        Make QuerySet return dicts instead of objects.

        If called after `.get()`, `.get_or_none()` or `.first()`, returns a dict instead of an object.

        You can specify which fields to include by:
        - Passing field names as positional arguments
        - Using kwargs in the format `field_name='name_in_dict'` to customize the keys in the resulting dict

        If no arguments are passed, it will default to a dict containing all fields.

        :raises FieldError: If duplicate key has been provided.
        z%.values() cannot be used with .only()zDuplicate key c                   s"   g | ]}| j jj v r|qS rA   )rd   rm   fields_db_projectionr0  r}   rB   rA   rD   r     s    z#QuerySet.values.<locals>.<listcomp>c                 S  s   i | ]}||qS rA   rA   r}   rA   rA   rD   
<dictcomp>  s    z#QuerySet.values.<locals>.<dictcomp>)rn   rd   r'  r(  r)  rS   r  r  r  r   r   rx   r+  r,  r-  )r   r.  r!   r   rd   rm   r   r0  r/  rh   ValuesQueryrf   rj   r   r   r   r   r   r   ri   r   r   r   )rC   rG   rO   rS   r   	return_as_fieldsrA   rB   rD   r]     sH   


zQuerySet.valuesDeleteQueryc              	   C  $   t | j| j| j| j| j| j| jdS )z1
        Delete all objects in QuerySet.
        )rn   rd   r'  r   rx   r  r   )r8  rf   rd   rj   rh   ri   r   r   rB   rA   rA   rD   delete     zQuerySet.deleteUpdateQueryc              
   K  s&   t | j| j|| j| j| j| j| jdS )a*  
        Update all objects in QuerySet with given kwargs.

        .. admonition: Example:

            .. code-block:: py3

                await Employee.filter(occupation='developer').update(salary=5000)

        Will instead of returning a resultset, update the data in the DB itself.
        )rn   rd   update_kwargsr'  r   rx   r  r   )r<  rf   rd   rj   rh   ri   r   r   rQ   rA   rA   rD   r$    s   zQuerySet.update
CountQueryc                 C  s,   t | j| j| j| j| j| j| j| j| j	d	S )zI
        Return count of objects in queryset instead of objects.
        )	rn   rd   r'  r   rx   r  r  r,  r-  )
r>  rf   rd   rj   rh   ri   r   r   r   r   rB   rA   rA   rD   count  s   zQuerySet.countExistsQueryc              	   C  r9  )z<
        Return True/False whether queryset exists.
        )rn   rd   r'  r   rx   r,  r-  )r@  rf   rd   rj   rh   ri   r   r   rB   rA   rA   rD   exists  r;  zQuerySet.existsc                 C  s   |   S )zf
        Return the whole QuerySet.
        Essentially a no-op except as the only operation.
        )r   rB   rA   rA   rD   all  s   zQuerySet.allr   RawSQLQueryc                 C  s   t | j| j|dS )z2
        Return the QuerySet from raw SQL
        )rd   rn   r   )rC  rd   rf   )rC   r   rA   rA   rD   raw&  s   zQuerySet.rawc                 C  s   |   }| S )zU
        Limit queryset to one object and return one object instead of list.
        )r   r  r   rA   rA   rD   first,  s   zQuerySet.firstc                 C  s`   |   }|jrdd |jD }n| jjj }r|jtjfg}n
td| jj	 d||_|
 S )zZ
        Limit queryset to one object and return the last object instead of list.
        c                 S  s*   g | ]\}}||t jkrt jnt jfqS rA   r   )r~   r   r   rA   rA   rD   r   :  s    z!QuerySet.last.<locals>.<listcomp>z#QuerySet has no ordering and model z has no pk defined)r   r   rd   rm   pkmodel_field_namer   r   r!   r^   r  )rC   r   r  rF  rA   rA   rD   last3  s   zQuerySet.lastQuerySetSingle[MODEL]c                 O  s&   | j |i |}d|_d|_d|_|S zC
        Fetch exactly one object matching the parameters.
           T)r  r   r   r   rC   rG   rO   r   rA   rA   rD   r   G  s
   zQuerySet.getid_listIterable[str | int]r   dict[str, MODEL]c                   s4   | j di   d|iI dH } fdd|D S )a  
        Return a dictionary mapping each of the given IDs to the object with
        that ID. If `id_list` isn't provided, evaluate the entire QuerySet.

        :param id_list: A list of field values
        :param field_name: Must be a unique field
        __inNc                   s   i | ]}t | |qS rA   )getattrr~   objr   rA   rD   r4  Z  s    z$QuerySet.in_bulk.<locals>.<dictcomp>rA   )r  )rC   rM  r   objsrA   rT  rD   in_bulkQ  s    zQuerySet.in_bulkNobjectsIterable[MODEL]
batch_size
int | Noneignore_conflictsupdate_fieldsIterable[str] | Noneon_conflictBulkCreateQuery[MODEL]c              	   C  sF   |r|rt d|s|r|r|r|st dt| j| j|||||dS )a&  
        This method inserts the provided list of objects into the database in an efficient manner
        (generally only 1 query, no matter how many objects there are).

        :param on_conflict: On conflict index name
        :param update_fields: Update fields when conflicts
        :param ignore_conflicts: Ignore conflicts when inserting
        :param objects: List of objects to bulk create
        :param batch_size: How many objects are created in a single query

        :raises ValueError: If params do not meet specifications
        z:ignore_conflicts and update_fields are mutually exclusive.z4update_fields and on_conflict need set in same time.)rn   rd   rW  rY  r[  r\  r^  )r.  BulkCreateQueryrf   rd   )rC   rW  rY  r[  r\  r^  rA   rA   rD   bulk_create\  s    zQuerySet.bulk_createIterable[str]BulkUpdateQuery[MODEL]c                 C  sD   t dd |D rtdt| j| j| j| j| j| j| j	|||d
S )aE  
        Update the given fields in each of the given objects in the database.

        :param objects: List of objects to bulk create
        :param fields: The fields to update
        :param batch_size: How many objects are created in a single query

        :raises ValueError: If objects have no primary key set
        c                 s  s    | ]}|j d u V  qd S r@   )rF  rR  rA   rA   rD   r     s    z'QuerySet.bulk_update.<locals>.<genexpr>z6All bulk_update() objects must have a primary key set.)
rn   rd   r'  r   rx   r  r   rW  r   rY  )
r   r.  BulkUpdateQueryrf   rd   rj   rh   ri   r   r   )rC   rW  r   rY  rA   rA   rD   bulk_update  s   zQuerySet.bulk_updatec                 O  s    | j |i |}d|_d|_|S rJ  )r  r   r   rL  rA   rA   rD   get_or_none  s   zQuerySet.get_or_nonerS   c                 G  s   |st d|  }||_|S )a9  
        Fetch ONLY the specified fields to create a partial model.

        Persisting changes on the model is allowed only when:

        * All the fields you want to update is specified in ``<model>.save(update_fields=[...])``
        * You included the Model primary key in the `.only(...)``

        To protect against common mistakes we ensure that errors get raised:

        * If you access a field that is not specified, you will get an ``AttributeError``.
        * If you do a ``<model>.save()`` a ``IncompleteInstanceError`` will be raised as the model is, as requested, incomplete.
        * If you do a ``<model>.save(update_fields=[...])`` and you didn't include the primary key in the ``.only(...)``,
          then ``IncompleteInstanceError`` will be raised indicating that updates can't be done without the primary key being known.
        * If you do a ``<model>.save(update_fields=[...])`` and one of the fields in ``update_fields`` was not in the ``.only(...)``,
          then ``IncompleteInstanceError`` as that field is not available to be updated.
        z#.only() requires at least one field)r.  r   r   )rC   rS   r   rA   rA   rD   rT     s
   zQuerySet.onlyc                 G  s"   |   }|D ]}|j| q|S )z
        Return a new QuerySet instance that will select related objects.

        If fields are specified, they must be ForeignKey fields and only those
        related objects are included in the selection.
        )r   r   add)rC   r   r   r   rA   rA   rD   rN     s   zQuerySet.select_relatedindex_namesc                 G  .   | j jr|  }|D ]}|j| q
|S | S )z
        The FORCE INDEX hint acts like USE INDEX (index_list),
        with the addition that a table scan is assumed to be very expensive.
        )rg   support_index_hintr   r   rg  rC   rh  r   
index_namerA   rA   rD   force_index  s   zQuerySet.force_indexc                 G  ri  )z}
        The USE INDEX (index_list) hint tells MySQL to use only one of the named indexes to find rows in the table.
        )rg   rj  r   r   rg  rk  rA   rA   rD   	use_index  s   zQuerySet.use_indexrH   c                 G  s   |   }i |_|D ]V}t|tr|| q	|d\}}}|| jjjvrH|| jjj	v r:t
d| d| jjj dt
d| d| jjj d||j vrUt |j|< |r_|j| | q	|S )z
        Like ``.fetch_related()`` on instance, but works on all objects in QuerySet.

        :raises FieldError: If the field to prefetch on is not a relation, or not found.
        r   Field z on z is not a relationz	Relation z for z
 not found)r   r   r   r.   resolve_for_querysetr   rd   rm   r   r   r!   	full_namer0  r   rg  )rC   rG   r   relationfirst_level_fieldr   forwarded_prefetchrA   rA   rD   rK     s*   

zQuerySet.prefetch_relatedc                   s8   |    |   | jj| j| jd| j I dH S )aT  Fetch and return information about the query execution plan.

        This is done by executing an ``EXPLAIN`` query whose exact prefix depends
        on the database backend, as documented below.

        - PostgreSQL: ``EXPLAIN (FORMAT JSON, VERBOSE) ...``
        - SQLite: ``EXPLAIN QUERY PLAN ...``
        - MySQL: ``EXPLAIN FORMAT=JSON ...``

        .. note::
            This is only meant to be used in an interactive environment for debugging
            and query optimization.
            **The output format may (and will) vary greatly depending on the database backend.**
        rd   rn   N)ru   r   rf   executor_classrd   execute_explainrc   r   rB   rA   rA   rD   explain  s   
zQuerySet.explainrf   BaseDBAsyncClient | Nonec                 C  s"   |   }|r||_|S |j|_|S )zc
        Executes query in provided db client.
        Useful for transactions workaround.
        )r   rf   )rC   rf   r   rA   rA   rD   using_db  s
   
zQuerySet.using_dblookup_expressiontuple[type[Model], Table]c                   s   t | j|}| j}| jjj d}|D ]E}tt|}||jf }|  |j| | jr+q|j	jj
}|j	t||j||f}|j	}|| jvrI| j| | jj fdd|D  | _q| fS )Nr@   c                   (   g | ]} |     d | qS .r   get_table_name)r~   r   rw   rA   rD   r   9  s    z1QuerySet._join_select_related.<locals>.<listcomp>)r1   rd   rm   r|   r   r,   rG  r   r   r   r   lenr   r   rc   select)rC   r{  r   rd   pathr   related_fieldsappend_itemrA   r  rD   _join_select_related  s4   




zQuerySet._join_select_relatedonly_lookup_expressionsc              
     s  t t}t|dd dD ]}|d\}}}|| | q|dd }|ri jjj j	 jt
| jdf z jj fdd|D   _W n- tyh } ztd	|jd
  d jj d|d }~ww  j	 jd
 jjj jdf t }	| D ]f\}}t j|}
 j }} jjjd}t|
D ]=\}}tt|}||jf } |j||}|j}||	v rq j	||t
|
d krt
|nd
||f |	| q jjfdd|D   _qd S )Nc                 S  s
   |  dS )Nr   )r?  xrA   rA   rD   <lambda>H  s   
 z(QuerySet._resolve_only.<locals>.<lambda>)r   r    r@   c                   s.   g | ]}| j vr jjj|  |qS rA   )rh   rd   rm   r3  r   r}   rC   rw   rA   rD   r   [  s    z*QuerySet._resolve_only.<locals>.<listcomp>Unknown field "r   " for model ""r{   c                   r}  r~  r  r}   r  rA   rD   r     s   ( )r   r/  sorted
rpartitionr   poprd   rm   r|   r   r  rc   r  KeyErrorr!   rG   r^   r   r   r1   	enumerater   r,   rG  r   r   rg  )rC   r  fetch_to_fields
expressionfetch_fields_lookupr   r   data_fieldseadded_pathsr   referring_modelrd   r  ifetch_fieldr   rA   r  rD   _resolve_only@  s   
	



zQuerySet._resolve_onlyc                 C  sl  g | _ g | _| jjj}| jr| jjj | _| 	| j n!t
| jjj| _| jt| jjjt| j || jdf}| j | | | j| jjj| j| j| j |   | jd ur`| j| j| j_| jd urn| j| j| j_| jrud| j_| jr| j| j| j| j| j| _| jr| jD ]}| | q| jrg | j_| jj | j | _| j!rg | j_!| jj"| j! | _d S d S )Nr@   T)#r   re   rd   rm   r|   r   	basequeryr  rc   r  r	   basequery_all_fieldsr  r   rh   r   r   r   r   r   _wrapper_clsr   r   r   
for_updater   r   r   r   r   r  r   rm  r   rn  )rC   rw   r  rN   rA   rA   rD   r     sZ   



zQuerySet._make_query!Generator[Any, None, list[MODEL]]c                 C  s,   | j d u r| | j| _ |   |   S r@   )rf   rt   r   r   r   rE   rB   rA   rA   rD   rE     s   
zQuerySet.__await__AsyncIterator[MODEL]c                 C    | I d H D ]}|V  qd S r@   rA   rC   valrA   rA   rD   	__aiter__     zQuerySet.__aiter__list[MODEL]c                   s   | j j| j| j | j| j| jdj| j dt	| j
 iI d H }| jr@t|dkr/|d S |s;| jr9t| jd S t| j|S )N)rd   rn   prefetch_mapprefetch_queriesselect_related_idxcustom_fieldsr{   r   )rf   rv  rd   r   r   r   execute_selectrc   r   r/  rh   r0  r   r  r   r    r#   rC   instance_listrA   rA   rD   r     s,   


zQuerySet._executer   )r=   r   )rG   r&   r   rX   rO   r   r=   r   )rG   r&   rO   r   r=   r   r   )r   r  r=   r  )r   rM   r=   r   )r=   r  )r   rM   r=   r  )r  r  r=   r   )r  r  r=   r   )r   r  r=   r   )FFrA   F)
r  rX   r  rX   r  r  r  rX   r=   r   )rO   rP   r=   r   )r   rM   r=   r   )rW   rM   rV   rX   r=   r&  )rG   rM   rO   rM   r=   r2  )r=   r8  )rO   r   r=   r<  )r=   r>  )r=   r@  )r   rM   r=   rC  )rG   r&   rO   r   r=   rI  )rM  rN  r   rM   r=   rO  NFNN)rW  rX  rY  rZ  r[  rX   r\  r]  r^  r]  r=   r_  r@   )rW  rX  r   rb  rY  rZ  r=   rc  )rG   r&   rO   r   r=   r  )rS   rM   r=   r   )rh  rM   r=   r   )rG   rH   r=   r   r   )rf   ry  r=   r   )r{  rM   r=   r|  )r  r  r=   rl   r   r=   r  )r=   r  )r=   r  )2r^   r_   r`   r   rp   r   r   r  r  r  r
  r  r  r  r  r  r  r  r!  rR   r%  rZ   r]   r:  r$  r?  rA  rB  rD  rE  rH  r   rV  ra  re  rf  rT   rN   rm  rn  rK   rx  rz  r  r  r   rE   r  r   __classcell__rA   rA   r   rD   r   5  sr    

"









!
&


$
7









)

	






	
!
U
3
r   c                      s@   e Zd ZdZd fddZdddZdddZd ddZ  ZS )!r<  )r=  r   r   r]   rd   rk   r=  dict[str, Any]rn   r   r'  list[Q]r   rx   dict[str, FilterInfoDict]r  rZ  r   list[tuple[str, str]]r=   rl   c	           	        s:   t  | || _|| _|| _|| _|| _|| _|| _d S r@   )	r   rp   r=  rj   rh   ri   rf   r   r   )	rC   rd   r=  rn   r'  r   rx   r  r   r   rA   rD   rp     s   
zUpdateQuery.__init__c              	   C  s  | j jj}| jj|| _| jjr(| j	r(| j
| j	| j_	| | j || j| j |   | j D ]\}}| j jj|}|sJtd| d| j  |jrUtd| dt|ttfr| j || |j}| j jj| j}| j jj| t||jjd }n9z	| j jj| }W n t y   td| dw t|t!r|"t#| j || j| j$dj%}n| j jj| |d }| j&||| _q1d S )NzUnknown keyword argument r   ro  z is PK and can not be updatedz" is virtual and can not be updatedrv   )'rd   rm   r|   rf   query_classr$  rc   rg   support_update_limit_order_byr   r  r   r   rh   r   r=  r   r   r   r!   rF  r"   r   r*   r+   _validate_relation_typer   to_db_valuerQ  to_field_instancerG  r3  r  r%   r   r(   ri   r   r   )rC   rw   r   r   r   fk_fielddb_fieldrA   rA   rD   r     sN   

	zUpdateQuery._make_queryGenerator[Any, None, int]c                 C     |  d |   |   S NTru   r   r   rE   rB   rA   rA   rD   rE   .     
zUpdateQuery.__await__r  c                      | j j| j  I d H d S Nr   rf   execute_queryrc   r   rB   rA   rA   rD   r   3     zUpdateQuery._execute)rd   rk   r=  r  rn   r   r'  r  r   r  rx   r  r  rZ  r   r  r=   rl   r   r=   r  r=   r  	r^   r_   r`   r   rp   r   rE   r   r  rA   rA   r   rD   r<    s    

*r<  c                      s@   e Zd ZdZd fddZdddZdddZdddZ  ZS ) r8  )rh   ri   r   r   rd   rk   rn   r   r'  r  r   r  rx   r  r  rZ  r   r  r=   rl   c                   s4   t  | || _|| _|| _|| _|| _|| _d S r@   )r   rp   rj   rh   ri   rf   r   r   )rC   rd   rn   r'  r   rx   r  r   r   rA   rD   rp   ?     

zDeleteQuery.__init__c                 C  sb   t | jjj| _| jjr'| jr'| j| j| j_| j	| j| jjj
| j| jd |   d| j_d S )Nrd   rw   r   r   T)r	   rd   rm   r  rc   rg   r  r   r  r   r|   r   rh   r   _delete_fromrB   rA   rA   rD   r   Q  s   zDeleteQuery._make_queryr  c                 C  r  r  r  rB   rA   rA   rD   rE   _  r  zDeleteQuery.__await__r  c                   r  r  r  rB   rA   rA   rD   r   d  r  zDeleteQuery._execute)rd   rk   rn   r   r'  r  r   r  rx   r  r  rZ  r   r  r=   rl   r   r  r  r  rA   rA   r   rD   r8  7  s    

r8  c                      s@   e Zd ZdZd fddZdddZdddZdddZ  ZS )r@  )r   r   rd   rk   rn   r   r'  r  r   r  rx   r  r,  set[str]r-  r=   rl   c                   s4   t  | || _|| _|| _|| _|| _|| _d S r@   )r   rp   rj   rf   rh   ri   r   r   )rC   rd   rn   r'  r   rx   r,  r-  r   rA   rD   rp   n  r  zExistsQuery.__init__c                 C  s   t | jjj| _|   | jd| j_| jt	ddd | j
r-g | j_
| jj| j
 | _| jr>g | j_| jj| j | _d S d S )Nr{   F)allow_parametrize)r	   rd   rm   r  rc   r   r  r   r   r   r   rm  r   rn  rB   rA   rA   rD   r     s   zExistsQuery._make_queryGenerator[Any, None, bool]c                 C     |    |   |   S r@   r  rB   rA   rA   rD   rE        zExistsQuery.__await__rX   c                   s&   | j j| j  I d H \}}t|S r@   )rf   r  rc   r   rX   )rC   resultr   rA   rA   rD   r     s   zExistsQuery._execute)rd   rk   rn   r   r'  r  r   r  rx   r  r,  r  r-  r  r=   rl   r   )r=   r  r   r  rA   rA   r   rD   r@  h  s    

r@  c                      s@   e Zd ZdZd fddZdddZd ddZd!ddZ  ZS )"r>  )r   r   r   r   rd   rk   rn   r   r'  r  r   r  rx   r  r  rZ  r  r,  r  r-  r=   rl   c
           
        sD   t  | || _|| _|| _|| _|pd| _|| _|| _|	| _	d S r  )
r   rp   rj   rh   ri   r   r   rf   r   r   )
rC   rd   rn   r'  r   rx   r  r  r,  r-  r   rA   rD   rp     s   

zCountQuery.__init__c                 C  s   t | jjj| _|   tt }| jjr|	 }g | j_
| j| | jr2g | j_| jj| j | _| jrCg | j_| jj| j | _d S d S r@   )r	   rd   rm   r  rc   r   r   r   	_groupbysover_selectsr   r   rm  r   rn  )rC   
count_termrA   rA   rD   r     s   
zCountQuery._make_queryr  c                 C  r  r@   r  rB   rA   rA   rD   rE     r  zCountQuery.__await__r  c                   s^   | j j| j  I d H \}}|sdS tt|d  d | j }| jr-|| jkr-| jS |S r  )	rf   r  rc   r   r/  dictr]   r   r   )rC   r   r  r?  rA   rA   rD   r     s   zCountQuery._execute)rd   rk   rn   r   r'  r  r   r  rx   r  r  rZ  r  rZ  r,  r  r-  r  r=   rl   r   r  r  r  rA   rA   r   rD   r>    s    

r>  c                      sF   e Zd Zd fddZdddZdddZdddZdddZ  ZS ) FieldSelectQueryrd   rk   r   r  r=   rl   c                   s   t  | || _d S r@   )r   rp   rh   )rC   rd   r   r   rA   rD   rp     s   
zFieldSelectQuery.__init__rw   r   r   rM   forwarded_fieldstuple[Table, str]c                 C  s   ||j jv r|s||j j| fS ||j jv r$|r$td| d|j d|| jj jv r5|s5td| dtt|j j	
|}|sMtd| d|j d| |||}|d\}}}| j|j|||d	S )
NzField "r  z" is not relationSelecting relation "9" is not possible, select concrete field on related modelr  r  r   rd   rw   r   r  )rm   r3  r!   r^   rd   r   r.  r   r,   r   r   r   r   !_join_table_with_forwarded_fieldsr   )rC   rd   rw   r   r  r   r   forwarded_fields_rA   rA   rD   r    s&   
z2FieldSelectQuery._join_table_with_forwarded_fieldsr6  c           
      C  s   | j jj}|| jv r| j| | j|< d S || j jjv r/| j jj| }| j|| | d S || j jjv r>t	d| d|
d\}}}|| j jjv rf| j| j |||d\}}	| j||	 | d S td| d| j j d)Nr  r  r   r  r  r  r  )rd   rm   r|   rh   r3  rc   _select_fieldr   r   r.  r   r  r!   r^   )
rC   r   r6  rw   r  field_r   r  r   related_db_fieldrA   rA   rD   add_field_to_select_query  s.   



z*FieldSelectQuery.add_field_to_select_queryr   c           	      C  s   ||j jv r
dd S |dd |j jD v rdd S || jv r2| j| }t|dd }|r.|jS dd S ||j jv r?|j j| jS |d\}}}||j jv rZ|j j| j}| 	||S t
d	| d
| d)Nc                 S     | S r@   rA   r  rA   rA   rD   r    rF   z:FieldSelectQuery.resolve_to_python_value.<locals>.<lambda>c                 s  s    | ]}|d  V  qdS )r{   NrA   )r~   r  rA   rA   rD   r      r   z;FieldSelectQuery.resolve_to_python_value.<locals>.<genexpr>c                 S  r  r@   rA   r  rA   rA   rD   r  !  rF   r   c                 S  r  r@   rA   r  rA   rA   rD   r  (  rF   r   r  r  r  )rm   r   db_native_fieldsrh   rQ  to_python_valuer   r   r   resolve_to_python_valuer!   )	rC   rd   r   r   r   r  r   r  	new_modelrA   rA   rD   r    s"   

z(FieldSelectQuery.resolve_to_python_valuefield_namesr/  c           	      G  s   g }|D ]9}|| j v r|t| q|d\}}}| j| j| jjj||d\}}|| |	  d| }|| q|S )Nr   r  )
rh   r   r   r   r  rd   rm   r|   r   r  )	rC   r  r+  r   r   r   r  r   r  rA   rA   rD   _resolve_group_bys4  s"   

z#FieldSelectQuery._resolve_group_bys)rd   rk   r   r  r=   rl   )
rd   rk   rw   r   r   rM   r  rM   r=   r  )r   rM   r6  rM   r=   rl   )rd   rk   r   rM   r=   r   )r  rM   r=   r/  )	r^   r_   r`   rp   r  r  r  r  r  rA   rA   r   rD   r    s    


r  c                      sf   e Zd ZdZd4 fdd Zd5d!d"Zed6d&d'Zed7d*d'Zd8d,d'Zd9d/d0Zd:d2d3Z	  Z
S );r1  )r   r   r   r   r   r   r   _fields_for_select_list_flatr   r   r   rd   rk   rn   r   r'  r  r(  rX   r)  r*  tuple[str, ...] | list[str]r  rZ  r  r  r   r  rV   r   r  rx   r  r+  r  r,  r  r-  r=   rl   c                   s   t  || |rt|dkrtddd t|D }|| _|| _|| _|	| _|
| _	|| _
|| _|| _|| _|| _|| _|| _|| _|| _|| _d S )Nr{   z2You can flat value_list only if contains one fieldc                 S  s   i | ]	\}}t ||qS rA   )rM   )r~   r  r   rA   rA   rD   r4  o  r   z,ValuesListQuery.__init__.<locals>.<dictcomp>)r   rp   r  r   r  r   r   r   r   r   ri   rj   r   r   r  r  rf   r   r   r   )rC   rd   rn   r'  r(  r)  r*  r  r  r  r   rV   r   rx   r+  r,  r-  rS   r   rA   rD   rp   X  s&   
zValuesListQuery.__init__c                 C  s   g | _ t| jjj| _| j D ]
\}}| || q| j	| j| jjj
| j| j| jd |   | jr<| j| j| j_| jrH| j| j| j_| jrOd| j_| jrZ| j| j | j_| jrig | j_| jj| j | _| jrzg | j_| jj| j | _d S d S )Nrd   rw   r   r   rS   T)re   r	   rd   rm   r  rc   r   r   r  r   r|   r   rh   r  r   r   r  r   r   r   r  r  r   rm  r   rn  )rC   positional_numberr   rA   rA   rD   r     s6   zValuesListQuery._make_queryrC   r&  +Generator[Any, None, list[tuple[Any, ...]]]c                 C  r?   r@   rA   rB   rA   rA   rD   rE        zValuesListQuery.__await__rY   %Generator[Any, None, tuple[Any, ...]]c                 C  r?   r@   rA   rB   rA   rA   rD   rE     r  1Generator[Any, None, list[Any] | tuple[Any, ...]]c                 C  r  r@   r  rB   rA   rA   rD   rE     r  ValuesListQuery[Any]AsyncIterator[Any]c                 C r  r@   rA   r  rA   rA   rD   r    r  zValuesListQuery.__aiter__list[Any] | tuplec                   s   j jj  I d H \}}fddj D  jr2 d d fdd}tt||}n fdd}tt||}j	r]t
|dkrL|d S |sXjrVtjd S tj|S )Nc                   "   g | ]\}}|   j|fqS rA   r  rd   )r~   r   namerB   rA   rD   r         z,ValuesListQuery._execute.<locals>.<listcomp>r   r{   c                   s    | d S )N0rA   entry)r   rA   rD   r    s    z*ValuesListQuery._execute.<locals>.<lambda>c                   s   t  fddD S )Nc                 3  s     | ]\}}| | V  qd S r@   rA   )r~   columnr   r  rA   rD   r     s    z=ValuesListQuery._execute.<locals>.<lambda>.<locals>.<genexpr>)tupler  )columnsr  rD   r    s    )rf   r  rc   r   r   r   r  r/  mapr   r  r   r    rd   r#   )rC   r   r  flatmap
lst_valueslistmaprA   )r  r   rC   rD   r     s(   


zValuesListQuery._execute)"rd   rk   rn   r   r'  r  r(  rX   r)  rX   r*  r  r  rZ  r  rZ  r  rX   r   r  rV   rX   r   r  rx   r  r+  r  r,  r  r-  r  r=   rl   r   )rC   r&  r=   r  )rC   rY   r=   r  )r=   r  )rC   r  r=   r  )r=   r  r^   r_   r`   r   rp   r   r   rE   r  r   r  rA   rA   r   rD   r1  H  s    
(

r1  c                      sf   e Zd ZdZd3 fddZd4d d!Zed5d%d&Zed6d)d&Zd7d+d&Zd8d.d/Zd9d1d2Z	  Z
S ):r5  )
r   r   r   r   r   r   r   r   r   r   rd   rk   rn   r   r'  r  r(  rX   r)  rS   dict[str, str]r  rZ  r  r  r   r  r   r  rx   r  r+  r  r,  r  r-  r=   rl   c                   s`   t  || || _|| _|| _|	| _|
| _|| _|| _|| _	|| _
|| _|| _|| _|| _d S r@   )r   rp   r   r   r   r   r   ri   rj   r   r   rf   r   r   r   )rC   rd   rn   r'  r(  r)  rS   r  r  r  r   r   rx   r+  r,  r-  r   rA   rD   rp     s   
zValuesQuery.__init__c                   s  g  _ t jjj _ j D ]
\}} || q j	 j jjj
 j j j d     fdd jjD  j_ jrK j j j_ jrW j j j_ jr^d j_ jri j j  j_ jrxg  j_ jj j  _ jrg  j_ jj j  _d S d S )Nr  c                   s   g | ]
}|j  jv r|qS rA   )aliasr   )r~   r  rB   rA   rD   r     s
    z+ValuesQuery._make_query.<locals>.<listcomp>T)re   r	   rd   rm   r  rc   r   r   r  r   r|   r   rh   r0  r   r  r   r  r   r   r   r  r  r   rm  r   rn  )rC   r6  r   rA   rB   rD   r     s<   

zValuesQuery._make_queryrC   r2  *Generator[Any, None, list[dict[str, Any]]]c                 C  r?   r@   rA   rB   rA   rA   rD   rE     r  zValuesQuery.__await__r[   $Generator[Any, None, dict[str, Any]]c                 C  r?   r@   rA   rB   rA   rA   rD   rE   $  r  ;Generator[Any, None, list[dict[str, Any]] | dict[str, Any]]c                 C  r  r@   r  rB   rA   rA   rD   rE   )  s   ValuesQuery[Any]AsyncIterator[dict[str, Any]]c                 C r  r@   rA   r  rA   rA   rD   r  0  r  zValuesQuery.__aiter__list[dict] | dictc                   s    j j j  I d H }dd  fdd j D D }|r4|D ]}|D ]\}}||| ||< q&q" jrRt|dkrA|d S |sM jrKt	 j
d S t j
|S )Nc                 S  s    g | ]}t |d  tjs|qS )r{   )r   types
LambdaType)r~   r  rA   rA   rD   r   6  s    z(ValuesQuery._execute.<locals>.<listcomp>c                   r  rA   r  )r~   r  r   rB   rA   rD   r   8  r  r{   r   )rf   execute_query_dictrc   r   r   r   r   r  r   r    rd   r#   )rC   r  r  rowcolr   rA   rB   rD   r   4  s*   
	

zValuesQuery._execute) rd   rk   rn   r   r'  r  r(  rX   r)  rX   rS   r  r  rZ  r  rZ  r  rX   r   r  r   r  rx   r  r+  r  r,  r  r-  r  r=   rl   r   )rC   r2  r=   r  )rC   r[   r=   r  )r=   r  )rC   r  r=   r  )r=   r  r
  rA   rA   r   rD   r5    s    
!%

r5  c                      s6   e Zd ZdZd fd
dZdddZdddZ  ZS )rC  )_sqlrf   rd   rk   rn   r   r   rM   r=   rl   c                   s   t  | || _|| _d S r@   )r   rp   r  rf   )rC   rd   rn   r   r   rA   rD   rp   R  s   
zRawSQLQuery.__init__r   c                   s:   | j j| j| j dt| j| j jjg I d H }|S )Nru  )	rf   rv  rd   r  r'   r  r   r  SQL_CONTEXTr  rA   rA   rD   r   W  s   zRawSQLQuery._executer  c                 C  s   |    |   S r@   )ru   r   rE   rB   rA   rA   rD   rE   ^  s   zRawSQLQuery.__await__)rd   rk   rn   r   r   rM   r=   rl   r   r  )r^   r_   r`   r   rp   r   rE   r  rA   rA   r   rD   rC  O  s
    
rC  c                      sP   e Zd ZdZ	d'd( fddZd)ddZd*ddZd+d!d"Zd,d-d%d&Z  Z	S ).rd  )r   _objects_batch_size_queriesNrd   rk   rn   r   r'  r  r   r  rx   r  r  rZ  r   r  rW  rX  r   rb  rY  c              
     s8   t  j|i ||||||d |	| _|| _|
| _g | _d S )N)r=  rn   r'  r   rx   r  r   )r   rp   r   r  r  r  )rC   rd   rn   r'  r   rx   r  r   rW  r   rY  r   rA   rD   rp   f  s   

zBulkUpdateQuery.__init__r=   list[tuple[str, list[Any]]]c                 C  s|  | j jj}| jj|| _| jjr)| j	r)| j
| j	| j_	| j| j || j| jd |   | j jj}| j jj| jp;|}t|}t| j| jD ]n}t| j}| jD ]]}t }g }	|D ]E}
| j jj| |
jd }|
jj| }|t|
||
}|||k| jjjdkrt| j
|| | jjjdn| j
| |	!| qZ|"||}|#|$|	}qQ| j%!| qGdd | j%D S )Nr  postgresSQL_TYPEc                 S  s   g | ]}|  qS rA   )r   )r~   rc   rA   rA   rD   r         z1BulkUpdateQuery._make_queries.<locals>.<listcomp>)&rd   rm   r|   rf   r  r$  rc   rg   r  r   r  r   r   rh   r   pk_attrr   r   r   r4   r  r  r	   r   r   r  rF  rQ  whenschema_generatorDIALECTr   r   r   r   whereisinr  )rC   rw   r!  source_pk_attrrF  objects_itemrc   r   casepk_listrS  pk_value	field_objfield_valuerA   rA   rD   _make_queries  sN   






zBulkUpdateQuery._make_queriesqueries_with_paramsr  c                   s4   d}|D ]\}}|| j ||I d H d 7 }q|S r  )rf   r  )rC   r/  r?  r   r]   rA   rA   rD   _execute_many  s
   zBulkUpdateQuery._execute_manyGenerator[Any, Any, int]c                 C  s    |  d |  }| | S r  ru   r.  r0  rE   )rC   queriesrA   rA   rD   rE     s   
zBulkUpdateQuery.__await__FrM   c                 C  s$   |    |  }ddd |D S )N;c                 S  s   g | ]\}}|qS rA   rA   )r~   r   r   rA   rA   rD   r     r   z'BulkUpdateQuery.sql.<locals>.<listcomp>)ru   r.  r   )rC   r   r3  rA   rA   rD   r     s   zBulkUpdateQuery.sqlr@   )rd   rk   rn   r   r'  r  r   r  rx   r  r  rZ  r   r  rW  rX  r   rb  rY  rZ  )r=   r  )r/  r  r=   r  )r=   r1  r   r   
r^   r_   r`   r   rp   r.  r0  rE   r   r  rA   rA   r   rD   rd  c  s    

,
rd  c                      sV   e Zd ZdZ				d"d# fddZd$ddZd%ddZd&ddZd'd(d d!Z  Z	S ))r`  )r  _ignore_conflictsr  rf   	_executor_update_fields_on_conflictNFrd   rk   rn   r   rW  rX  rY  rZ  r[  rX   r\  r]  r^  c                   s4   t  | || _|| _|| _|| _|| _|| _d S r@   )r   rp   r  r6  r  rf   r8  r9  )rC   rd   rn   rW  rY  r[  r\  r^  r   rA   rD   rp     r  zBulkCreateQuery.__init__r=   tuple[str, str]c                 C  s   | j j| j| j d| _| js| jrt| j \}}| jj|| jd}|}| jjj	r;| jjdd\}}| jj|d| jd}| jrld| jjj
 }||j| jpOg  }||j| jpZg  }| jD ]}||}||}q_| | fS | jj| jjfS )Nru  )r[  T)include_generatedF)has_generatedr[  new_)rf   rv  rd   r7  r6  r8  _prepare_insert_columns_prepare_insert_statementrm   generated_db_fieldsdb_tabler   r^  r9  	do_updater   insert_queryinsert_query_all)rC   r   r  rC  rD  columns_allr  update_fieldrA   rA   rD   r.    s2   



zBulkCreateQuery._make_queries
insert_sqlrM   insert_sql_allrl   c                   s   | j jj t| j| jD ]E}g }g }|D ]$jr*| fdd| jj	D  q| fdd| jj
D  q|rF| j||I d H  |rR| j||I d H  qd S )Nc                   "   g | ]} |  t|qS rA   r  rQ  r~   r   r   instancerA   rD   r          
z1BulkCreateQuery._execute_many.<locals>.<listcomp>c                   rI  rA   rJ  rK  rL  rA   rD   r   	  rN  )rd   rm   r   r4   r  r  _custom_generated_pkr   r7  regular_columns_allregular_columnsrf   execute_many)rC   rG  rH  instance_chunkvalues_lists_allvalues_listsrA   rL  rD   r0    s.   
	zBulkCreateQuery._execute_manyGenerator[Any, None, None]c                 C  s&   |  d |  \}}| || S r  r2  )rC   rG  rH  rA   rA   rD   rE     s   
zBulkCreateQuery.__await__c                 C  sR   |    |  \}}tdd | jD r|S tdd | jD r"|S d||gS )Nc                 s  s    | ]}|j V  qd S r@   rO  r~   orA   rA   rD   r     s    z&BulkCreateQuery.sql.<locals>.<genexpr>c                 s  s    | ]}|j  V  qd S r@   rW  rX  rA   rA   rD   r   !  r   r4  )ru   r.  rB  r  r   )rC   r   rG  rH  rA   rA   rD   r     s   zBulkCreateQuery.sqlr  )rd   rk   rn   r   rW  rX  rY  rZ  r[  rX   r\  r]  r^  r]  )r=   r:  )rG  rM   rH  rM   r=   rl   )r=   rV  r   r   r5  rA   rA   r   rD   r`    s    


r`  )[
__future__r   r  collectionsr   collections.abcr   r   r   r   r   r	   typingr
   r   r   r   r   r   r   pypika_tortoiser   r   r   pypika_tortoise.analyticsr   pypika_tortoise.functionsr   pypika_tortoise.queriesr   pypika_tortoise.termsr   r   r   r   r   typing_extensionsr   r   tortoise.backends.base.clientr   r   tortoise.exceptionsr    r!   r"   r#   r$   tortoise.expressionsr%   r&   r'   r(   r)   tortoise.fields.relationalr*   r+   r,   tortoise.filtersr-   tortoise.query_utilsr.   r/   r0   r1   r2   tortoise.routerr3   tortoise.utilsr4   r5   __annotations__r#  r6   r7   r9   rX   r;   r<   rb   r   r<  r8  r@  r>  r  r1  r5  rC  rd  r`  rA   rA   rA   rD   <module>   sb    $ d       :O11?p  \