관리-도구
편집 파일: ranges.cpython-38.pyc
U -?�f�� � @ s� d dl mZ d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d d lm Z d d lm Z d d lm Z d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z d d l m Z d d l!mZ d d l"m#Z# d d l$m%Z% d d!l&m'Z' e�r�d d"l(m)Z) d d#l"m*Z* d d$l"m+Z+ ed%e d&�Z,e'd' Z-e%�r�d(d)iZ.d*d)iZ/ni Z.i Z/ej0f d+d)ie.��G d,d-� d-e e, ��Z1G d.d/� d/e e1e, �Z2G d0d1� d1e j#e, �Z3G d2d3� d3e3e1e, �Z4G d4d5� d5e4e, �Z5G d6d7� d7e3ee1e, �Z6G d8d9� d9e6e, �Z7G d:d;� d;e4e8 �Z9G d<d=� d=e4e8 �Z:G d>d?� d?e4e �Z;G d@dA� dAe4e �Z<G dBdC� dCe4e �Z=G dDdE� dEe4e �Z>G dFdG� dGe6e8 �Z?G dHdI� dIe6e8 �Z@G dJdK� dKe6e �ZAG dLdM� dMe6e �ZBG dNdO� dOe6e �ZCG dPdQ� dQe6e �ZDdRZEdSZFdTdUdV�dWdX�ZGdS )Y� )� annotationsN)�date)�datetime)� timedelta)�Decimal)�Any)�cast)�Generic)�List)�Optional)�overload)�Sequence)�Tuple)�Type)� TYPE_CHECKING)�TypeVar)�Union� )� ADJACENT_TO)� CONTAINED_BY)�CONTAINS)�NOT_EXTEND_LEFT_OF)�NOT_EXTEND_RIGHT_OF)�OVERLAP)�STRICTLY_LEFT_OF)�STRICTLY_RIGHT_OF� )�types)� operators)� TypeEngine)�py310)�Literal)� ColumnElement)�_TE)�TypeEngineMixin�_T)�bound)z()�[)z(]z[]�slotsTZkw_only�frozenc @ s� e Zd ZU dZdZded< dZded< erTej dd�Z ded < ej d d�Z d ed < n8ej f d die ��Z ded < ej f d d ie ��Z d ed < e s�dXdd d�dddd d�dd�Zd d�dd�Zed d�dd��Zed d�dd��Zed d�dd��Zed d�dd ��Zed d�d d ��Zed d�d d ��Zed!d�d"d#��Zd$d d%�d&d'�Zd(d�d)d*�ZdYdd+dd+d d,d-�d.d/�Zd(d d0�d1d2�Zd3d d0�d4d5�Z d6d d%�d7d8�Z d3d d0�d9d:�Z d3d d0�d;d<�Z e Z d3d d0�d=d>�Z!e!Z"d3d d0�d?d@�Z#d3d d0�dAdB�Z$dd+dd+d dC�dDdE�Z%d3d d0�dFdG�Z&d3d3d0�dHdI�Z'd3d3d0�dJdK�Z(d3d3d0�dLdM�Z)d3d3d0�dNdO�Z*d3d3d0�dPdQ�Z+d3d3d0�dRdS�Z,d+d�dTdU�Z-d+d�dVdW�Z.dS )Z�Rangea_ Represent a PostgreSQL range. E.g.:: r = Range(10, 50, bounds="()") The calling style is similar to that of psycopg and psycopg2, in part to allow easier migration from previous SQLAlchemy versions that used these objects directly. :param lower: Lower bound value, or None :param upper: Upper bound value, or None :param bounds: keyword-only, optional string value that is one of ``"()"``, ``"[)"``, ``"(]"``, ``"[]"``. Defaults to ``"[)"``. :param empty: keyword-only, optional bool indicating this is an "empty" range .. versionadded:: 2.0 Nz Optional[_T]�lower�upperr' )�default� _BoundsType�boundsF�bool�emptyr- )r/ r1 �r+ r, r/ r1 c C s | j �||||d�� d S )Nr2 )�__dict__�update)�selfr+ r, r/ r1 � r6 �U/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/dialects/postgresql/ranges.py�__init__a s ��zRange.__init__��returnc C s | j S �N�r1 �r5 r6 r6 r7 �__bool__s s zRange.__bool__c C s | j S �z$A synonym for the 'empty' attribute.r< r= r6 r6 r7 �isemptyv s z Range.isemptyc C s | j S r? r< r= r6 r6 r7 �is_empty| s zRange.is_emptyc C s | j d dkS )z,Return True if the lower bound is inclusive.r �[�r/ r= r6 r6 r7 � lower_inc� s zRange.lower_incc C s | j o| jdkS )zKReturn True if this range is non-empty and lower bound is infinite.N)r1 r+ r= r6 r6 r7 � lower_inf� s zRange.lower_infc C s | j d dkS )z,Return True if the upper bound is inclusive.r �]rC r= r6 r6 r7 � upper_inc� s zRange.upper_incc C s | j o| jdkS )zOReturn True if this range is non-empty and the upper bound is infinite.N)r1 r, r= r6 r6 r7 � upper_inf� s zRange.upper_infzAbstractSingleRange[_T]c C s t � S r; )�AbstractSingleRanger= r6 r6 r7 �__sa_type_engine__� s zRange.__sa_type_engine__r% ��valuer: c C s� | j r dS | jdkr@| jdkp>| jd dkr6|| jk S || jkS | jdkrl| jd dkrb|| jkS || jkS | jd dkr�|| jkn|| jko�| jd dkr�|| jk S || jkS )z3Return True if this range contains the given value.FNr �)r �()r1 r+ r, r/ �r5 rL r6 r6 r7 �_contains_value� s* � � � � � � � �zRange._contains_valuer c C sb t | jt�st | jt�r dS t | jt�s4t | jt�r8dS t | jt�sPt | jt�rZtdd�S dS dS )uA Determine the “step” for this range, if it is a discrete one.r N)�days)� isinstancer+ �intr, r r r r= r6 r6 r7 �_get_discrete_step� s � zRange._get_discrete_step�strrS )�value1�bound1�value2�bound2� only_valuesr: c C sx |dk}|dk}|| kr$dkrBn n||kr4dS |r<dS dS n(|dkrV|rRdS dS |dkrj|rfdS dS ||kr~||kr~dS |dk}|dk} | � � } | dk r�|r�|s�|| 7 }d}n|r�|| 7 }d}|r�| s�|| 7 }d} n| r�|| 7 }d} ||k r�dS ||k�rdS |�rdS |�r | �r dS |�sH| �sH||k�r8dS |�rBdS dS n,|�s\|�rXdS dS | �sp|�rldS dS dS dS ) a Compare two range bounds. Return -1, 0 or 1 respectively when `value1` is less than, equal to or greater than `value2`. When `only_value` is ``True``, do not consider the *inclusivity* of the edges, just their values. > rN rB Nr ���r > rB rF TF)rT ) r5 rV rW rX rY rZ Zvalue1_is_lower_boundZvalue2_is_lower_boundZ value1_incZ value2_inc�stepr6 r6 r7 �_compare_edges� s^ zRange._compare_edges��otherr: c C s� t |t�stS | jr |jr dS | j|jkr.dS | j}| jd }|j}|jd }| j}| jd }|j}|jd } | �||||�dko�| �|||| �dkS )z�Compare this range to the `other` taking into account bounds inclusivity, returning ``True`` if they are equal. TFr r )rR r* �NotImplementedr1 r+ r/ r, r] � r5 r_ �slower�slower_b�olower�olower_b�supper�supper_b�oupper�oupper_br6 r6 r7 �__eq__+ s" �z Range.__eq__z Range[_T]c C s� | j r dS |j rdS | j}| jd }|j}|jd }| �||||�dk rLdS | j}| jd }|j}|jd } | �|||| �dkr�dS dS )z7Determine whether this range is a contained by `other`.TFr r )r1 r+ r/ r] r, ra r6 r6 r7 � contained_byF s" zRange.contained_byzUnion[_T, Range[_T]]c C s" t |t�r|�| �S | �|�S dS )z.Determine whether this range contains `value`.N)rR r* rk rP rO r6 r6 r7 �containsc s zRange.containsc C s� | j s |j rdS | j}| jd }| j}| jd }|j}|jd }|j}|jd } | �||||�dkr|| �|||| �dkr|dS | �||||�dkr�| �||||�dkr�dS dS )z3Determine whether this range overlaps with `other`.Fr r T�r1 r+ r/ r, r] � r5 r_ rb rc rf rg rd re rh ri r6 r6 r7 �overlapsk s* ����zRange.overlapsc C sD | j s |j rdS | j}| jd }|j}|jd }| �||||�dk S )zBDetermine whether this range is completely to the left of `other`.Fr r )r1 r, r/ r+ r] )r5 r_ rf rg rd re r6 r6 r7 �strictly_left_of� s zRange.strictly_left_ofc C sD | j s |j rdS | j}| jd }|j}|jd }| �||||�dkS )zCDetermine whether this range is completely to the right of `other`.Fr r rm )r5 r_ rb rc rh ri r6 r6 r7 �strictly_right_of� s zRange.strictly_right_ofc C sD | j s |j rdS | j}| jd }|j}|jd }| �||||�dkS )z>Determine whether this does not extend to the left of `other`.Fr )r1 r+ r/ r] )r5 r_ rb rc rd re r6 r6 r7 �not_extend_left_of� s zRange.not_extend_left_ofc C sD | j s |j rdS | j}| jd }|j}|jd }| �||||�dkS )z?Determine whether this does not extend to the right of `other`.Fr r )r1 r, r/ r] )r5 r_ rf rg rh ri r6 r6 r7 �not_extend_right_of� s zRange.not_extend_right_of)rV rW rX rY r: c C s� | � ||||d�}|dkrr| �� }|dkr.dS |dkrT|dkrJ||| kS ||kS q�|dkrd||kS ||| kS n`|dkr�|dkr�|dks�|dkr�|d kr�| �� }|dk r�dS |dkr�|dkp�|dko�|d kS dS dS ) zTDetermine whether an upper bound is immediately successive to a lower bound.Tr[ NFrF rB r rM rN )r] rT )r5 rV rW rX rY �resr\ r6 r6 r7 � _upper_edge_adjacent_to_lower� s> ������z#Range._upper_edge_adjacent_to_lowerc C sp | j s |j rdS | j}| jd }| j}| jd }|j}|jd }|j}|jd } | �||||�pn| �|| ||�S )z8Determine whether this range is adjacent to the `other`.Fr r )r1 r+ r/ r, ru rn r6 r6 r7 � adjacent_to� s. � ��zRange.adjacent_toc C s� | j r |S |j r| S | �|�s0| �|�s0td��| j}| jd }| j}| jd }|j}|jd }|j}|jd } | �||||�dk r�|} |} n|} |} | �|||| �dkr�|} |} n|} | } t| | t t | | �d�S )z�Compute the union of this range with the `other`. This raises a ``ValueError`` exception if the two ranges are "disjunct", that is neither adjacent nor overlapping. zAAdding non-overlapping and non-adjacent ranges is not implementedr r rC ) r1 ro rv � ValueErrorr+ r/ r, r] r* r r. �r5 r_ rb rc rf rg rd re rh ri Zrlower�rlower_bZrupper�rupper_br6 r6 r7 �union s> � �z Range.unionc C s | � |�S r; )r{ �r5 r_ r6 r6 r7 �__add__> s z Range.__add__c C s� | j s |j r| S | j}| jd }| j}| jd }|j}|jd }|j}|jd } | �||||�} | �|||| �} | dk r�| dkr�td��| �|||| �} | �||||�} | dks�| dk r�| S | dkr�| dkr�tdddd�S | dk�rZ| dk�rZ| dk�rZ|dk�rdnd }|dk�rB|d k�rB| �||||�dk�rBtdddd�S t||tt|| �d �S | dk�r�| dk�r�| dk�r�| d k�r�d nd}|dk�r�|d k�r�| �||||�dk�r�tdddd�S t||tt|| �d �S d �s�t d | � d|� ���dS )z�Compute the difference between this range and the `other`. This raises a ``ValueError`` exception if the two ranges are "disjunct", that is neither adjacent nor overlapping. r r z5Subtracting a strictly inner range is not implementedNTr< rB rM rF rC rN FzUnhandled case computing z - ) r1 r+ r/ r, r] rw r* r r. �AssertionError)r5 r_ rb rc rf rg rd re rh ri Zsl_vs_olZsu_vs_ouZsl_vs_ouZsu_vs_olrz ry r6 r6 r7 � differenceA sn � ���� � ���� �zRange.differencec C s | � |�S r; )r r| r6 r6 r7 �__sub__� s z Range.__sub__c C s� | j s|j s| �|�s$tdddd�S | j}| jd }| j}| jd }|j}|jd }|j}|jd } | �||||�dk r�|} |} n|} |} | �|||| �dkr�|} | } n|} |} t| | tt| | �d�S )zdCompute the intersection of this range with the `other`. .. versionadded:: 2.0.10 NTr< r r rC ) r1 ro r* r+ r/ r, r] r r. rx r6 r6 r7 � intersection� s2 �zRange.intersectionc C s | � |�S r; )r� r| r6 r6 r7 �__mul__� s z Range.__mul__c C s | � � S r; )� _stringifyr= r6 r6 r7 �__str__� s z Range.__str__c C s^ | j r dS | j| j }}|d kr$dn|}|d kr4dn|}td| j�\}}|� |� d|� |� �S )Nr1 � zTuple[str, str]�,)r1 r+ r, r r/ )r5 �l�rZb0�b1r6 r6 r7 r� � s zRange._stringify)NN)F)/�__name__� __module__� __qualname__�__doc__r+ �__annotations__r, r � dataclasses�fieldr/ r1 � dc_kwonlyr r8 r> �propertyr@ rA rD rE rG rH rJ rP rT r] rj rk rl ro rp � __lshift__rq � __rshift__rr rs ru rv r{ r} r r� r� r� r� r� r6 r6 r6 r7 r* ; sl �� �Y /.L&r* c @ s" e Zd ZdZedd�dd��ZdS )� MultiRangea? Represents a multirange sequence. This list subclass is an utility to allow automatic type inference of the proper multi-range SQL type depending on the single range values. This is useful when operating on literal multi-ranges:: import sqlalchemy as sa from sqlalchemy.dialects.postgresql import MultiRange, Range value = literal(MultiRange([Range(2, 4)])) select(tbl).where(tbl.c.value.op("@")(MultiRange([Range(-3, 7)]))) .. versionadded:: 2.0.26 .. seealso:: - :ref:`postgresql_multirange_list_use`. zAbstractMultiRange[_T]r9 c C s t � S r; )�AbstractMultiRanger= r6 r6 r7 rJ � s z MultiRange.__sa_type_engine__N)r� r� r� r� r� rJ r6 r6 r6 r7 r� � s r� c sx e Zd ZdZdZdZedddd�dd��Zed dd d�d d��Zd dd d�� fd d� ZG dd� dej e e �Z � Z S )� AbstractRangez0Base class for single and multi Range SQL types.Tz Type[_TE]r r# )�cls�kwr: c K s d S r; r6 �r5 r� r� r6 r6 r7 �adapt� s zAbstractRange.adaptzType[TypeEngineMixin]zTypeEngine[Any]c K s d S r; r6 r� r6 r6 r7 r� � s z-Type[Union[TypeEngine[Any], TypeEngineMixin]]c sL t |ttf�r<|| jk r<| j}t|� d�|| jfd|i�� S t� �|�S dS )a Dynamically adapt a range type to an abstract impl. For example ``INT4RANGE().adapt(_Psycopg2NumericRange)`` should produce a type that will have ``_Psycopg2NumericRange`` behaviors and also render as ``INT4RANGE`` in SQL and DDL. Z RangeImpl�__visit_name__N)� issubclass�AbstractSingleRangeImpl�AbstractMultiRangeImpl� __class__r� �type�superr� )r5 r� r� Z visit_name�r� r6 r7 r� � s �� �c @ s� e Zd ZdZdddd�dd�Zddd�dd �Zddd�d d �Zddd�d d �ZeZddd�dd�Z e Z ddd�dd�Z ddd�dd�Z ddd�dd�Z ddd�dd�Zddd�dd�Zddd�dd �Zd S ) z AbstractRange.comparator_factoryz-Define comparison operations for range types.r zColumnElement[bool])r_ r� r: c K s | j �t|�S )a Boolean expression. Returns true if the right hand operand, which can be an element or a range, is contained within the column. kwargs may be ignored by this operator but are required for API conformance. )�expr�operater )r5 r_ r� r6 r6 r7 rl s z)AbstractRange.comparator_factory.containsr^ c C s | j �t|�S )zsBoolean expression. Returns true if the column is contained within the right hand operand. )r� r� r r| r6 r6 r7 rk ! s z-AbstractRange.comparator_factory.contained_byc C s | j �t|�S )z�Boolean expression. Returns true if the column overlaps (has points in common with) the right hand operand. )r� r� r r| r6 r6 r7 ro ' s z)AbstractRange.comparator_factory.overlapsc C s | j �t|�S )zsBoolean expression. Returns true if the column is strictly left of the right hand operand. )r� r� r r| r6 r6 r7 rp - s z1AbstractRange.comparator_factory.strictly_left_ofc C s | j �t|�S )ztBoolean expression. Returns true if the column is strictly right of the right hand operand. )r� r� r r| r6 r6 r7 rq 5 s z2AbstractRange.comparator_factory.strictly_right_ofc C s | j �t|�S )z�Boolean expression. Returns true if the range in the column does not extend right of the range in the operand. )r� r� r r| r6 r6 r7 rs = s z4AbstractRange.comparator_factory.not_extend_right_ofc C s | j �t|�S )z�Boolean expression. Returns true if the range in the column does not extend left of the range in the operand. )r� r� r r| r6 r6 r7 rr C s z3AbstractRange.comparator_factory.not_extend_left_ofc C s | j �t|�S )z}Boolean expression. Returns true if the range in the column is adjacent to the range in the operand. )r� r� r r| r6 r6 r7 rv I s z,AbstractRange.comparator_factory.adjacent_toc C s | j �tj|�S �z�Range expression. Returns the union of the two ranges. Will raise an exception if the resulting range is not contiguous. )r� r� r �addr| r6 r6 r7 r{ O s z&AbstractRange.comparator_factory.unionc C s | j �tj|�S r� )r� r� r �subr| r6 r6 r7 r V s z+AbstractRange.comparator_factory.differencezColumnElement[Range[_T]]c C s | j �tj|�S )z�Range expression. Returns the intersection of the two ranges. Will raise an exception if the resulting range is not contiguous. )r� r� r �mulr| r6 r6 r7 r� ] s z-AbstractRange.comparator_factory.intersectionN)r� r� r� r� rl rk ro rp r� rq r� rs rr rv r{ r r� r6 r6 r6 r7 �comparator_factory s r� )r� r� r� r� Zrender_bind_cast� __abstract__r r� r Z Comparatorr* r r� � __classcell__r6 r6 r� r7 r� � s "r� c @ s$ e Zd ZdZdZddd�dd�ZdS ) rI z�Base for PostgreSQL RANGE types. These are types that return a single :class:`_postgresql.Range` object. .. seealso:: `PostgreSQL range functions <https://www.postgresql.org/docs/current/static/functions-range.html>`_ Tz Range[Any]r rK c C s� |j d k r|j n|j}t|t�r6t|�r.t� S t� S nFt|ttf�rJt � S t|t �rf|j s`t � S t � S t|t�rvt� S tjS d S r; )r+ r, rR rS � _is_int32� INT4RANGE� INT8RANGEr �float�NUMRANGEr �tzinfo�TSRANGE� TSTZRANGEr � DATERANGE�sqltypes�NULLTYPE)r5 rL �specr6 r6 r7 �_resolve_for_literalr s z(AbstractSingleRange._resolve_for_literalN�r� r� r� r� r� r� r6 r6 r6 r7 rI e s rI c @ s e Zd ZdZdS )r� zQMarker for AbstractSingleRange that will apply a subclass-specific adaptationN�r� r� r� r� r6 r6 r6 r7 r� � s r� c @ s$ e Zd ZdZdZddd�dd�ZdS ) r� z�Base for PostgreSQL MULTIRANGE types. these are types that return a sequence of :class:`_postgresql.Range` objects. TzSequence[Range[Any]]r rK c C s� |s t jS |d }|jd k r"|jn|j}t|t�rRtdd� |D ��rJt� S t� S nFt|t t f�rft � S t|t �r�|j s|t� S t� S t|t�r�t� S t jS d S )Nr c s s | ]}t |�V qd S r; )r� )�.0r� r6 r6 r7 � <genexpr>� s z:AbstractMultiRange._resolve_for_literal.<locals>.<genexpr>)r� r� r+ r, rR rS �all�INT4MULTIRANGE�INT8MULTIRANGEr r� � NUMMULTIRANGEr r� � TSMULTIRANGE�TSTZMULTIRANGEr �DATEMULTIRANGE)r5 rL �firstr� r6 r6 r7 r� � s z'AbstractMultiRange._resolve_for_literalNr� r6 r6 r6 r7 r� � s r� c @ s e Zd ZdZdS )r� zPMarker for AbstractMultiRange that will apply a subclass-specific adaptationNr� r6 r6 r6 r7 r� � s r� c @ s e Zd ZdZd ZdS )r� z(Represent the PostgreSQL INT4RANGE type.N�r� r� r� r� r� r6 r6 r6 r7 r� � s r� c @ s e Zd ZdZd ZdS )r� z(Represent the PostgreSQL INT8RANGE type.Nr� r6 r6 r6 r7 r� � s r� c @ s e Zd ZdZd ZdS )r� z'Represent the PostgreSQL NUMRANGE type.Nr� r6 r6 r6 r7 r� � s r� c @ s e Zd ZdZd ZdS )r� z(Represent the PostgreSQL DATERANGE type.Nr� r6 r6 r6 r7 r� � s r� c @ s e Zd ZdZd ZdS )r� �&Represent the PostgreSQL TSRANGE type.Nr� r6 r6 r6 r7 r� � s r� c @ s e Zd ZdZd ZdS )r� �(Represent the PostgreSQL TSTZRANGE type.Nr� r6 r6 r6 r7 r� � s r� c @ s e Zd ZdZd ZdS )r� z-Represent the PostgreSQL INT4MULTIRANGE type.Nr� r6 r6 r6 r7 r� � s r� c @ s e Zd ZdZd ZdS )r� z-Represent the PostgreSQL INT8MULTIRANGE type.Nr� r6 r6 r6 r7 r� � s r� c @ s e Zd ZdZd ZdS )r� z,Represent the PostgreSQL NUMMULTIRANGE type.Nr� r6 r6 r6 r7 r� � s r� c @ s e Zd ZdZd ZdS )r� z-Represent the PostgreSQL DATEMULTIRANGE type.Nr� r6 r6 r6 r7 r� � s r� c @ s e Zd ZdZd ZdS )r� r� Nr� r6 r6 r6 r7 r� � s r� c @ s e Zd ZdZd ZdS )r� r� Nr� r6 r6 r6 r7 r� � s r� i���i �z Range[int]r0 )r� r: c C sH | j d ks$t| j ko tkn oF| jd kpFt| j koBtkS S r; )r+ � _min_int_32� _max_int_32r, )r� r6 r6 r7 r� s $"�r� )H� __future__r r� r r r �decimalr �typingr r r r r r r r r r r r r r r r r r r r r r� r r� �sqlZ sql.type_apir �utilr Z util.typingr! Z sql.elementsr" r# r$ r% r. Zdc_slotsr� � dataclassr* r� r� rI r� r� r� rS r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r6 r6 r6 r7 �<module> s� # $