관리-도구
편집 파일: writeonly.cpython-38.pyc
U -?�f!W � @ s� d Z ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z dd lm Z dd lm Z dd lm Z dd lm Z dd lmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z d d lm Z d d lm Z d d lm!Z! d d lm"Z" d d!l#m$Z$ d d"l#m%Z% d d#l#m&Z& d d$l#m'Z' d d%l(m)Z) d d&l(m*Z* d d'l(m+Z+ d d(l,m-Z- e�rbdd)lm.Z. dd*l/m0Z0 dd+lm1Z1 dd,lm2Z2 dd-l3m4Z4 dd.l3m5Z5 dd/l6m7Z7 dd0lm8Z8 dd1l9m:Z: dd2l"m;Z; d d3l<m=Z= d d4l>m?Z? d d5l>m@Z@ ed6ed7�ZAG d8d9� d9eeA �ZBG d:d;� d;ejCejD�ZEe!jFejGjHd<d=�G d>d?� d?ejIe!jJ���ZKG d@dA� dA�ZLG dBdC� dCeeA �ZMG dDdE� dEeMeA �ZNdFS )Gz�Write-only collection API. This is an alternate mapped attribute style that only supports single-item collection mutation operations. To read the collection, a select() object must be executed each time. .. versionadded:: 2.0 � )� annotations)�Any)� Collection)�Dict)�Generic)�Iterable)�Iterator)�List)�NoReturn)�Optional)�overload)�Tuple)�Type)� TYPE_CHECKING)�TypeVar)�Union)� bindparam� )� attributes)� interfaces)� relationships)� strategies)� NEVER_SET)� object_mapper)� PassiveFlag)�RelationshipDirection� )�exc)�inspect)�log)�util)�delete)�insert)�select)�update)�Delete)�Insert)�Update)�Literal)�QueryableAttribute)� _InstanceDict)�AttributeEventToken)�LoaderCallableStatus)�_AdaptedCollectionProtocol)�CollectionAdapter)�Mapper)�_RelationshipOrderByArg)� InstanceState)� AliasedClass)� _Dispatch)� FromClause)�Select�_T)�boundc @ s� e Zd ZU dZded< ded< ded< ded< d%d d d d d d�dd�Zedd�dd��Zedd�dd��Zdd�dd�Z ddd �d d �Z d d d �d!d"�Z d d d �d#d$�Z dS )&�WriteOnlyHistoryzDOverrides AttributeHistory to receive append/remove events directly.zutil.OrderedIdentitySet�unchanged_items� added_items� deleted_items�bool�_reconcile_collectionN�WriteOnlyAttributeImpl�InstanceState[_T]r z Optional[WriteOnlyHistory[_T]]�None)�attr�state�passive�apply_to�returnc C sp |rH|t j@ r&t�d|� d|� d���|j| _|j| _|j| _|j| _n$t� � | _t� � | _t� � | _d| _d S )Nz Attribute z� can't load the existing state from the database for this operation; full iteration is not permitted. If this is a delete operation, configure passive_deletes=True on the z- relationship in order to resolve this error.F) r �SQL_OKr �InvalidRequestErrorr9 r: r; r= r ZOrderedIdentitySet)�selfrA rB rC rD � rI �H/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/orm/writeonly.py�__init__U s � zWriteOnlyHistory.__init__zList[_T]�rE c C s t | j�| j��S �N)�listr: �unionr9 �rH rI rI rJ �added_plus_unchangedp s z%WriteOnlyHistory.added_plus_unchangedc C s t | j�| j��| j��S rM )rN r: rO r9 r; rP rI rI rJ � all_itemst s ��zWriteOnlyHistory.all_items�attributes.Historyc C s` | j r0| j�| j�}| j�| j�}| j�|�}n| j| j| j }}}t�t|�t|�t|��S rM ) r= r: � differencer9 r; � intersectionr ZHistoryrN )rH �addedZdeletedZ unchangedrI rI rJ � as_history| s � zWriteOnlyHistory.as_historyzUnion[int, slice]zUnion[List[_T], _T])�indexrE c C s t | j�| S rM )rN r: )rH rX rI rI rJ �indexed� s zWriteOnlyHistory.indexedr6 )�valuerE c C s | j �|� d S rM )r: �add�rH rZ rI rI rJ � add_added� s zWriteOnlyHistory.add_addedc C s( || j kr| j �|� n | j�|� d S rM )r: �remover; r[ r\ rI rI rJ � add_removed� s z WriteOnlyHistory.add_removed)N) �__name__� __module__� __qualname__�__doc__�__annotations__rK �propertyrQ rR rW rY r] r_ rI rI rI rJ r8 M s � r8 c s� e Zd ZU dZded< dZded< dZded< dZded< dZded< dZ ded < d Z d ed < e Z d ed< ded< ddddd dd�� fdd� Z ejfdddd d �d d �ZedZddd!d"d#d$�d%d&��Zed[ddd'dd#d$�d(d&��Zed\ddd)dd*d$�d+d&��Zd,ejfddd)dd*d$�d-d&�Zejd.d/�d0d1��Zejd.d/�d2d3��Zd]dddd4d5d6d7�d8d9�Zd^dddd4d5d6d7�d:d;�Zddd<d=�d>d?�Zd,ejd,ddfdddd4ddddd6d@� dAdB�ZdddCdD�dEdF�ZddddCdG�dHdI�Z ej fddddJd �dKdL�Z ej fddddMd �dNdO�Z ddd<dP�dQdR�Z!ej fdddd4dd6dS�dTdU�Z"ej fdddd4dd6dS�dVdW�Z#ej fdddd4dd6dS�dXdY�Z$� Z%S )_r> Tr< � uses_objectsF� default_accepts_scalar_loader�supports_population�_supports_dynamic_iteration� collection�dynamicrI r0 �order_byzType[WriteOnlyHistory[Any]]�collection_history_clsz Type[WriteOnlyCollection[Any]]� query_classz#Union[Type[Any], AliasedClass[Any]]�strz"_Dispatch[QueryableAttribute[Any]]z Mapper[_T]r )�class_�key�dispatch� target_mapperrl �kwc s4 t � j||d |f|� || _t| _|r0t|�| _d S rM )�superrK rs �WriteOnlyCollectionrn �tuplerl )rH rp rq rr rs rl rt �� __class__rI rJ rK � s z WriteOnlyAttributeImpl.__init__zInstanceState[Any]r* r z8Union[util.OrderedIdentitySet, WriteOnlyCollection[Any]])rB �dict_rC rE c C s* |t j@ s| �|t j�jS | �| |�S d S rM )r rF �_get_collection_history�PASSIVE_NO_INITIALIZEr: rn )rH rB rz rC rI rI rJ �get� s �zWriteOnlyAttributeImpl.get.z Literal[None]z Literal[PassiveFlag.PASSIVE_OFF]r. )rB rz � user_datarC rE c C s d S rM rI �rH rB rz r~ rC rI rI rJ �get_collection� s z%WriteOnlyAttributeImpl.get_collectionr- c C s d S rM rI r rI rI rJ r� � s z$Optional[_AdaptedCollectionProtocol]zIUnion[Literal[LoaderCallableStatus.PASSIVE_NO_RESULT], CollectionAdapter]c C s d S rM rI r rI rI rJ r� � s Nc C s4 |t j@ s| �||�j}n| �||�}|j}t|�S rM )r rF r{ r: rQ �DynamicCollectionAdapter)rH rB rz r~ rC �data�historyrI rI rJ r� � s z attributes.AttributeEventTokenrL c C s t �| t j�S rM )r r+ Z OP_APPENDrP rI rI rJ � _append_token� s z$WriteOnlyAttributeImpl._append_tokenc C s t �| t j�S rM )r r+ Z OP_REMOVErP rI rI rJ � _remove_token� s z$WriteOnlyAttributeImpl._remove_tokenz Optional[AttributeEventToken]z Optional[WriteOnlyHistory[Any]]r@ )rB rz rZ � initiator�collection_historyrE c C sd |d kr| � ||�}|�|� | jjD ]}||||p8| j�}q&| jr`|d k r`| �t�|�|d� d S �NT) �_modified_eventr] rr �appendr� � trackparent� sethasparentr �instance_state�rH rB rz rZ r� r� �fnrI rI rJ �fire_append_event� s z(WriteOnlyAttributeImpl.fire_append_eventc C sd |d kr| � ||�}|�|� | jr@|d k r@| �t�|�|d� | jjD ]}||||pZ| j� qHd S )NF) r� r_ r� r� r r� rr r^ r� r� rI rI rJ �fire_remove_event s z(WriteOnlyAttributeImpl.fire_remove_eventzWriteOnlyHistory[Any])rB rz rE c C sH | j |jkr$| �| |tj�|j| j < |�|| t� d|| j < |j| j S r� )rq �committed_staterm r �PASSIVE_NO_FETCHr� r )rH rB rz rI rI rJ r� s � z&WriteOnlyAttributeImpl._modified_event) rB rz rZ r� rC � check_old�pop�_adaptrE c C s� |r|j | j krd S |r$|d kr$d S |} t| �} |jrd| jsNt�d| � d���t�| j|||d��} | � ||�} |js~| j } n | � | j �} | � | �} t�| �� | �}| � | �}| D ] }||kr�| j|||d | d� q�|D ]}| j|||d | d� q�d S )Nz Collection "zV" does not support implicit iteration; collection replacement operations can't be used�rC )r� )Z parent_tokenrN � has_identityri r rG r Z IdentitySetr} r� r: rO rU rT r� r� )rH rB rz rZ r� rC r� r� r� �iterable� new_valuesZold_collectionr� � constants� additionsZremovals�memberrI rI rJ �set/ sP �� � ��zWriteOnlyAttributeImpl.setr )�args�kwargsrE c O s t � �d S rM ��NotImplementedError)rH r� r� rI rI rJ r! l s z WriteOnlyAttributeImpl.delete)rB rz rZ rE c C s t d��d S )Nz7Dynamic attributes don't support collection population.r� )rH rB rz rZ rI rI rJ �set_committed_valueo s �z*WriteOnlyAttributeImpl.set_committed_valuerS c C s | � ||�}|�� S rM )r{ rW �rH rB rz rC �crI rI rJ � get_historyv s z"WriteOnlyAttributeImpl.get_historyz$List[Tuple[InstanceState[Any], Any]]c C s | � ||�}dd� |jD �S )Nc S s g | ]}t �|�|f�qS rI )r r� )�.0�xrI rI rJ � <listcomp>� s z:WriteOnlyAttributeImpl.get_all_pending.<locals>.<listcomp>)r{ rR r� rI rI rJ �get_all_pending s z&WriteOnlyAttributeImpl.get_all_pending)rB rC rE c C sT | j |jkr|j| j }n| �| |tj�}|jrL|tj@ rL| j| |||d�S |S d S )N)rD )rq r� rm r r� r� ZINIT_OK)rH rB rC r� rI rI rJ r{ � s � �z.WriteOnlyAttributeImpl._get_collection_history)rB rz rZ r� rC rE c C s || k r| � ||||� d S rM )r� �rH rB rz rZ r� rC rI rI rJ r� � s z WriteOnlyAttributeImpl.appendc C s || k r| � ||||� d S rM )r� r� rI rI rJ r^ � s z WriteOnlyAttributeImpl.removec C s | j |||||d� d S )Nr� )r^ r� rI rI rJ r� � s zWriteOnlyAttributeImpl.pop)..)..)..)N)N)&r` ra rb rf rd rg rh ri rj rk rl r8 rm rK r Z PASSIVE_OFFr} r r� r Zmemoized_propertyr� r� r� r� r� r� r! r� r� r� r| r� r{ r� r^ r� � __classcell__rI rI rx rJ r> � sn � � � �� � �� = � � ���r> Z write_only)Zlazyc @ s e Zd ZeZddd�dd�ZdS )�WriteOnlyLoaderz Mapper[Any]r@ )�mapperrE c C s\ d| _ | jr | jjtjtjfkr0t�d| j ��t j | j|d| j | jj | jj | jjd� d S )NTztOn relationship %s, 'dynamic' loaders cannot be used with many-to-one/one-to-one relationships and/or uselist=False.)Z useobject� impl_classrs rl rn )Zis_class_levelZuselistZparent_property� directionr � ONETOMANYZ MANYTOMANYr rG r Z_register_attributer� r� rl rn )rH r� rI rI rJ �init_class_attribute� s& ����z$WriteOnlyLoader.init_class_attributeN)r` ra rb r> r� r� rI rI rI rJ r� � s r� c @ s` e Zd ZU dZded< dd�dd�Zdd�d d �Zd d�d d �Zdd�dd�Zdd�dd�Z dS )r� z9simplified CollectionAdapter for internal API consistencyzCollection[Any]r� �r� c C s || _ d S rM r� )rH r� rI rI rJ rK � s z!DynamicCollectionAdapter.__init__z Iterator[Any]rL c C s t | j�S rM )�iterr� rP rI rI rJ �__iter__� s z!DynamicCollectionAdapter.__iter__r@ c C s d S rM rI rP rI rI rJ � _reset_empty� s z%DynamicCollectionAdapter._reset_empty�intc C s t | j�S rM )�lenr� rP rI rI rJ �__len__� s z DynamicCollectionAdapter.__len__r< c C s dS r� rI rP rI rI rJ �__bool__� s z!DynamicCollectionAdapter.__bool__N) r` ra rb rc rd rK r� r� r� r� rI rI rI rJ r� � s r� c @ sZ e Zd ZU dZesdZded< ded< ddd �d d �Zd d d�dd�Zdd d�dd�Z dS )�AbstractCollectionWriterzsVirtual collection which includes append/remove methods that synchronize into the attribute event system. rI r6 �instancezTuple[FromClause, ...]� _from_objr> r? )rA rB c C s� |� � }tr|st�|| _|| _t|�}|j| jj }|jd k rT|j � � |jf| _ nd| _ |j |dd�f| _ | jjr�| jj| _nd| _d S )NrI F)Zalias_secondary)�objr �AssertionErrorr� rA r �_propsrq Z secondaryr� Z__clause_element__r� Z _with_parent�_where_criteriarl �_order_by_clauses)rH rA rB r� r� �proprI rI rJ rK � s � z!AbstractCollectionWriter.__init__� Iterable[_T]r@ ��iteratorrE c C s0 |D ]&}| j �t�| j�t�| j�|d � qd S rM )rA r� r r� r� � instance_dict)rH r� �itemrI rI rJ � _add_all_impl s �z&AbstractCollectionWriter._add_all_impl�r� rE c C s& | j �t�| j�t�| j�|d � d S rM )rA r^ r r� r� r� �rH r� rI rI rJ � _remove_impl" s �z%AbstractCollectionWriter._remove_implN) r` ra rb rc r � __slots__rd rK r� r� rI rI rI rJ r� � s ! r� c @ s� e Zd ZdZdZdd�dd�Zdd�dd �Zd d�d d �Zd d�dd�Zdd�dd�Z ddd�dd�Z ddd�dd�Z ddd�d d �Z d S ) rv a� Write-only collection which can synchronize changes into the attribute event system. The :class:`.WriteOnlyCollection` is used in a mapping by using the ``"write_only"`` lazy loading strategy with :func:`_orm.relationship`. For background on this configuration, see :ref:`write_only_relationship`. .. versionadded:: 2.0 .. seealso:: :ref:`write_only_relationship` )r� rA r� r� r� r rL c C s t d��d S )Nz�WriteOnly collections don't support iteration in-place; to query for collection items, use the select() method to produce a SQL statement and execute it with session.scalars().)� TypeErrorrP rI rI rJ r� D s �z WriteOnlyCollection.__iter__zSelect[Tuple[_T]]c C s< t | jj�j| j� }| jr&|j| j� }| jr8|j| j� }|S )z�Produce a :class:`_sql.Select` construct that represents the rows within this instance-local :class:`_orm.WriteOnlyCollection`. ) r# rA rs �wherer� r� � select_fromr� rl )rH �stmtrI rI rJ r# K s zWriteOnlyCollection.selectr& c C s� t | j�}|j}|j| jj }|jtjk r4t � d��i }|j D ],\}}|� |||j |�}td|d�||j< q>t| jj�jf |�S )aN For one-to-many collections, produce a :class:`_dml.Insert` which will insert new rows in terms of this this instance-local :class:`_orm.WriteOnlyCollection`. This construct is only supported for a :class:`_orm.Relationship` that does **not** include the :paramref:`_orm.relationship.secondary` parameter. For relationships that refer to a many-to-many table, use ordinary bulk insert techniques to produce new objects, then use :meth:`_orm.AbstractCollectionWriter.add_all` to associate them with the collection. z�Write only bulk INSERT only supported for one-to-many collections; for many-to-many, use a separate bulk INSERT along with add_all().N)Z callable_)r r� r� r� rA rq r� r r� r rG Zsynchronize_pairsZ_get_attr_w_warn_on_none�dictr r" rs �values)rH rB r� r� rz �l�rr� rI rI rJ r" W s" ��zWriteOnlyCollection.insertr' c C s t | jj�j| j� S )z�Produce a :class:`_dml.Update` which will refer to rows in terms of this instance-local :class:`_orm.WriteOnlyCollection`. )r$ rA rs r� r� rP rI rI rJ r$ s zWriteOnlyCollection.updater% c C s t | jj�j| j� S )z�Produce a :class:`_dml.Delete` which will refer to rows in terms of this instance-local :class:`_orm.WriteOnlyCollection`. )r! rA rs r� r� rP rI rI rJ r! � s zWriteOnlyCollection.deleter� r@ r� c C s | � |� dS )z�Add an iterable of items to this :class:`_orm.WriteOnlyCollection`. The given items will be persisted to the database in terms of the parent instance's collection on the next flush. N�r� )rH r� rI rI rJ �add_all� s zWriteOnlyCollection.add_allr6 r� c C s | � |g� dS )z�Add an item to this :class:`_orm.WriteOnlyCollection`. The given item will be persisted to the database in terms of the parent instance's collection on the next flush. Nr� r� rI rI rJ r[ � s zWriteOnlyCollection.addc C s | � |� dS )z�Remove an item from this :class:`_orm.WriteOnlyCollection`. The given item will be removed from the parent instance's collection on the next flush. N)r� r� rI rI rJ r^ � s zWriteOnlyCollection.removeN) r` ra rb rc r� r� r# r" r$ r! r� r[ r^ rI rI rI rJ rv + s ( rv N)Orc � __future__r �typingr r r r r r r r r r r r r r r Zsqlalchemy.sqlr � r r r r �baser r r r r r r r �sqlr! r" r# r$ Zsql.dmlr% r& r' Z util.typingr( r) Z_typingr* r+ r, � collectionsr- r. r� r/ r0 rB r1 r2 �eventr3 Zsql.selectabler4 r5 r6 r8 ZHasCollectionAdapterZ AttributeImplr> Z class_loggerZRelationshipPropertyZ strategy_forZAbstractRelationshipLoaderZ Identifiedr� r� r� rv rI rI rI rJ �<module> s� I � ' ?