관리-도구
편집 파일: pipeline.cpython-38.pyc
U /?�f�] � @ s� d Z ddlmZ ddlZddlZddlZddlZddlmZ ddl m Z ddl m Z ddl mZ ddlmZmZ dd lmZmZmZmZmZmZmZmZmZ ddl Z dd l m Z er�dd l m Z! dd l"m#Z# dd l$m%Z& ej'dk r�e(e)�Z*n ddl+m*Z* dddgZ,e&ddi�Z-e f e-�G dd� d��Z.e G dd� d��Z/e f e-�G dd� d��Z0e f e-�G dd � d ��Z1e f e-�G d d � d ��Z2e f e-�G d d � d ��Z3e f e-�G d!d"� d"��Z4e f e-�G d#d$� d$��Z5e f e-�G d%d&� d&��Z6ee j7e j8e j9e j:e j;e j<e j=e j>e j?e3e4e5e6ee@ f ZAe f e-�G d'd(� d(��ZBee.e/e0e1e2eBf ZCed)�ZDed*�ZEed+�ZFG d,d-� d-�ZGe f e&�G d.d/� d/eeDeEf ��ZHeHeef d0�jIZIeHeef d0�jJZJeHeef e.eG�f�jKZKd1d2d3d4d5�d6d7�ZLd8d3d9d:d4d;�d<d=�ZMd3d>d?d9d:d4d@�dAdB�ZNd3dCd9d4dD�dEdF�ZOd3dGd4dH�dIdJ�ZPG dKdL� dLe jQe jRe�ZSG dMdN� dNe�ZTedOe jUdP�ZVedQe jRdP�ZWedRe jXdP�ZYedSe jQdP�ZZedTeTdP�Z[edUe j\dP�Z]edVe j^dP�Z_edWejdP�Z`edXeSdP�ZaedY�ZbedZ�ZcdS )[zZExperimental pipeline API functionality. Be careful with this API, it's subject to change.� )� annotationsN)�deque)� Container)� dataclass)�Decimal)�cached_property�partial) � TYPE_CHECKING�Any�Callable�Generic�Pattern�Protocol�TypeVar�Union�overload)� Annotated�� core_schema)�GetCoreSchemaHandler)� slots_true)� � )� EllipsisType� validate_as�validate_as_deferred� transform�frozenTc @ s" e Zd ZU ded< dZded< dS )� _ValidateAs� type[Any]�tpF�bool�strictN)�__name__� __module__� __qualname__�__annotations__r" � r' r' �L/opt/hc_python/lib/python3.8/site-packages/pydantic/experimental/pipeline.pyr $ s r c @ s( e Zd ZU ded< edd�dd��ZdS )�_ValidateAsDeferzCallable[[], type[Any]]�funcr ��returnc C s | � � S �N)r* ��selfr' r' r( r . s z_ValidateAsDefer.tpN)r# r$ r% r& r r r' r' r' r( r) * s r) c @ s e Zd ZU ded< dS )� _Transform�Callable[[Any], Any]r* N�r# r$ r% r& r' r' r' r( r0 3 s r0 c @ s e Zd ZU ded< ded< dS )� _PipelineOr�_Pipeline[Any, Any]�left�rightNr2 r' r' r' r( r3 8 s r3 c @ s e Zd ZU ded< ded< dS )� _PipelineAndr4 r5 r6 Nr2 r' r' r' r( r7 > s r7 c @ s e Zd ZU ded< dS )�_Eqr �valueNr2 r' r' r' r( r8 D s r8 c @ s e Zd ZU ded< dS )�_NotEqr r9 Nr2 r' r' r' r( r: I s r: c @ s e Zd ZU ded< dS )�_In�Container[Any]�valuesNr2 r' r' r' r( r; N s r; c @ s e Zd ZU ded< dS )�_NotInr< r= Nr2 r' r' r' r( r> S s r> c @ s e Zd ZU ded< dS )� _Constraint�_ConstraintAnnotation� constraintNr2 r' r' r' r( r? l s r? �_InT�_OutT�_NewOutTc @ s e Zd ZdS )�_FieldTypeMarkerN�r# r$ r% r' r' r' r( rE x s rE c @ s� e Zd ZU dZded< ddd�dd�Zed d �d d dd �dd��Zed d �dd dd �dd��Zdd �dd dd �dd�Zddd�dd�Zedddd�d d ��Z ed d d d�d d ��Z ed!d"d!d�d#d ��Z ed$d%d$d�d&d ��Z ed'd(d'd�d)d ��Z edd*dd�d+d ��Z ed,d-d,d�d.d ��Z ed/d0d/d�d1d ��Z ed2d3d2d�d4d ��Z ed/d5d/d�d6d ��Z ed/d7d/d�d8d ��Z ed/d9d/d�d:d ��Z ed/d;d/d�d<d ��Z edd=dd�d>d ��Z d?d@dA�dBd �Z ddCddD�dEdF�Z d dGd dH�dIdJ�Z d$dKd$dL�dMdN�Z ddOddP�dQdR�Z d!dSd!dT�dUdV�Zd�d'dXdYd'dZ�d[d\�Zed]d^d]d_�d`da��Zedbdcdbd_�ddda��Zdd@dd_�deda�Zd/dfd/dg�dhdi�Zd/dfd/dg�djdk�Zd/dld/dm�dndo�Zd/dld/dm�dpdq�Zdrdrds�dtdu�Zdrdrds�dvdw�Zdrdxdrdy�dzd{�Zdrd|drdy�d}d~�Zddds�d�d��Zddds�d�d��Zddds�d�d��Zddds�d�d��Z dd�dd��d�d��Z dd�dd��d�d��Z dd�dd��d�d��Z dd�dd��d�d��Z d�d�d��d�d��Z!e!Z"d�d�d��d�d��Z#e#Z$d@d�d�d��d�d��Z%dfd d��d�d��Z&dWS )�� _PipelinezTAbstract representation of a chain of validation, transformation, and parsing steps.ztuple[_Step, ...]�_stepszCallable[[_OutT], _NewOutT]z_Pipeline[_InT, _NewOutT])r* r, c C s t ttf | jt|�f �S )z�Transform the output of the previous step. If used as the first step in a pipeline, the type of the field is used. That is, the transformation is applied to after the value is parsed to the field's type. )rG rB rD rH r0 �r/ r* r' r' r( r � s z_Pipeline.transform.�r" ztype[_NewOutT]r! )r r" r, c C s d S r- r' �r/ r r" r' r' r( r � s z_Pipeline.validate_asr z_Pipeline[_InT, Any]c C s d S r- r' rK r' r' r( r � s Fz type[_NewOutT] | EllipsisTypec C sJ t |t�r*tttf | jtt|d�f �S tttf | jt||d�f �S )z�Validate / parse the input into a new type. If no type is provided, the type of the field is used. Types are parsed in Pydantic's `lax` mode by default, but you can enable `strict` mode by passing `strict=True`. rJ ) � isinstancer rG rB r rH r rE rD rK r' r' r( r � s z Callable[[], type[_NewOutT]]c C s t ttf | jt|�f �S )z�Parse the input into a new type, deferring resolution of the type until the current class is fully defined. This is useful when you need to reference the class in it's own type annotations. )rG rB rD rH r) rI r' r' r( r � s z _Pipeline.validate_as_deferredz_Pipeline[_InT, _NewOutGe]zannotated_types.Ge)r/ rA r, c C s d S r- r' �r/ rA r' r' r( � constrain� s z_Pipeline.constrainz_Pipeline[_InT, _NewOutGt]zannotated_types.Gtc C s d S r- r' rM r' r' r( rN � s z_Pipeline[_InT, _NewOutLe]zannotated_types.Lec C s d S r- r' rM r' r' r( rN � s z_Pipeline[_InT, _NewOutLt]zannotated_types.Ltc C s d S r- r' rM r' r' r( rN � s z_Pipeline[_InT, _NewOutLen]zannotated_types.Lenc C s d S r- r' rM r' r' r( rN � s zannotated_types.MultipleOfc C s d S r- r' rM r' r' r( rN � s z _Pipeline[_InT, _NewOutDatetime]zannotated_types.Timezonec C s d S r- r' rM r' r' r( rN � s z_Pipeline[_InT, _OutT]zannotated_types.Predicatec C s d S r- r' rM r' r' r( rN � s z _Pipeline[_InT, _NewOutInterval]zannotated_types.Intervalc C s d S r- r' rM r' r' r( rN � s r8 c C s d S r- r' rM r' r' r( rN � s r: c C s d S r- r' rM r' r' r( rN � s r; c C s d S r- r' rM r' r' r( rN � s r> c C s d S r- r' rM r' r' r( rN � s z Pattern[str]c C s d S r- r' rM r' r' r( rN � s r@ r )rA r, c C s t ttf | jt|�f �S )a Constrain a value to meet a certain condition. We support most conditions from `annotated_types`, as well as regular expressions. Most of the time you'll be calling a shortcut method like `gt`, `lt`, `len`, etc so you don't need to call this directly. )rG rB rC rH r? rM r' r' r( rN � s zCallable[[_NewOutT], bool])r/ r* r, c C s | � t�|��S )z.Constrain a value to meet a certain predicate.)rN �annotated_types� PredicaterI r' r' r( � predicate� s z_Pipeline.predicate� _NewOutGt)r/ �gtr, c C s | � t�|��S )z5Constrain a value to be greater than a certain value.)rN rO �Gt)r/ rS r' r' r( rS � s z _Pipeline.gt� _NewOutLt)r/ �ltr, c C s | � t�|��S )z2Constrain a value to be less than a certain value.)rN rO �Lt)r/ rV r' r' r( rV � s z _Pipeline.lt� _NewOutGe)r/ �ger, c C s | � t�|��S )zAConstrain a value to be greater than or equal to a certain value.)rN rO �Ge)r/ rY r' r' r( rY � s z _Pipeline.ge� _NewOutLe)r/ �ler, c C s | � t�|��S )z>Constrain a value to be less than or equal to a certain value.)rN rO �Le)r/ r\ r' r' r( r\ � s z _Pipeline.leN�intz int | None)r/ �min_len�max_lenr, c C s | � t�||��S )z+Constrain a value to have a certain length.)rN rO �Len)r/ r_ r` r' r' r( �len� s z _Pipeline.lenz_Pipeline[_InT, _NewOutDiv]� _NewOutDiv)r/ � multiple_ofr, c C s d S r- r' �r/ rd r' r' r( rd s z_Pipeline.multiple_ofz_Pipeline[_InT, _NewOutMod]� _NewOutModc C s d S r- r' re r' r' r( rd s c C s | � t�|��S )z7Constrain a value to be a multiple of a certain number.)rN rO � MultipleOfre r' r' r( rd s rC )r/ r9 r, c C s | � t|��S )z1Constrain a value to be equal to a certain value.)rN r8 �r/ r9 r' r' r( �eq s z _Pipeline.eqc C s | � t|��S )z5Constrain a value to not be equal to a certain value.)rN r: rh r' r' r( �not_eq s z_Pipeline.not_eqzContainer[_OutT])r/ r= r, c C s | � t|��S )z)Constrain a value to be in a certain set.)rN r; �r/ r= r' r' r( �in_ s z _Pipeline.in_c C s | � t|��S )z-Constrain a value to not be in a certain set.)rN r> rk r' r' r( �not_in s z_Pipeline.not_inz"_Pipeline[_InT, datetime.datetime])r/ r, c C s | � t�d ��S r- �rN rO �Timezoner. r' r' r( �datetime_tz_naive s z_Pipeline.datetime_tz_naivec C s | � t�d��S )N.rn r. r' r' r( �datetime_tz_aware s z_Pipeline.datetime_tz_awarezdatetime.tzinfo)r/ �tzr, c C s | � t�|��S r- rn �r/ rr r' r' r( � datetime_tz" s z_Pipeline.datetime_tzzdatetime.tzinfo | Nonec C s | � ttjj|d��S )N)�tzinfo)r r �datetime�replacers r' r' r( �datetime_with_tz' s z_Pipeline.datetime_with_tzz_Pipeline[_InT, str]c C s | � tj�S r- )r �str�lowerr. r' r' r( � str_lower- s z_Pipeline.str_lowerc C s | � tj�S r- )r ry �upperr. r' r' r( � str_upper0 s z_Pipeline.str_upperc C s | � tj�S r- )r ry �titler. r' r' r( � str_title3 s z_Pipeline.str_titlec C s | � tj�S r- )r ry �stripr. r' r' r( � str_strip6 s z_Pipeline.str_stripry )r/ �patternr, c C s | � t�|��S r- )rN �re�compile)r/ r� r' r' r( � str_pattern9 s z_Pipeline.str_pattern)r/ � substringr, c s | � � fdd��S )Nc s � | kS r- r' ��v�r� r' r( �<lambda>= � z(_Pipeline.str_contains.<locals>.<lambda>�rQ )r/ r� r' r� r( � str_contains< s z_Pipeline.str_contains)r/ �prefixr, c s | � � fdd��S )Nc s | � � �S r- )� startswithr� �r� r' r( r� @ r� z+_Pipeline.str_starts_with.<locals>.<lambda>r� )r/ r� r' r� r( �str_starts_with? s z_Pipeline.str_starts_with)r/ �suffixr, c s | � � fdd��S )Nc s | � � �S r- )�endswithr� �r� r' r( r� C r� z)_Pipeline.str_ends_with.<locals>.<lambda>r� )r/ r� r' r� r( � str_ends_withB s z_Pipeline.str_ends_withz _Pipeline[_OtherIn, _OtherOut]z-_Pipeline[_InT | _OtherIn, _OutT | _OtherOut])�otherr, c C s t t| |�f�S )zxCombine two validation chains, returning the result of the first chain if it succeeds, and the second chain if it fails.)rG r3 �r/ r� r' r' r( � otherwiseF s z_Pipeline.otherwisez_Pipeline[_OutT, _OtherOut]z_Pipeline[_InT, _OtherOut]c C s t t| |�f�S )z5Pipe the result of one validation chain into another.)rG r7 r� r' r' r( �thenL s z_Pipeline.thenr � cs.CoreSchema)� source_type�handlerr, c C sF ddl m} t| j�}d }|r6|�� }t||||�}q|p@|�� }|S )Nr r )� pydantic_corer r rH �popleft� _apply_stepZ any_schema)r/ r� r� �cs�queue�s�stepr' r' r( � __get_pydantic_core_schema__R s z&_Pipeline.__get_pydantic_core_schema__)�_r, c C s t �d S r- )�NotImplementedError)r/ r� r' r' r( �__supports_type__` s z_Pipeline.__supports_type__)N)'r# r$ r% �__doc__r& r r r r rN rQ rS rV rY r\ rb rd ri rj rl rm rp rq rt rx r{ r} r r� r� r� r� r� r� �__or__r� �__and__r� r� r' r' r' r( rG � s� rG r' zCallable[[Any], bool]zstr | Callable[[], str]zcs.CoreSchema | Noner� )r* � predicate_errr� r, c sD ddl m} ddd�� �fdd� }|d kr4|�|�S |�||�S d S )Nr r r �r� r, c s. � | �r | S t dt�t�r �n�� � ���d S )Nz Expected )� ValueErrorrL ry r� �r* r� r' r( r� n s z _check_func.<locals>.handler)r� r � no_info_plain_validator_function� no_info_after_validator_function)r* r� r� r� r� r' r� r( � _check_funci s r� �_Stepr r )r� r� r� r� r, c C s� ddl m} t| t�r,t|| j| j||�}n�t| t�rJt|| jd||�}n�t| t�rdt || j |�}nft| t �r|t || j �}nNt| t�r�|�|| j�|| j�g�}n(t| t�s�t�|�|| j�|| j�g�}|S )Nr r F)r� r rL r � _apply_parser r" r) r0 �_apply_transformr* r? �_apply_constraintrA r3 Z union_schemar5 r6 r7 �AssertionError� chain_schema)r� r� r� r� r� r' r' r( r� y s r� r r! )r� r r" r� r� r, c C st ddl m} ddlm} |tkr(||�S |r:t||� f }| rR| d dkrR||�S | rh|�| ||�g�S ||�S d S )Nr r )�Strict�type�any)r� r �pydanticr� rE r r� )r� r r" r� r� r� r� r' r' r( r� � s r� r1 )r� r* r� r, c C s� ddl m} | d kr |�|�S | d dkr�|tjkrH| �� } d| d< | S |tjkrf| �� } d| d< | S |tjkr�| �� } d| d< | S |�|| �S ) Nr r r� ry TZstrip_whitespaceZto_lowerZto_upper) r� r r� ry r� �copyrz r| r� )r� r* r� r� r' r' r( r� � s" r� r@ )r� rA r, c s t � tj�r�� j�| r�| d dkr�| �� } | d dkrJt �t�rJ�| d< q�| d dkrjt �t�rj�| d< q�| d dkr�t �t�r��| d< n&ddd ��fd d � }t|d �� �| �} �nJt � tj ��rx� j �| �rN| d dk�rN| �� } | d dk�rt �t��r�| d < nF| d dk�r,t �t��r,�| d < n"| d dk�rNt �t��rN�| d < ddd ��fdd� }t|d�� �| �} �n�t � tj ��r<� j �| �r| d dk�r| �� } | d dk�r�t �t��r̈| d< nF| d dk�r�t �t��r�| d< n"| d dk�rt �t��r�| d< ddd ��fdd� }t|d�� �| �} �n�t � tj ��r � j�| �r�| d dk�r�| �� } | d dk�r�t �t��r��| d< nF| d dk�r�t �t��r��| d< n"| d dk�r�t �t��rֈ| d< ddd ��fdd� }t|d�� �| �} �n�t � tj��r� j�� j�| �r�| d dk�r�| d dk�s�| d dk�s�| d d k�s�| d d k�s�| d d k�s�| d d k�s�t�| �� } �d k�r��| d!< �d"k �r��| d#< ddd ���fd$d%� }t|d&�� d'�� �| �} �nt � tj��r�� j�| �r~| d dk�r~| �� } | d dk�r8t �t��r8�| d(< nF| d dk�r\t �t��r\�| d(< n"| d dk�r~t �t��r~�| d(< ddd ��fd)d*� }t|d+�� d,�| �} �nTt � tj��rf� j}|d-k�r | �r�| d d.k�r�| �� } d/| d0< n d1dd �d2d3�} t| d4| �} nV|d"k�rZ| �r<| d d.k�r<| �� } d5| d0< n d1dd �d6d7�} t| d8| �} ntd9���n�t � tj��r� j �r�t| t� � j ��} � j�r�t| t�� j��} � j�r�t| t� � j��} � j �r�t| t� � j ��} | d"k �s�t��nt � tj��r�� j} | j d:k�r�d d"l } zb| � | �� � } | � d;��r>| d"d<� } d=d>�!d>�!| �"d?�d@d"� ��"dA�d@d"� �� � d= }W n t#k �r� dB}Y nX t| || �} nt| | j | �} �nJt � t$��r� j%�ddd ��fdCdD� }t|dE�� �| �} �nt � t&��r*� j%�ddd ��fdFdG� }t|dH�� �| �} n�t � t'��rd� j(� ddd �� fdIdJ� }t|dK� � �| �} n�t � t)��r�� j(� ddd �� fdLdM� }t|dN� � �| �} n`t � t*��s�t�| �r�| d dk�r�| �� } � j+| dO< n(d1dd �� fdPdQ� }t|dR� j+� �| �} | S )Sz&Apply a single constraint to a schema.r� > r^ �float�decimalr^ rS r� r� r r! r� c s | � kS r- r' r� )rS r' r( �check_gt� s z#_apply_constraint.<locals>.check_gtz> rY c s | � kS r- r' r� )rY r' r( �check_ge� s z#_apply_constraint.<locals>.check_gez>= rV c s | � k S r- r' r� )rV r' r( �check_lt� s z#_apply_constraint.<locals>.check_ltz< r\ c s | � kS r- r' r� )r\ r' r( �check_le� s z#_apply_constraint.<locals>.check_lez<= > �tuple�set� frozensetry �dict�listry r� r� r� r� r� r � min_lengthN� max_lengthc s, � d k r �t | �ko t | �� kS �t | �kS r- )rb r� )r` r_ r' r( � check_len s z$_apply_constraint.<locals>.check_lenz length >= z and length <= rd c s | � dkS )Nr r' r� )rd r' r( �check_multiple_of! s z,_apply_constraint.<locals>.check_multiple_ofz% z == 0.rv ZawareZ tz_constraint�objectc S s t | tj�st�| jd k S r- �rL rv r� ru r� r' r' r( �check_tz_aware. s z)_apply_constraint.<locals>.check_tz_awareztimezone awareZnaivec S s t | tj�st�| jd kS r- r� r� r' r' r( �check_tz_naive9 s z)_apply_constraint.<locals>.check_tz_naiveztimezone naivez8Constraining to a specific timezone is not yet supportedz<lambda>�)����`� zlambda � �:�lambdac s t �| � �S r- )�operator�__ne__r� �r9 r' r( � check_not_eqc s z'_apply_constraint.<locals>.check_not_eqz!= c s t �| � �S r- )r� �__eq__r� r� r' r( �check_eqj s z#_apply_constraint.<locals>.check_eqz== c s t �� | �S r- )r� � __contains__r� �r= r' r( �check_inq s z#_apply_constraint.<locals>.check_inzin c s t �t �� | ��S r- )r� �__not__r� r� r� r' r( � check_not_inx s z'_apply_constraint.<locals>.check_not_inznot in r� c s t | t�st�� �| �d k S r- )rL ry r� �matchr� )rA r' r( � check_pattern� s z(_apply_constraint.<locals>.check_patternz~ ),rL rO rT rS r� r^ r� r r� rZ rY rW rV r] r\ ra r� r� r� rg rd ro rr r� �Intervalr� rP r* r# �inspect� getsourcer� r� �join�split�OSErrorr: r9 r8 r; r= r> r r� )r� rA r� r� r� r� r� r� rr r� r� r* r� �sourceZlambda_source_coder� r� r� r� r� r' ) rA rY rS r\ rV r` r_ rd r9 r= r( r� � s � � � � � � < r� c @ s e Zd ZdS )�_SupportsRangeNrF r' r' r' r( r� � s r� c @ s e Zd Zdd�dd�ZdS )� _SupportsLenr^ r+ c C s d S r- r' r. r' r' r( �__len__� r� z_SupportsLen.__len__N)r# r$ r% r� r' r' r' r( r� � s r� rR )�boundrX rU r[ � _NewOutLenrc rf �_NewOutDatetime�_NewOutInterval�_OtherIn� _OtherOut)dr� � __future__r rv r� r� �sys� collectionsr �collections.abcr � dataclassesr r� r � functoolsr r �typingr r r r r r r r r rO �typing_extensionsr r� r r� r� r Z&pydantic._internal._internal_dataclassr Z _slots_true� version_infor� �Ellipsisr �types�__all__Z _slots_frozenr r) r0 r3 r7 r8 r: r; r> r] rZ rW rT ra rg ro r� rP ry r@ r? r� rB rC rD rE rG r r r r� r� r� r� r� Z SupportsLeZ SupportsGer� r� Z SupportsGtrR rX Z SupportsLtrU r[ r� Z SupportsDivrc Z SupportsModrf r� r� r� r� r' r' r' r( �<module> s� , �� d P