OwlCyberSecurity - MANAGER
Edit File: query.cpython-39.pyc
a ���e�� � @ s4 d Z ddlZddlZddlZddlmZmZ ddlmZ ddl Z ddl m Z ddl m Z ddlmZmZmZmZmZmZ ddlmZmZmZmZmZ dd lm Z m Z dd l m Z dd l m!Z!m"Z"m#Z#m$Z$ dd l%m&Z&m'Z' dd l(m)Z)m*Z* ddl+m,Z,m-Z- ddl.m/Z/m0Z0m1Z1 ddl2m3Z3 ddl4m5Z5 ddl6m7Z7m8Z8 dZ9dZ:G dd� d�Z;G dd� de;�Z<G dd� de;�Z=G dd � d e;�Z>G d d � d e;�Z?G d d � d e?�Z@G d!d"� d"e;�ZAG d#d$� d$e/�ZBG d%d&� d&eC�ZDG d'd(� d(eDd)�ZEG d*d+� d+�ZFG d,d-� d-�ZGd:d.d/�ZHd0d1� ZId2d3� ZJd4d5� ZKG d6d7� d7�ZLd8d9� ZMdS );zM The main QuerySet implementation. This provides the public API for the ORM. � N)�chain�islice)� sync_to_async)�settings)� exceptions)�DJANGO_VERSION_PICKLE_KEY�IntegrityError�NotSupportedError� connections�router� transaction)� AutoField� DateField� DateTimeField�Field�sql)� LOOKUP_SEP� OnConflict)� Collector)�Case�F�Value�When)�Cast�Trunc)�FilteredRelation�Q)�CURSOR�GET_ITERATOR_CHUNK_SIZE)� AltersData�create_namedtuple_class�resolve_callables)�timezone)�RemovedInDjango50Warning)�cached_property� partition� � c @ s* e Zd Zdefdd�Zdd� Zdd� ZdS ) � BaseIterableFc C s || _ || _|| _d S �N)�queryset� chunked_fetch� chunk_size)�selfr* r+ r, � r. �W/home/gouroczh/virtualenv/pat/3.9/lib/python3.9/site-packages/django/db/models/query.py�__init__/ s zBaseIterable.__init__c sL � � � }� fdd�}t|�|�I d H }|D ] }|V q*t|�� jk rqHqd S )Nc s t t| � j��S r) )�listr r, )�gen�r- r. r/ � next_slice< s z1BaseIterable._async_generator.<locals>.next_slice)�__iter__r �lenr, )r- Zsync_generatorr4 �chunk�itemr. r3 r/ �_async_generator6 s z BaseIterable._async_generatorc C s | � � S r) )r9 r3 r. r. r/ � __aiter__N s zBaseIterable.__aiter__N)�__name__� __module__� __qualname__r r0 r9 r: r. r. r. r/ r( . s � r( c @ s e Zd ZdZdd� ZdS )� ModelIterablez3Iterable that yields a model instance for each row.c # sj | j � � j}� jj|d�}|j| j| jd�}|j|j|j }}}|d }|d }|d |d d } } dd � || | � D �} t |||�} � fd d �� j � � D �} |� |�D ]�}|�|| || | � �}| D ]}|�||� q�|�r|� � D ]\}}t|||| � q�| D ]T\}}}|�|��r"�q||�}z || }W n t�yJ Y n0 t||j|� �q|V q�d S ) N��using�r+ r, �model� select_fieldsr ���� c S s g | ]}|d j j�qS �r ��target�attname��.0�fr. r. r/ � <listcomp>f s z*ModelIterable.__iter__.<locals>.<listcomp>c s2 g | ]*\� }� |t j� �fd d�� jD �� f�qS )c s* g | ]"}|d kr� j n�jj�|�j �qS r3 )rI rB �_meta� get_field)rK Z from_field)�fieldr* r. r/ rM o s ��z5ModelIterable.__iter__.<locals>.<listcomp>.<listcomp>)�operator� attrgetterZ from_fields)rK Z related_objs�r* )rP r/ rM j s � ���)r* �db�query� get_compiler� execute_sqlr+ r, �select� klass_info�annotation_col_map�get_related_populators�_known_related_objects�items� results_iter�from_db�populate�setattr� is_cached�KeyError�name)r- rT �compiler�resultsrX rY rZ � model_clsrC Zmodel_fields_startZmodel_fields_end� init_list�related_populatorsZknown_related_objects�row�objZ rel_populator� attr_nameZcol_posrP Zrel_objsZ rel_getterZ rel_obj_id�rel_objr. rS r/ r5 U sN �� � �� zModelIterable.__iter__N�r; r< r= �__doc__r5 r. r. r. r/ r> R s r> c @ s e Zd ZdZdd� ZdS )�RawModelIterablezQ Iterable that yields a model instance for each row from a raw queryset. c # s8 � j j}� j j}t| }|j�d�|||�}t|�}zވ j �� \}}}� j j} | j j j |vrft � d��� fdd�� j jD �} |�dd� | D ��} | r�|�|| �}|D ]L��fdd�|D �} | �||| �} |r�|D ]\}}t| |�| � q�| V q�W t|d��r4|j�r4|j�� n t|d��r2|j�r2|j�� 0 d S )NZ SQLCompilerz&Raw query must include the primary keyc s g | ]}� j j�|��qS r. )r* � model_fields�get)rK �cr3 r. r/ rM � � z-RawModelIterable.__iter__.<locals>.<listcomp>c S s$ g | ] }|r |� |jjj�nd �qS r) )Zget_colrB rN Zdb_tablerJ r. r. r/ rM � rt c s g | ] }� | �qS r. r. )rK �pos)�valuesr. r/ rM � rt �cursor)r* rT rU r �opsre �iter�resolve_model_init_orderrB rN �pkrI r �FieldDoesNotExist�columnsZget_convertersZapply_convertersr_ ra �hasattrrw �close)r- rT rU � connectionre Zquery_iterator�model_init_namesZmodel_init_pos�annotation_fieldsrg �fields� convertersZmodel_init_values�instance�columnru r. )r- rv r/ r5 � sB �� � �zRawModelIterable.__iter__Nrn r. r. r. r/ rp � s rp c @ s e Zd ZdZdd� ZdS )�ValuesIterablezQ Iterable returned by QuerySet.values() that yields a dict for each row. c # sl | j }|j}|�|j�}g |j�|j�|j�� tt� ��}|j | j | j d�D ]�� �fdd�|D �V qLd S )NrA c s i | ]}� | �| �qS r. r. �rK �i��namesrj r. r/ � <dictcomp>� rt z+ValuesIterable.__iter__.<locals>.<dictcomp>) r* rU rV rT � extra_select� values_select�annotation_select�ranger6 r^ r+ r, )r- r* rU re Zindexesr. r� r/ r5 � s ��� � zValuesIterable.__iter__Nrn r. r. r. r/ r� � s r� c @ s e Zd ZdZdd� ZdS )�ValuesListIterableze Iterable returned by QuerySet.values_list(flat=False) that yields a tuple for each row. c s� | j ��j}|��j�}�jr�g |j�|j�|j�}g �j��fdd�|jD ��}||kr�dd� t|�D �� t j � fdd�|D �� }t ||j | j | jd��S |j d| j | jd �S ) Nc 3 s | ]}|� j vr|V qd S r) )�_fieldsrJ rS r. r/ � <genexpr>� rt z.ValuesListIterable.__iter__.<locals>.<genexpr>c S s i | ]\}}||�qS r. r. )rK �idxrd r. r. r/ r� � rt z/ValuesListIterable.__iter__.<locals>.<dictcomp>c s g | ] }� | �qS r. r. rJ )� index_mapr. r/ rM � rt z/ValuesListIterable.__iter__.<locals>.<listcomp>rA T)Ztuple_expectedr+ r, )r* rU rV rT r� r� r� r� � enumeraterQ � itemgetter�mapr^ r+ r, )r- rU re r� r� Z rowfactoryr. )r� r* r/ r5 � s< ��������zValuesListIterable.__iter__Nrn r. r. r. r/ r� � s r� c s e Zd ZdZ� fdd�Z� ZS )�NamedValuesListIterablezj Iterable returned by QuerySet.values_list(named=True) that yields a namedtuple for each row. c # s^ | j }|jr|j}n |j}g |j�|j�|j�}t|� }tj}t � � � D ]}|||�V qHd S r) ) r* r� rU r� r� r� r �tuple�__new__�superr5 )r- r* r� rU Z tuple_class�newrj �� __class__r. r/ r5 s ���z NamedValuesListIterable.__iter__)r; r< r= ro r5 � __classcell__r. r. r� r/ r� � s r� c @ s e Zd ZdZdd� ZdS )�FlatValuesListIterablez] Iterable returned by QuerySet.values_list(flat=True) that yields single values. c c s: | j }|j�|j�}|j| j| jd�D ]}|d V q&d S )NrA r )r* rU rV rT r^ r+ r, )r- r* re rj r. r. r/ r5 s � z FlatValuesListIterable.__iter__Nrn r. r. r. r/ r� s r� c @ sV e Zd ZdZd�dd�Zedd� �Zejdd� �Zdd � Zd e_ e e�Zd d � Z d d� Z dd� Z dd� Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd�d'd(�Zd�d*d+�Zd,d-� Zd.d/� Z d0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z!d:d;� Z"d<d=� Z#d>d?� Z$d�dAdB�Z%d�dCdD�Z&d�dEdF�Z'd e'_(d�dGdH�Z)d e)_(d�dIdJ�Z*d�dKdL�Z+d�dMdN�Z,d�dOdP�Z-dQdR� Z.dSdT� Z/dUdV� Z0dWdX� Z1dYdZ� Z2d[d\� Z3d]d^� Z4d_d`� Z5dadb� Z6dcdd� Z7d�dedf�dgdh�Z8d�dedf�didj�Z9dkdl� Z:d e:_(d e:_ dmdn� Z;d e;_(d e;_ dodp� Z<d e<_(dqdr� Z=d e=_(dsdt� Z>d e>_(dudv� Z?d e?_(d@e?_ dwdx� Z@dydz� ZAd{d|� ZBd}d~� ZCdd�� ZDdd��d�d��ZEdd��d�d��ZFd�d�d��ZGd�d�� ZHd�d�� ZId@d@d��d�d��ZJd�d�d��ZKd�deLjMfd�d��ZNd�d�� ZOd�d�� ZPd�d�� ZQd�d�� ZRd�d�� ZSd�d�� ZTd�d�� ZUd@d��d�d��ZVd@d��d�d��ZWd�d�� ZXd�d�� ZYd�d�d��ZZd�d�� Z[d�d�� Z\d�d�� Z]d�d�� Z^d�d�d��Z_d�d�� Z`d�d�� Zad�d�d��Zbd�d�� Zcd�d�� Zdd�dÄ Zed�dń Zfed�dDŽ �Zged�dɄ �Zhd�d�d˄Zid ei_(d@ei_ d�d�d̈́Zjd�dτ Zkd�dф Zld�dӄ Zmd�dՄ Znd�dׄ Zod�dل Zpd�dۄ Zqd eq_ d�d݄ Zrd�d߄ Zsetd�d� �Zud�d� Zvd�d� Zwd�d� ZxdS )��QuerySetz6Represent a lazy database lookup for a set of objects.Nc C sh || _ || _|pi | _|p$t�| j �| _d | _d| _d| _d| _ d| _ i | _ t | _ d | _d| _d | _d S )NFr. )rB �_db�_hintsr ZQuery�_query� _result_cache�_sticky_filter� _for_write�_prefetch_related_lookups�_prefetch_doner\ r> �_iterable_classr� �_defer_next_filter�_deferred_filter)r- rB rU r@ �hintsr. r. r/ r0 & s zQuerySet.__init__c C s, | j r&| j \}}}| �|||� d | _ | jS r) )r� �_filter_or_exclude_inplacer� �r- �negate�args�kwargsr. r. r/ rU 6 s zQuerySet.queryc C s |j r t| _|| _d S r) )r� r� r� r� )r- �valuer. r. r/ rU > s c C s" ddl m} |�| �� }d|_|S )Nr )�ManagerT)Zdjango.db.models.managerr� Z from_querysetZ_built_with_as_manager)�clsr� �managerr. r. r/ � as_managerD s zQuerySet.as_managerTc C sF | � � }| j�� D ].\}}|dkr.d|j|< qt�||�|j|< q|S )z$Don't populate the QuerySet's cache.r� N)r� �__dict__r] �copy�deepcopy)r- �memork �k�vr. r. r/ � __deepcopy__S s zQuerySet.__deepcopy__c C s | � � i | j�ttji�S r) )� _fetch_allr� r �django� __version__r3 r. r. r/ � __getstate__] s zQuerySet.__getstate__c C sT |� t�}|r4|tjkrDtjd|tjf tdd� ntjdtdd� | j�|� d S )NzTPickled queryset instance's Django version %s does not match the current version %s.� )� stacklevelz<Pickled queryset instance's Django version is not specified.) rr r r� r� �warnings�warn�RuntimeWarningr� �update)r- �stateZpickled_versionr. r. r/ � __setstate__b s ���zQuerySet.__setstate__c C s8 t | d td � �}t|�tkr(d|d< d| jj|f S )NrE z$...(remaining elements truncated)...rD z<%s %r>)r1 �REPR_OUTPUT_SIZEr6 r� r; )r- �datar. r. r/ �__repr__u s zQuerySet.__repr__c C s | � � t| j�S r) �r� r6 r� r3 r. r. r/ �__len__{ s zQuerySet.__len__c C s | � � t| j�S )a� The queryset iterator protocol uses three nested iterators in the default case: 1. sql.compiler.execute_sql() - Returns 100 rows at time (constants.GET_ITERATOR_CHUNK_SIZE) using cursor.fetchmany(). This part is responsible for doing some column masking, and returning the rows in chunks. 2. sql.compiler.results_iter() - Returns one row at time. At this point the rows are still just tuples. In some cases the return values are converted to Python values at this location. 3. self.iterator() - Responsible for turning the rows into model objects. �r� ry r� r3 r. r. r/ r5 s zQuerySet.__iter__c s � fdd�}|� S )Nc s( t � j�� I d H � jD ] } | V qd S r) �r r� r� �r8 r3 r. r/ � generator� s z%QuerySet.__aiter__.<locals>.generatorr. �r- r� r. r3 r/ r: � s zQuerySet.__aiter__c C s | � � t| j�S r) �r� �boolr� r3 r. r. r/ �__bool__� s zQuerySet.__bool__c C s t |ttf�s tdt|�j ��t |t�r2|dk sdt |t�rl|jdurP|jdk sd|jdurl|jdk rltd��| j dur�| j | S t |t�r�| � � }|jdur�t|j�}nd}|jdur�t|j�}nd}|j � ||� |j r�t|�dd|j � S |S | � � }|j � ||d � |�� |j d S )z2Retrieve an item or slice from the set of results.z4QuerySet indices must be integers or slices, not %s.r Nz#Negative indexing is not supported.rE )� isinstance�int�slice� TypeError�typer; �start�stop� ValueErrorr� �_chainrU � set_limits�stepr1 r� )r- r� �qsr� r� r. r. r/ � __getitem__� sD ������� zQuerySet.__getitem__c O s | S r) r. )r� r� r� r. r. r/ �__class_getitem__� s zQuerySet.__class_getitem__c C sZ | � |d� | �|� t|t�r$|S t| t�r2| S | �� }|�|� |j�|jtj � |S )N�&) �_check_operator_queryset�_merge_sanity_checkr� � EmptyQuerySetr� � _merge_known_related_objectsrU �combiner �AND)r- �other�combinedr. r. r/ �__and__� s zQuerySet.__and__c C s� | � |d� | �|� t| t�r$|S t|t�r2| S | j�� r@| n| jjj| � d�d�}|� � }|� |� |j�� s�|jjj|� d�d�}|j� |jt j� |S )N�|r{ �Zpk__in)r� r� r� r� rU � can_filterrB � _base_manager�filterrv r� r� r� r �OR�r- r� rU r� r. r. r/ �__or__� s �� zQuerySet.__or__c C s� | � |d� | �|� t| t�r$|S t|t�r2| S | j�� r@| n| jjj| � d�d�}|� � }|� |� |j�� s�|jjj|� d�d�}|j� |jt j� |S )N�^r{ r� )r� r� r� r� rU r� rB r� r� rv r� r� r� r ZXORr� r. r. r/ �__xor__� s �� zQuerySet.__xor__c c sl | j | ||pdd�}| jr"|d u r0|E d H d S t|�}tt||�� }rht|g| j�R � |E d H q8d S )N�� rA )r� r� ry r1 r �prefetch_related_objects)r- �use_chunked_fetchr, �iterable�iteratorrf r. r. r/ � _iterator� s � zQuerySet._iteratorc C sP |du r | j r0tjdtdd� n|dkr0td��t| j j�d� }| � ||�S )z� An iterator over the results from applying this QuerySet to the database. chunk_size must be provided for QuerySets that prefetch related objects. Otherwise, a default chunk_size of 2000 is supplied. Nz_Using QuerySet.iterator() after prefetch_related() without specifying chunk_size is deprecated.r� )�categoryr� r �%Chunk size must be strictly positive.�DISABLE_SERVER_SIDE_CURSORS) r� r� r� r# r� r rT � settings_dictrr r )r- r, r� r. r. r/ r� s � �zQuerySet.iteratorr� c C s\ | j rtd��|dkr td��t| j j�d� }| j| ||d�2 z3 dH W }|V qB6 dS )zp An asynchronous iterator over the results from applying this QuerySet to the database. zEUsing QuerySet.aiterator() after prefetch_related() is not supported.r r r rA N)r� r r� r rT r rr r� )r- r, r� r8 r. r. r/ � aiterator& s � ��zQuerySet.aiteratorc O s~ | j jrtd��| jg |�|�� �R dd� |D ]8}z |j W n ttfy\ td��Y n0 |||j< q0| j �� � | j |�S )z� Return a dictionary containing the calculations (aggregation) over the current queryset. If args is present the expression is passed as a kwarg using the Aggregate object's default alias. z/aggregate() + distinct(fields) not implemented.� aggregate�� method_namez#Complex aggregates require an alias) rU �distinct_fields�NotImplementedError� _validate_values_are_expressionsrv � default_alias�AttributeErrorr� r Zget_aggregationrT )r- r� r� �argr. r. r/ r 9 s � zQuerySet.aggregatec � s t | j�|i |��I d H S r) )r r �r- r� r� r. r. r/ � aaggregateR s zQuerySet.aaggregatec C s$ | j durt| j �S | jj| jd�S )z� Perform a SELECT COUNT() and return the number of records as an integer. If the QuerySet is already fully cached, return the length of the cached results set to avoid multiple SELECT COUNT(*) calls. Nr? )r� r6 rU Z get_countrT r3 r. r. r/ �countU s zQuerySet.countc � s t | j�� I d H S r) )r r r3 r. r. r/ �acountb s zQuerySet.acountc O s� | j jr |s|r td| j j ��| j jr0| �� n| j|i |��}| j �� rZ| j jsZ|�� }d}|j jrtt |j j j r�t }|j j|d� t|�}|dkr�|jd S |s�| j�d| jjj ��| j�d| jjj|r�||k r�|n d|d f ��dS ) zl Perform the query and return a single object matching the given keyword arguments. zCCalling QuerySet.get(...) with filters after %s() is not supported.N��highrE r z!%s matching query does not exist.z2get() returned more than one %s -- it returned %s!z more than %s)rU � combinatorr r� r� r� r �order_by�select_for_updater rT �featuresZ%supports_select_for_update_with_limit�MAX_GET_RESULTSr� r6 r� rB � DoesNotExistrN � object_nameZMultipleObjectsReturned)r- r� r� �clone�limit�numr. r. r/ rr e s< �� � � ����z QuerySet.getc � s t | j�|i |��I d H S r) )r rr r r. r. r/ �aget� s z QuerySet.agetc K s* | j f i |��}d| _|jd| jd� |S )z� Create a new object with the given kwargs, saving it to the database and returning the created object. T)Z force_insertr@ )rB r� �saverT )r- r� rk r. r. r/ �create� s zQuerySet.createc � s t | j�f i |��I d H S r) )r r! �r- r� r. r. r/ �acreate� s zQuerySet.acreatec C s4 |D ]*}|j d u r"|jj �|�|_ |jdd� qd S )N� bulk_create)�operation_name)r{ rN Zget_pk_value_on_save� _prepare_related_fields_for_save)r- �objsrk r. r. r/ �_prepare_for_bulk_create� s z!QuerySet._prepare_for_bulk_createc C s� |r|rt d��t| j j}|r4|js.td��tjS |r�|jsFtd��|sRt d��|rd|j sdtd��|sv|j rvt d��t dd� |D ��r�t d ��t d d� |D ��r�t d ��|r�t d d� |D ��r�t d ��tj S d S )Nz=ignore_conflicts and update_conflicts are mutually exclusive.z:This database backend does not support ignoring conflicts.z:This database backend does not support updating conflicts.zUFields that will be updated when a row insertion fails on conflicts must be provided.ztThis database backend does not support updating conflicts with specifying unique fields that can trigger the upsert.z;Unique fields that can trigger the upsert must be provided.c s s | ]}|j p|jV qd S r) �ZconcreteZ many_to_manyrJ r. r. r/ r� � rt z6QuerySet._check_bulk_create_options.<locals>.<genexpr>zEbulk_create() can only be used with concrete fields in update_fields.c s s | ] }|j V qd S r) �� primary_keyrJ r. r. r/ r� � rt z@bulk_create() cannot be used with primary keys in update_fields.c s s | ]}|j p|jV qd S r) r) rJ r. r. r/ r� � rt zEbulk_create() can only be used with concrete fields in unique_fields.) r� r rT r Zsupports_ignore_conflictsr r ZIGNOREZsupports_update_conflictsZ%supports_update_conflicts_with_target�anyZUPDATE)r- �ignore_conflicts�update_conflicts� update_fields� unique_fieldsZ db_featuresr. r. r/ �_check_bulk_create_options� sV � ��� � ����z#QuerySet._check_bulk_create_optionsFc s& |dur|dkrt d���jj�� D ] }|jj�jjjur$t d��q$|sL|S �jj� |rl� �fdd�|D �}|r��fdd�|D �}��||||�}d�_� j} t|�}�� |� t j �j d d ���N t d d � |�\} } | �rV�j| | ||||d �} t| | �D ]:\} }t|� j�D ]$\}}|� jk�rt| |j|� �qq�| D ]} d | j_�j | j_ �q<| �rdd� | D �} �j| | ||||d �} t�j }|jj�r�|du �r�t| �t| �k�s�J �t| | �D ]B\}}t|� j�D ]\}}t||j|� �q�d |j_�j |j_ �q�W d � n1 �s0 Y |S )ab Insert each of the instances into the database. Do *not* call save() on each of the instances, do not send any pre/post_save signals, and do not set the primary key attribute if it is an autoincrement field (except if features.can_return_rows_from_bulk_insert=True). Multi-table models are not supported. Nr �&Batch size must be a positive integer.z/Can't bulk create a multi-table inherited modelc s* g | ]"}�j j�|d kr � jjn|��qS �r{ )rB rN rO r{ rd �rK rd ��optsr- r. r/ rM � s �z(QuerySet.bulk_create.<locals>.<listcomp>c s g | ]}� j j�|��qS r. �rB rN rO r4 r3 r. r/ rM rt TF�r@ Z savepointc S s | j d u S r) r3 )�or. r. r/ �<lambda> rt z&QuerySet.bulk_create.<locals>.<lambda>�� on_conflictr/ r0 c S s g | ]}t |t�s|�qS r. )r� r rJ r. r. r/ rM " rt )r� rB rN Zget_parent_list�concrete_modelr1 r� �concrete_fieldsr1 r( r �atomicrT r% �_batched_insert�zip�db_returning_fieldsr{ ra rI �_stateZaddingr r � can_return_rows_from_bulk_insertr6 )r- r' � batch_sizer- r. r/ r0 �parentr< r� Z objs_with_pkZobjs_without_pkZreturned_columnsZ obj_with_pkrf �resultrP r� Zobj_without_pkr. r5 r/ r$ � s� �� � � ��.zQuerySet.bulk_createc � s t | j�||||||d�I d H S )N)r' rE r- r. r/ r0 )r r$ )r- r' rE r- r. r/ r0 r. r. r/ � abulk_create9 s �zQuerySet.abulk_createc s( � dur� dkrt d��|s$t d��t���tdd� �D ��rFt d���fdd �|D �}td d� |D ��rrt d ��td d� |D ��r�t d ���s�dS �D ]}|jd|d� q�d�_t�j }|j�ddg| ��}� r�t � |�n|� |j j }� �fdd�t dt ��� �D �}g } |D ]�} i } |D ]x} g } | D ]>}t|| j�}t|d��sTt|| d�}| �t|j|d�� �q,t| d| i�}|�r�t|| d�}|| | j< �q | �dd � | D �| f� �qd}���j�}tj�jdd��< | D ]&\}} ||j|d�jf i | ��7 }�q�W d � n1 �s0 Y |S )zW Update the given fields in each of the given objects in the database. Nr r2 z+Field names must be given to bulk_update().c s s | ]}|j d u V qd S r) r3 �rK rk r. r. r/ r� T rt z'QuerySet.bulk_update.<locals>.<genexpr>z6All bulk_update() objects must have a primary key set.c s g | ]}� j j�|��qS r. r7 r4 r3 r. r/ rM V rt z(QuerySet.bulk_update.<locals>.<listcomp>c s s | ]}|j p|jV qd S r) r) rJ r. r. r/ r� W rt z4bulk_update() can only be used with concrete fields.c s s | ] }|j V qd S r) r* rJ r. r. r/ r� Y rt z5bulk_update() cannot be used with primary key fields.� bulk_update)r% r� Tr{ c 3 s | ]}�||� � V qd S r) r. r� �rE r' r. r/ r� h rt �resolve_expression�� output_field)r{ ZthenrN c S s g | ] }|j �qS r. r3 rI r. r. r/ rM w rt Fr8 r� ) r� r� r, r& r� r rT rx �bulk_batch_size�minr Z requires_casted_case_in_updatesr� r6 �getattrrI r~ r �appendr r{ r r r@ r r? r� r� )r- r' r� rE rk r� �max_batch_sizeZrequires_castingZbatchesZupdatesZ batch_objsZ update_kwargsrP Zwhen_statements�attrZcase_statementZ rows_updatedr* Zpksr. )rE r' r- r/ rJ K sZ � @zQuerySet.bulk_updatec � s t | j�|||d�I d H S )N)r' r� rE )r rJ )r- r' r� rE r. r. r/ � abulk_update� s �zQuerySet.abulk_updatec K s� d| _ z| jf i |��dfW S | jjy� | j|fi |��}zXtj| jd��6 tt |��}| j f i |��dfW d � W Y S 1 s�0 Y W nJ t y� z | jf i |��dfW Y Y S | jjy� Y n0 � Y n0 Y n0 dS )z� Look up an object with the given kwargs, creating one if necessary. Return a tuple of (object, created), where created is a boolean specifying whether an object was created. TFr? N) r� rr rB r �_extract_model_paramsr r? rT �dictr! r! r )r- �defaultsr� �paramsr. r. r/ � get_or_create� s < zQuerySet.get_or_createc � s t | j�f d|i|��I d H S �NrX )r rZ �r- rX r� r. r. r/ �aget_or_create� s ��zQuerySet.aget_or_createc K s |pi }d| _ tj| jd��� | �� j|fi |��\}}|rR||fW d � S t|�D ]\}}t|||� qZt|�}| j j j }|� |�r�| j j j D ]<} | js�| jjtju s�|�| j� | j| jkr�|�| j� q�|j| j|d� n|j| jd� W d � n1 �s 0 Y |dfS )z� Look up an object with the given kwargs, updating one with defaults if it exists, otherwise create a new one. Return a tuple (object, created), where created is a boolean specifying whether an object was created. Tr? N)r@ r/ F)r� r r? rT r rZ r! ra �setrB rN Z _non_pk_concrete_field_names� issupersetZlocal_concrete_fieldsr+ r� Zpre_saver �addrd rI r ) r- rX r� rk �createdr� r� r/ Zconcrete_field_namesrP r. r. r/ �update_or_create� s, � � .zQuerySet.update_or_createc � s t | j�f d|i|��I d H S r[ )r rb r\ r. r. r/ �aupdate_or_create� s ��zQuerySet.aupdate_or_createc K s� |pi }dd� |� � D �}|�|� | jjj}g }|D ]L}z| jj�|� W q6 tjy� ||v rrt| j|�j s||� |� Y q60 q6|r�t� d| jjj d� t|��f ��|S )z Prepare `params` for creating a model instance based on the given kwargs; for use by get_or_create(). c S s i | ]\}}t |vr||�qS r. )r )rK r� r� r. r. r/ r� � rt z2QuerySet._extract_model_params.<locals>.<dictcomp>z)Invalid field name(s) for model %s: '%s'.z', ')r] r� rB rN Z_property_namesrO r r| rQ �fsetrR � FieldErrorr �join�sorted)r- rX r� rY Zproperty_namesZinvalid_params�paramr. r. r/ rV � s( ���z QuerySet._extract_model_paramsc G sx |r |}n&t | jjd�}|r0t|ttf�s0|f}|du r@td��| �� }|jj dd� |jj dd� |jj |� |� � S )zy Return the earliest object according to fields (if given) or by the model's Meta.get_latest_by. Z get_latest_byNzmearliest() and latest() require either fields as positional arguments or 'get_latest_by' in the model's Meta.rE r T��force) rQ rB rN r� r� r1 r� r� rU r� �clear_ordering� add_orderingrr )r- r� r rk r. r. r/ � _earliest� s � zQuerySet._earliestc G s | j jrtd��| j|� S )N�2Cannot change a query once a slice has been taken.)rU � is_slicedr� rm �r- r� r. r. r/ �earliest s zQuerySet.earliestc � s t | j�|� I d H S r) )r rq rp r. r. r/ � aearliest s zQuerySet.aearliestc G s | j jrtd��| �� j|� S )zw Return the latest object according to fields (if given) or by the model's Meta.get_latest_by. rn )rU ro r� �reverserm rp r. r. r/ �latest s zQuerySet.latestc � s t | j�|� I d H S r) )r rt rp r. r. r/ �alatest s zQuerySet.alatestc C s> | j r | }n| jdd� | �d�}|dd� D ] }| S dS )z@Return the first object of a query or None if no match is found.�first��methodr{ NrE )�ordered�/_check_ordering_first_last_queryset_aggregationr �r- r* rk r. r. r/ rv s zQuerySet.firstc � s t | j�� I d H S r) )r rv r3 r. r. r/ �afirst$ s zQuerySet.afirstc C sB | j r| �� }n| jdd� | �d�}|dd� D ] }| S dS )z?Return the last object of a query or None if no match is found.�lastrw z-pkNrE )ry rs rz r r{ r. r. r/ r} ' s z QuerySet.lastc � s t | j�� I d H S r) )r r} r3 r. r. r/ �alast1 s zQuerySet.alastr{ �� field_namec s | j jrtd��| jj}dd� |jD �}� dkr^|�� �js^� |vr^| j j� fkr^t d� ��|du�r|spi S d� � �}t | j j j}t|�}|r�|t|�k r�d}td t|�|�D ]4}|||| � } |t| jf i || i���� �7 }q�n| jf i ||i���� }n| �� }� fd d �|D �S ) z� 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. z.Cannot use 'limit' or 'offset' with in_bulk().c S s$ g | ] }t |j�d kr|jd �qS )rE r )r6 r� )rK � constraintr. r. r/ rM < s �z$QuerySet.in_bulk.<locals>.<listcomp>r{ z;in_bulk()'s field_name must be a unique field but %r isn't.Nz{}__inr. r c s i | ]}t |� �|�qS r. )rQ rI r r. r/ r� \ rt z$QuerySet.in_bulk.<locals>.<dictcomp>)rU ro r� rB rN Ztotal_unique_constraintsrO �uniquer r� �formatr rT r Zmax_query_paramsr� r6 r� r� r r� ) r- �id_listr� r6 r0 Z filter_keyrE r� �offset�batchr. r r/ �in_bulk4 sB �� �� ��� $zQuerySet.in_bulkc � s t | j�||d�I d H S )N)r� r� )r r� )r- r� r� r. r. r/ �ain_bulk^ s �zQuerySet.ain_bulkc C s� | � d� | jjrtd��| jjs*| jjr2td��| jdurDtd��| �� }d|_d|j_ d|j_ |jj dd� t |j | d �}|�|� |�� \}}d| _||fS ) z+Delete the records in the current QuerySet.�deletez-Cannot use 'limit' or 'offset' with delete().z'Cannot call delete() after .distinct().Nz6Cannot call delete() after .values() or .values_list()TFri )r@ �origin)� _not_support_combined_queriesrU ro r� �distinctr r� r� r� r �select_relatedrk r rT Zcollectr� r� )r- Z del_query� collectorZdeletedZ _rows_countr. r. r/ r� d s" zQuerySet.deletec � s t | j�� I d H S r) )r r� r3 r. r. r/ �adelete� s zQuerySet.adeletec C sT | j �� }tj|_|�|��t�}|rP|� |jW d � S 1 sF0 Y dS )z� Delete objects found from the given queryset in single direct SQL query. No signals are sent and there is no protection for cascades. Nr ) rU r r Z DeleteQueryr� rV rW r Zrowcount)r- r@ rU rw r. r. r/ � _raw_delete� s $zQuerySet._raw_deletec K s� | � d� | jjrtd��d| _| j�tj�}|�|� g }|j D ]F}|j � |� }r~t |dd�rrt �d|� ���|�|� qB|�|� qBt|�|_ i |_ tj| jd��" |�| j��t�}W d � n1 s�0 Y d| _|S ) z~ Update all elements in the current QuerySet, setting all the given fields to the appropriate values. r� �2Cannot update a query once a slice has been taken.T�contains_aggregateFz-Cannot update when ordering by an aggregate: r? N)r� rU ro r� r� r r � UpdateQueryZadd_update_valuesr � annotationsrr rQ r re rR r� r Zmark_for_rollback_on_errorrT rV rW r r� )r- r� rU Z new_order_by�col� annotation�rowsr. r. r/ r� � s* � 0zQuerySet.updatec � s t | j�f i |��I d H S r) )r r� r" r. r. r/ �aupdate� s zQuerySet.aupdatec C sF | j jrtd��| j �tj�}|�|� i |_d| _|� | j �� t �S )a A version of update() that accepts field objects instead of field names. Used primarily for model saving and not intended for use by general code (it requires too much poking around at model internals to be useful at that level). r� N) rU ro r� r r r� Zadd_update_fieldsr� r� rV rT rW r )r- rv rU r. r. r/ �_update� s zQuerySet._updatec C s$ | j du r| jj| jd�S t| j �S )zV Return True if the QuerySet would have any results, False otherwise. Nr? )r� rU Z has_resultsrT r� r3 r. r. r/ �exists� s zQuerySet.existsc � s t | j�� I d H S r) )r r� r3 r. r. r/ �aexists� s zQuerySet.aexistsc C s� | � d� | jdur td��z |jj| jjjkr6W dS W n tyR td��Y n0 |jdu rftd��| j durz|| j v S | j |jd�� � S )za Return True if the QuerySet contains the provided obj, False otherwise. �containsNzBCannot call QuerySet.contains() after .values() or .values_list().Fz 'obj' must be a model instance.z6QuerySet.contains() cannot be used on unsaved objects.r3 ) r� r� r� rN r= rB r r{ r� r� r� r� �r- rk r. r. r/ r� � s � zQuerySet.containsc � s t | j�|d�I d H S )N)rk )r r� r� r. r. r/ � acontains� s zQuerySet.acontainsc C s t | jg| j�R � d| _d S �NT�r� r� r� r� r3 r. r. r/ �_prefetch_related_objects� s z"QuerySet._prefetch_related_objects)r� c K s | j jf | j|d�|��S )zp Runs an EXPLAIN on the SQL query this QuerySet would perform, and returns the results. )r@ r� )rU �explainrT �r- r� �optionsr. r. r/ r� � s zQuerySet.explainc � s t | j�f d|i|��I d H S )Nr� )r r� r� r. r. r/ �aexplain s zQuerySet.aexplainr. c C s6 |d u r| j }t|| j|||d�}| jd d � |_|S )N)rB rY � translationsr@ )rT � RawQuerySetrB r� )r- � raw_queryrY r� r@ r� r. r. r/ �raw s �z QuerySet.rawc O s2 | � � }|r |jf i |��}||_|j�|� |S r) )r� �annotater� rU Z set_values�r- r� � expressionsr r. r. r/ �_values s zQuerySet._valuesc O s& |t |�7 }| j|i |��}t|_|S r) )r� r� r� r� r� r. r. r/ rv s zQuerySet.values)�flat�namedc G s� |r|rt d��|r(t|�dkr(t d��dd� |D �}g }i }d}|D ]`}t|d�r�t|d|jj�� �} | t|� } |d7 }| |vrhq�qh||| < |�| � qF|�|� qF| j |i |��} |r�t n |r�t nt | _ | S )Nz*'flat' and 'named' can't be used together.rE zH'flat' is not valid when values_list is called with more than one field.c S s h | ]}t |d �s|�qS )rL �r~ rJ r. r. r/ � <setcomp>. rt z'QuerySet.values_list.<locals>.<setcomp>rL r )r� r6 r~ rQ r� r; �lower�strrR r� r� r� r� r� ) r- r� r� r� � field_namesr� r� �counterrP Zfield_id_prefixZfield_idr r. r. r/ � values_list% s< � � ���zQuerySet.values_list�ASCc C sj |dvrt d��|dvr t d��| jt||t� d�t|�d�jddd ��� jd d ��|d kr`d ndd �S )z� Return a list of date objects representing all available dates for the given field_name, scoped to 'kind'. )�year�month�week�dayz8'kind' must be one of 'year', 'month', 'week', or 'day'.�r� �DESC�''order' must be either 'ASC' or 'DESC'.rM )� datefield� plain_fieldr� T�r� F�Zplain_field__isnullr� �-� ) r� r� r r r r� r� r� r )r- r� �kind�orderr. r. r/ �datesK s �� ���zQuerySet.datesc C s� |dvrt d��|dvr t d��tjr8|du r<t�� }nd}| jt||t� ||d�t|�d�j dd d �� � j d d �� |d kr�dndd �S )z� Return a list of datetime objects representing all available datetimes for the given field_name, scoped to 'kind'. )r� r� r� r� �hour�minute�secondzT'kind' must be one of 'year', 'month', 'week', 'day', 'hour', 'minute', or 'second'.r� r� N)rN �tzinfo�is_dst)� datetimefieldr� r� Tr� Fr� r� r� r� ) r� r ZUSE_TZr" Zget_current_timezoner� r r r r� r� r� r )r- r� r� r� r� r� r. r. r/ � datetimesa s6 � �� � � ��zQuerySet.datetimesc C s | � � }|j�� |S )zReturn an empty QuerySet.)r� rU Z set_empty�r- r r. r. r/ �none� s z QuerySet.nonec C s | � � S )z� Return a new QuerySet that is a copy of the current one. This allows a QuerySet to proxy for a model manager in some cases. )r� r3 r. r. r/ �all� s z QuerySet.allc O s | � d� | �d||�S )za Return a new QuerySet instance with the args ANDed to the existing set. r� F�r� �_filter_or_excluder r. r. r/ r� � s zQuerySet.filterc O s | � d� | �d||�S )zc Return a new QuerySet instance with NOT (args) ANDed to the existing set. �excludeTr� r r. r. r/ r� � s zQuerySet.excludec C sL |s|r| j jrtd��| �� }| jr:d| _|||f|_n|�|||� |S )Nz2Cannot filter a query once a slice has been taken.F)rU ro r� r� r� r� r� )r- r� r� r� r r. r. r/ r� � s zQuerySet._filter_or_excludec C s8 |r | j �t|i |�� � n| j �t|i |��� d S r) )r� �add_qr r� r. r. r/ r� � s z#QuerySet._filter_or_exclude_inplacec C s6 t |t�r"| �� }|j�|� |S | jdd|d�S dS )aB Return a new QuerySet instance with filter_obj added to the filters. filter_obj can be a Q object or a dictionary of keyword lookup arguments. This exists to support framework features such as 'limit_choices_to', and usually it will be more natural to use other methods. Fr. )r� r� N)r� r r� rU r� r� )r- Z filter_objr r. r. r/ �complex_filter� s zQuerySet.complex_filter)r� c G sR | � � }|jjdd� |j�� | jftdd� |D �� |j_||j_||j_|S )NTri c s s | ] }|j V qd S r) )rU )rK r� r. r. r/ r� � s z-QuerySet._combinator_query.<locals>.<genexpr>)r� rU rk Z clear_limitsr� Zcombined_queriesr Zcombinator_all)r- r r� �other_qsr r. r. r/ �_combinator_query� s �zQuerySet._combinator_queryc G sp t | t�rXdd� |D �}|s | S t|�dkr4|d S |d jdg|dd � �R d|i�S | jdg|�R d|i�S )Nc S s g | ]}t |t�s|�qS r. )r� r� )rK �qr. r. r/ rM � rt z"QuerySet.union.<locals>.<listcomp>rE r �unionr� )r� r� r6 r� )r- r� r� r� r. r. r/ r� � s $zQuerySet.unionc G s< t | t�r| S |D ]}t |t�r| S q| jdg|�R � S )N� intersection�r� r� r� )r- r� r� r. r. r/ r� � s zQuerySet.intersectionc G s t | t�r| S | jdg|�R � S )N� differencer� )r- r� r. r. r/ r� � s zQuerySet.differencec C sJ |r|rt d��| �� }d|_d|j_||j_||j_||j_||j_|S )zi Return a new QuerySet instance that will select objects with a FOR UPDATE lock. z2The nowait option cannot be used with skip_locked.T) r� r� r� rU r Zselect_for_update_nowaitZ select_for_update_skip_lockedZselect_for_update_ofZselect_for_no_key_update)r- ZnowaitZ skip_lockedZofZno_keyrk r. r. r/ r � s zQuerySet.select_for_updatec G sT | � d� | jdur td��| �� }|dkr6d|j_n|rH|j�|� nd|j_|S )a 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. If select_related(None) is called, clear the list. r� Nz>Cannot call select_related() after .values() or .values_list()r) FT)r� r� r� r� rU r� Zadd_select_related)r- r� rk r. r. r/ r� � s � zQuerySet.select_relatedc G sp | � d� | �� }|dkr"d|_nJ|D ]8}t|t�r:|j}|�td�d }|| jj v r&t d��q&|j| |_|S )aN Return a new QuerySet instance that will prefetch the specified Many-To-One and Many-To-Many related objects when the QuerySet is evaluated. When prefetch_related() is called more than once, append to the list of prefetch lookups. If prefetch_related(None) is called, clear the list. �prefetch_relatedr) r. rE r z:prefetch_related() is not supported with FilteredRelation.) r� r� r� r� �Prefetch� prefetch_to�splitr rU �_filtered_relationsr� )r- �lookupsr �lookupr. r. r/ r� s � zQuerySet.prefetch_relatedc O s | � d� | j||dd�S )z Return a query set in which the returned objects have been annotated with extra data or aggregations. r� T�rX �r� � _annotater r. r. r/ r� 0 s zQuerySet.annotatec O s | � d� | j||dd�S )zW Return a query set with added aliases for extra data or aggregations. �aliasFr� r� r r. r. r/ r� 8 s zQuerySet.aliasc C sP | j |t|�� � dd� i }|D ]F}z |j|v r@td|j ��W n ty\ td��Y n0 |||j< q"|�|� | �� }| j}|d u r�t t � dd� | j j �� D ���}|�� D ]H\}} ||v r�td| ��t| t�r�|j�| |� q�|jj| ||d� q�|jj�� D ]B\}} ||v �r| j�r|jd u �r8d |j_n |j�� �qL�q|S ) Nr� r zQThe named annotation '%s' conflicts with the default name for another annotation.z$Complex annotations require an aliasc s s, | ]$}t |d �r |j|jfn|jfV qdS )rI N)r~ rd rI )rK rP r. r. r/ r� U s ��z%QuerySet._annotate.<locals>.<genexpr>z8The annotation '%s' conflicts with a field on the model.r� T)r r� rv r r� r� r� r� r� r^ r � from_iterablerB rN Z get_fieldsr] r� r rU Zadd_filtered_relationZadd_annotationr� r� �group_byZ set_group_by) r- r� r� rX r� r r r� r� r� r. r. r/ r� ? sZ � �� �� �� � zQuerySet._annotatec G s8 | j jrtd��| �� }|j jddd� |j j|� |S )z9Return a new QuerySet instance with the ordering changed.z3Cannot reorder a query once a slice has been taken.TF)rj Z clear_default)rU ro r� r� rk rl �r- r� rk r. r. r/ r u s zQuerySet.order_byc G s2 | � d� | jjrtd��| �� }|jj|� |S )zX Return a new QuerySet instance that will select only distinct results. r� z:Cannot create distinct fields once a slice has been taken.)r� rU ro r� r� Zadd_distinct_fieldsr� r. r. r/ r� ~ s � zQuerySet.distinctc C s< | � d� | jjrtd��| �� }|j�||||||� |S )z%Add extra SQL fragments to the query.�extrarn )r� rU ro r� r� Z add_extra)r- rX �whererY Ztablesr Z select_paramsr r. r. r/ r� � s zQuerySet.extrac C s* | j jrtd��| �� }|j j |j _|S )z%Reverse the ordering of the QuerySet.z3Cannot reverse a query once a slice has been taken.)rU ro r� r� Zstandard_orderingr� r. r. r/ rs � s zQuerySet.reversec G sH | � d� | jdur td��| �� }|dkr8|j�� n |j�|� |S )a% Defer the loading of data for certain fields until they are accessed. Add the set of deferred fields to any existing set of deferred fields. The only exception to this is if None is passed in as the only parameter, in which case removal all deferrals. �deferNz5Cannot call defer() after .values() or .values_list()r) )r� r� r� r� rU Zclear_deferred_loadingZadd_deferred_loading)r- r� r r. r. r/ r� � s zQuerySet.deferc G sr | � d� | jdur td��|dkr,td��|D ](}|�td�d }|| jjv r0td��q0| �� }|j� |� |S ) z� Essentially, the opposite of defer(). Only the fields passed into this method and that are not already specified as deferred are loaded immediately when the queryset is evaluated. �onlyNz4Cannot call only() after .values() or .values_list()r) z*Cannot pass None as an argument to only().rE r z.only() is not supported with FilteredRelation.) r� r� r� r� r rU r� r� r� Zadd_immediate_loading)r- r� rP r r. r. r/ r� � s z QuerySet.onlyc C s | � � }||_|S )z;Select which database this QuerySet should execute against.)r� r� )r- r� r r. r. r/ r@ � s zQuerySet.usingc C sJ t | t�rdS | jjs | jjr"dS | jjrB| j�� jrB| jjsBdS dS dS )z� Return True if the QuerySet is ordered -- i.e. has an order_by() clause or a default ordering on the model (or is empty). TFN) r� r� rU Zextra_order_byr Zdefault_orderingZget_metaZorderingr� r3 r. r. r/ ry � s � ��zQuerySet.orderedc C s> | j r"| jp tj| jfi | j��S | jp<tj| jfi | j��S �z7Return the database used if this query is executed now.)r� r� r Z db_for_writerB r� � db_for_readr3 r. r. r/ rT � s z QuerySet.dbc C sJ d| _ |du r| j}tj| j|||d�} | j|||d� | j|d��|�S )z� Insert a new record for the given model. This provides an interface to the InsertQuery class and is how Model.save() is implemented. TNr; )r� r? )r� rT r Z InsertQueryrB Z insert_valuesrV rW ) r- r' r� �returning_fieldsr� r@ r< r/ r0 rU r. r. r/ �_insert� s �zQuerySet._insertc s� t | j }|j}t|�|��d�} � r0t� | �n| � g } |jj} � �fdd�tdt ��� �D �D ]J} | r�|du r�| � | j | || j| j j jd�� q`| j | || j|||d� q`| S )zU Helper method for bulk_create() to insert objs one batch at a time. rE c s g | ]}�||� � �qS r. r. r� rK r. r/ rM $ rt z,QuerySet._batched_insert.<locals>.<listcomp>r N)r� r@ r� )r� r@ r< r/ r0 )r rT rx �maxrO rP r rD r� r6 �extendr� rB rN rB ) r- r' r� rE r< r/ r0 r� rx rS Z inserted_rowsZ bulk_returnr8 r. rK r/ r@ s2 $ �� �zQuerySet._batched_insertc C s | � � }|jr d|j_d|_|S )zc Return a copy of the current QuerySet that's ready for another operation. TF)�_cloner� rU Zfilter_is_stickyr� r. r. r/ r� 9 s zQuerySet._chainc C sZ | j | j| j�� | j| jd�}| j|_| j|_| jdd� |_| j |_ | j |_ | j |_ |S )zi Return a copy of the current QuerySet. A lightweight alternative to deepcopy(). )rB rU r@ r� N) r� rB rU r r� r� r� r� r� r\ r� r� �r- rs r. r. r/ r� D s �zQuerySet._clonec C s2 | j d u rt| �| ��| _ | jr.| js.| �� d S r) )r� r1 r� r� r� r� r3 r. r. r/ r� W s zQuerySet._fetch_allc C s d| _ | S )a Indicate that the next filter call and the one following that should be treated as a single filter. This is only important when it comes to determining when to reuse tables for many-to-many filters. Required so that we can filter naturally on the results of related managers. This doesn't return a clone of the current QuerySet (it returns "self"). The method is only used internally and should be immediately followed by a filter() that does create a clone. T)r� r3 r. r. r/ �_next_is_sticky] s zQuerySet._next_is_stickyc C sf | j durbt| jj�t|jj�ksRt| jj�t|jj�ksRt| jj�t|jj�krbtd| jj ��dS )z.Check that two QuerySet classes may be merged.Nz?Merging '%s' classes must involve the same values in each case.) r� r^ rU r� r� r� r� r� r; �r- r� r. r. r/ r� k s �����z QuerySet._merge_sanity_checkc C s, |j �� D ] \}}| j �|i ��|� q dS )zX Keep track of all known related objects from either QuerySet instance. N)r\ r] � setdefaultr� )r- r� rP �objectsr. r. r/ r� w s z%QuerySet._merge_known_related_objectsc O s: | j r t| j �dkr td��| jj|i |��}| j|_|S )NrE z0Cannot use multi-field values as a filter value.)r� r6 r� rU rL r� )r- r� r� rU r. r. r/ rL ~ s zQuerySet.resolve_expressionc K s | j �|� dS )z} Update hinting information for use by routers. Add new key/values or overwrite existing key/values. N)r� r� )r- r� r. r. r/ � _add_hints� s zQuerySet._add_hintsc C s | j �� S )z� Check if this QuerySet has any filtering going on. This isn't equivalent with checking if all objects are present in results, for example, qs[1:]._has_filters() -> False. )rU Z has_filtersr3 r. r. r/ � _has_filters� s zQuerySet._has_filtersc C s0 t dd� | D ��}|r,td|d�|�f ��d S )Nc s s | ]}t |d �st|�V qdS )rL N)r~ r� )rK r r. r. r/ r� � s z<QuerySet._validate_values_are_expressions.<locals>.<genexpr>z-QuerySet.%s() received non-expression(s): %s.z, )rg r� rf )rv r Z invalid_argsr. r. r/ r � s ����z)QuerySet._validate_values_are_expressionsc C s | j jr td|| j jf ��d S )Nz2Calling QuerySet.%s() after %s() is not supported.)rU r r )r- r% r. r. r/ r� � s ��z&QuerySet._not_support_combined_queriesc C s$ | j js|j jr td|� d���d S )Nz Cannot use z! operator with combined queryset.)rU r r� )r- r� Z operator_r. r. r/ r� � s z!QuerySet._check_operator_querysetc s< t � jjt�r8t� fdd�� jjD ��s8td|� d���d S )Nc 3 s | ]}|j � jjju V qd S r) )rN rB rN r{ )rK r� r3 r. r/ r� � s zKQuerySet._check_ordering_first_last_queryset_aggregation.<locals>.<genexpr>zCannot use QuerySet.zT() on an unordered queryset performing aggregation. Add an ordering with order_by().)r� rU r� r� r, r� )r- rx r. r3 r/ rz � s � �z8QuerySet._check_ordering_first_last_queryset_aggregation)NNNN)N)r� )NFFNN)NFFNN)N)N)N)N)N)N)N)N)r. NN)r� )FFr. F)T)NNNNNN)NFNNNN)NNN)yr; r< r= ro r0 �propertyrU �setterr� Z queryset_only� classmethodr� r� r� r� r� r5 r: r� r� r� r� r� r� r r� r r r r r rr r r! r# r( r1 r$ rH rJ Z alters_datarU rZ r] rb rc rV rm rq rr rt ru rv r| r} r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� rv r� r� r" Z NOT_PASSEDr� r� r� r� r� r� r� r� r� r� r� r� r r� r� r� r� r� r r� r� rs r� r� r@ ry rT r� r@ r� r� r� r� r� r� rL r� r� � staticmethodr r� r� rz r. r. r. r/ r� # s< % # 9 � g � 4 & * & � $ 6 � � � ' r� c @ s e Zd Zdd� ZdS )�InstanceCheckMetac C s t |t�o|j�� S r) )r� r� rU Zis_empty)r- r� r. r. r/ �__instancecheck__� s z#InstanceCheckMeta.__instancecheck__N)r; r< r= r r. r. r. r/ r � s r c @ s e Zd ZdZdd� ZdS )r� zz Marker class to checking if a queryset is empty by .none(): isinstance(qs.none(), EmptyQuerySet) -> True c O s t d��d S )Nz#EmptyQuerySet can't be instantiated)r� r r. r. r/ r0 � s zEmptyQuerySet.__init__N)r; r< r= ro r0 r. r. r. r/ r� � s r� )� metaclassc @ s� e Zd ZdZd&dd�Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zd d � Zed d � �Zd d!� Zed"d#� �Zed$d%� �ZdS )'r� zo Provide an iterator which converts the results of raw SQL queries into annotated model instances. Nr. c C sZ || _ || _|| _|pi | _|p0tj|| j|d�| _|| _|p@i | _ d | _ d| _ d| _ d S )N)r r@ rY r. F) r� rB r� r� r ZRawQueryrT rU rY r� r� r� r� )r- r� rB rU rY r� r@ r� r. r. r/ r0 � s zRawQuerySet.__init__c sl t �j jj� � �fdd��jjjD �}�fdd�t�j�D �}� �fdd�|D �}dd� |D �}|||fS )z1Resolve the init field names and value positions.c s g | ]}� |j ��jv r|�qS r. )r� r} rJ �� converterr- r. r/ rM � s z8RawQuerySet.resolve_model_init_order.<locals>.<listcomp>c s" g | ]\}}|� j vr||f�qS r. )rq )rK ru r� r3 r. r/ rM � s �c s g | ]}�j �� |j���qS r. )r} �indexr� rJ r r. r/ rM � s c S s g | ] }|j �qS r. �rI rJ r. r. r/ rM � rt ) r rT � introspection�identifier_converterrB rN r� r� r} )r- Zmodel_init_fieldsr� Zmodel_init_orderr� r. r r/ rz � s � � �z$RawQuerySet.resolve_model_init_orderc G s( | � � }|dkrd|_n |j| |_|S )z#Same as QuerySet.prefetch_related()r) r. )r� r� )r- r� r r. r. r/ r� � s z RawQuerySet.prefetch_relatedc C s t | jg| j�R � d| _d S r� r� r3 r. r. r/ r� � s z%RawQuerySet._prefetch_related_objectsc C s: | j | j| j| j| j| j| j| jd�}| jdd� |_|S )zSame as QuerySet._clone())rB rU rY r� r@ r� N) r� r� rB rU rY r� r� r� r� r� r. r. r/ r� s � zRawQuerySet._clonec C s0 | j d u rt| �� �| _ | jr,| js,| �� d S r) )r� r1 r� r� r� r� r3 r. r. r/ r� s zRawQuerySet._fetch_allc C s | � � t| j�S r) r� r3 r. r. r/ r� s zRawQuerySet.__len__c C s | � � t| j�S r) r� r3 r. r. r/ r� s zRawQuerySet.__bool__c C s | � � t| j�S r) r� r3 r. r. r/ r5 s zRawQuerySet.__iter__c s � fdd�}|� S )Nc s( t � j�� I d H � jD ] } | V qd S r) r� r� r3 r. r/ r� & s z(RawQuerySet.__aiter__.<locals>.generatorr. r� r. r3 r/ r: # s zRawQuerySet.__aiter__c c s t | �E d H d S r) )rp r3 r. r. r/ r� - s zRawQuerySet.iteratorc C s d| j j| jf S )Nz<%s: %s>)r� r; rU r3 r. r. r/ r� 0 s zRawQuerySet.__repr__c C s t | �| S r) )r1 )r- r� r. r. r/ r� 3 s zRawQuerySet.__getitem__c C s | j ptj| jfi | j��S r� )r� r r� rB r� r3 r. r. r/ rT 6 s zRawQuerySet.dbc C s& t | j| j| jj|d�| j| j|d�S )z<Select the database this RawQuerySet should execute against.r? )rB rU rY r� r@ )r� r� rB rU r rY r� )r- r� r. r. r/ r@ ; s �zRawQuerySet.usingc C sL | j �� }| j�� D ]2\}}z|�|�}W n ty< Y q0 |||< q|S )zg A list of model field names in the order they'll appear in the query results. )rU Z get_columnsr� r] r r� )r- r} Z query_nameZ model_namer r. r. r/ r} F s zRawQuerySet.columnsc C s> t | j jj}i }| jjjD ] }|�� \}}||||�< q |S )z1A dict mapping column names to model field names.)r rT r r rB rN r� Zget_attname_column)r- r rq rP rd r� r. r. r/ rq X s zRawQuerySet.model_fields)NNr. NNN)r; r< r= ro r0 rz r� r� r� r� r� r� r5 r: r� r� r� r� rT r@ r$ r} rq r. r. r. r/ r� � s6 � r� c @ sN e Zd Zddd�Zdd� Zdd� Zdd � Zd d � Zd d � Zdd� Z dd� Z dS )r� Nc C sp || _ || _|d ur<t|t�s4t|d�r<t|jt�s<td��|r`t � |� t �d d� |g �| _|| _ || _ d S )Nr� zAPrefetch querysets cannot use raw(), values(), and values_list().rD )�prefetch_throughr� r� r� r~ � issubclassr� r> r� r rf r� r* �to_attr)r- r� r* r r. r. r/ r0 d s$ �� ���zPrefetch.__init__c C s6 | j �� }| jd ur2| j�� }g |_d|_||d< |S )NTr* )r� r� r* r� r� r� )r- Zobj_dictr* r. r. r/ r� { s zPrefetch.__getstate__c C s$ |t | j | _|t | j | _d S r) )r r r� )r- �prefixr. r. r/ � add_prefix� s zPrefetch.add_prefixc C s t �| j�t �d |d � �S �NrE )r rf r� r� �r- �levelr. r. r/ �get_current_prefetch_to� s z Prefetch.get_current_prefetch_toc C s2 | j �t�}|| }| jo(|t|�d k}||fS r )r� r� r r r6 )r- r �partsr �as_attrr. r. r/ �get_current_to_attr� s z Prefetch.get_current_to_attrc C s | � |�| jkr| jS d S r) )r r� r* r r. r. r/ �get_current_queryset� s z Prefetch.get_current_querysetc C s t |t�stS | j|jkS r) )r� r� �NotImplementedr� r� r. r. r/ �__eq__� s zPrefetch.__eq__c C s t | j| jf�S r) )�hashr� r� r3 r. r. r/ �__hash__� s zPrefetch.__hash__)NN) r; r< r= r0 r� r r r r r r r. r. r. r/ r� c s r� c C s< g }| D ].}t |t�s t|�}|r,|�|� |�|� q|S )z(Normalize lookups into Prefetch objects.)r� r� r rR )r� r �retr� r. r. r/ �normalize_prefetch_lookups� s r c s� | sdS i }t � }t � }tt|��}|�r~|�� }|j|v rV|jdur$td|j ��q$| }|j�t �}t |�D �] \} } |s� q$|� | �} | |v r�|| }qnd} |D ]:} t | d�s�z i | _ W q� ttfy� d} Y q�Y q�0 q�| s� q$|d }|�| �d }t|| |�\}}}� |�s0td| |jj|jf ��| t|�d k�rZ|du �rZtd |j ��d}|du�rz� fd d �|D �}|�r�t|||| �\}}| |v �r�||v �r�||v �s�||| < tt|�| �}|�|� |�|� |�|� qng }|D ]�} | t| dd �v �rt| j �| ��}n,zt| | �}W n t j �y@ Y �q�Y n0 |du �rP�q�t |t��rh|�|� n |� |� �q�|}qnq$dS ) z| Populate prefetched object caches for a list of model instances based on the lookups/Prefetch instances given. Nzl'%s' lookup was already seen with a different queryset. You may need to adjust the ordering of your lookups.T�_prefetched_objects_cacheFr zQCannot find '%s' on %s object, '%s' is an invalid parameter to prefetch_related()rE zp'%s' does not resolve to an item that supports prefetching - this is an invalid parameter to prefetch_related().c s g | ]}� |�s|�qS r. r. rI �� is_fetchedr. r/ rM rt z,prefetch_related_objects.<locals>.<listcomp>r. ) r^ r �reversed�popr� r* r� r r� r r� r r~ r r r� r �get_prefetcherr� r; r6 �prefetch_one_levelr� r� r` rQ r1 rr r ZObjectDoesNotExistr� rR )Zmodel_instancesZrelated_lookupsZ done_queriesZ auto_lookupsZfollowed_descriptorsZ all_lookupsr� Zobj_listZ through_attrsr � through_attrr� Z good_objectsrk Z first_objr � prefetcherZ descriptor� attr_foundZ obj_to_fetch�additional_lookupsZ new_lookupsZ new_obj_listZnew_objr. r r/ r� � s� �� � ��� �� � ���� r� c s� �fdd�}d}|}t | j� d�}|du r6t| � �}ntd}|r�t|d�rT|}|j}nVt | � �}t|d�rl|}� �kr�tt | j�d�t�r��fdd�} | }n� fdd �} | }||||fS ) a� For the attribute 'through_attr' on the given instance, find an object that has a get_prefetch_queryset(). Return a 4 tuple containing: (the object with get_prefetch_queryset (or None), the descriptor object representing this relationship (or None), a boolean that is False if the attribute was not found at all, a function that takes an instance and returns a boolean that is True if the attribute has already been fetched for that instance) c s t | � �S r) r� �r� �r r. r/ �has_to_attr_attributeR s z-get_prefetcher.<locals>.has_to_attr_attributeNT�get_prefetch_querysetc s � | j v S r) )r� r) r* r. r/ �has_cached_propertyt s z+get_prefetcher.<locals>.has_cached_propertyc s � | j v S r) )r r) )r% r. r/ �in_prefetched_cachez s z+get_prefetcher.<locals>.in_prefetched_cache)rQ r� r~ rb r� r$ ) r� r% r r+ r& r Zrel_obj_descriptorr' rm r- r. r. )r% r r/ r# F s. � r# c C s� |� | |�|��\}}}}}} dd� t|dd�D �} | r>d|_t|�} i } | D ] } || �}| �|g ��| � qN|�|�\}}|r�| r�| d j}z|j � |� W n t j y� Y n0 d}t |�||j���t|j�t��d |k}| D ]�}||�}| �|g �}|�rL|�r|d nd}|�r*t|||� n | �r>t|||� n ||jj|< q�|�r`t|||� q�t||�}|�r�|jdu�r�|�|j�}n|�� }||_d |_ ||j |< q�| | fS ) aL Helper function for prefetch_related_objects(). Run prefetches on all instances using the prefetcher object, assigning results to relevant caches in instance. Return the prefetched objects along with any additional prefetches that must be done due to prefetch_related lookups found from default managers. c S s g | ]}t � |��qS r. )r� )rK Zadditional_lookupr. r. r/ rM � s �z&prefetch_one_level.<locals>.<listcomp>r� r. r z2to_attr={} conflicts with a field on the {} model.rE NT) r, r rQ r� r1 r� rR r r� rN rO r r| r� r� r; r6 r r� r rr ra rC Z fields_cacher* Z_apply_rel_filtersZ get_querysetr� r� r )Z instancesr& r� r Zrel_qsZ rel_obj_attrZ instance_attr�singleZ cache_nameZ is_descriptorr( Zall_related_objectsZ rel_obj_cacherm Z rel_attr_valr r rB �msg�leafrk Zinstance_attr_val�vals�valr� r� r. r. r/ r$ � s` � � r$ c @ s e Zd ZdZdd� Zdd� ZdS )�RelatedPopulatora� RelatedPopulator is used for select_related() object instantiation. The idea is that each select_related() model will be populated by a different RelatedPopulator instance. The RelatedPopulator instances get klass_info and select (computed in SQLCompiler) plus the used db as input for initialization. That data is used to compute which columns to use, how to instantiate the model, and how to populate the links between the objects. The actual creation of the objects is done in populate() method. This method gets row and from_obj as input and populates the select_related() model instance. c s� || _ |d }|d }|sV|d | _|d d | _dd� �| j| j� D �| _d | _nX�fdd �|D �� d d � |d jjD �}� fd d�|D �| _tj� fdd�| jD �� | _|d | _ | j� | j jj j �| _ t|�| j �| _|d | _|d | _d S )NrC � from_parentr rD rE c S s g | ]}|d j j�qS rF rG rJ r. r. r/ rM s z-RelatedPopulator.__init__.<locals>.<listcomp>c s i | ]}� | d j j|�qS rF rG )rK r� r� r. r/ r� s z-RelatedPopulator.__init__.<locals>.<dictcomp>c s s | ] }|j V qd S r) r rJ r. r. r/ r� ! s z,RelatedPopulator.__init__.<locals>.<genexpr>rB c s g | ]}|� v r|�qS r. r. �rK rI ��attname_indexesr. r/ rM $ s c s g | ] }� | �qS r. r. r6 r7 r. r/ rM ( rt � local_setter� remote_setter)rT � cols_start�cols_endrh �reorder_for_initrN r> rQ r� rg r r{ rI �pk_idxr[ ri r9 r: )r- rY rX rT rC r5 Zmodel_init_attnamesr. )r8 rX r/ r0 � s6 � � � �� zRelatedPopulator.__init__c C s� | j r| � |�}n|| j| j� }|| j d u r6d }n,| j�| j| j|�}| jD ]}|� ||� qP| � ||� |d ur�| � ||� d S r) ) r= r; r< r> rg r_ rT rh ri r` r9 r: )r- rj Zfrom_objZobj_datark Zrel_iterr. r. r/ r` 1 s zRelatedPopulator.populateN)r; r< r= ro r0 r` r. r. r. r/ r4 � s 6r4 c C s4 g }| � dg �}|D ]}t|||�}|�|� q|S )N�related_klass_infos)rr r4 rR )rY rX rT � iteratorsr? Zrel_klass_infoZrel_clsr. r. r/ r[ A s r[ )N)Nro r� rQ r� � itertoolsr r Z asgiref.syncr r� Z django.confr Z django.corer Z django.dbr r r r r r Zdjango.db.modelsr r r r r Zdjango.db.models.constantsr r Zdjango.db.models.deletionr Z django.db.models.expressionsr r r r Zdjango.db.models.functionsr r Z django.db.models.query_utilsr r Z django.db.models.sql.constantsr r Zdjango.db.models.utilsr r r! Z django.utilsr" Zdjango.utils.deprecationr# Zdjango.utils.functionalr$ r% r r� r( r> rp r� r� r� r� r� r� r r� r� r� r r� r# r$ r4 r[ r. r. r. r/ �<module> sn $A,' % = ;jV