U ovgҀ@sddlmZddlZddlmZddlmZddlmZddlmZddlm Z ddlm Z dd lm Z dd lm Z dd lm Z dd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZ ddl!mZddl"m#Z#dd l$m%Z%dd!l&m'Z'erdd"l(m)Z)dd#l"m*Z*dd$l"m+Z+ed%e d&Z,e'd'Z-e%rd(d)iZ.d*d)iZ/niZ.iZ/ej0fd+d)ie.Gd,d-d-e e,Z1Gd.d/d/e e1e,Z2Gd0d1d1e j#e,Z3Gd2d3d3e3e1e,Z4Gd4d5d5e4e,Z5Gd6d7d7e3ee1e,Z6Gd8d9d9e6e,Z7Gd:d;d;e4e8Z9Gdd?d?e4eZ;Gd@dAdAe4eZGdFdGdGe6e8Z?GdHdIdIe6e8Z@GdJdKdKe6eZAGdLdMdMe6eZBGdNdOdOe6eZCGdPdQdQe6eZDdRZEdSZFdTdUdVdWdXZGdS)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_onlyfrozenc@seZdZUdZdZded<dZded<erTej ddZ ded <ej d dZ d ed <n8ej fd die Z ded <ej fd d ie Z d ed <e sdXdd ddddd dddZd dddZed dddZed dddZed dddZed dddZed dddZed ddd Zed!dd"d#Zd$d d%d&d'Zd(dd)d*ZdYdd+dd+d d,d-d.d/Zd(d d0d1d2Zd3d d0d4d5Zd6d d%d7d8ZeZd3d d0d9d:Zd3d d0d;d<Z e Z!d3d d0d=d>Z"e"Z#d3d d0d?d@Z$d3d d0dAdBZ%dd+dd+d dCdDdEZ&d3d d0dFdGZ'd3d3d0dHdIZ(d3d3d0dJdKZ)d3d3d0dLdMZ*d3d3d0dNdOZ+d3d3d0dPdQZ,d3d3d0dRdSZ-d+ddTdUZ.d+ddVdWZ/dS)ZRangea_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]lowerupperr')default _BoundsTypeboundsFboolemptyr-)r/r1r+r,r/r1cCs|j||||ddS)Nr2)__dict__update)selfr+r,r/r1r6I/tmp/pip-unpacked-wheel-uqd9otp_/sqlalchemy/dialects/postgresql/ranges.py__init__as zRange.__init__returncCs|j SNr1r5r6r6r7__bool__sszRange.__bool__cCs|jSz$A synonym for the 'empty' attribute.r<r=r6r6r7isemptyvsz Range.isemptycCs|jSr?r<r=r6r6r7is_empty|szRange.is_emptycCs|jddkS)z,Return True if the lower bound is inclusive.r[r/r=r6r6r7 lower_incszRange.lower_inccCs|j o|jdkS)zKReturn True if this range is non-empty and lower bound is infinite.N)r1r+r=r6r6r7 lower_infszRange.lower_infcCs|jddkS)z,Return True if the upper bound is inclusive.r]rCr=r6r6r7 upper_incszRange.upper_inccCs|j o|jdkS)zOReturn True if this range is non-empty and the upper bound is infinite.N)r1r,r=r6r6r7 upper_infszRange.upper_infzAbstractSingleRange[_T]cCstSr;)AbstractSingleRanger=r6r6r7__sa_type_engine__szRange.__sa_type_engine__r%valuer:cCs|jr dS|jdkr@|jdkp>|jddkr6||jkS||jkS|jdkrl|jddkrb||jkS||jkS|jddkr||jkn||jko|jddkr||jkS||jkS)z3Return True if this range contains the given value.FNr)r()r1r+r,r/r5rLr6r6r7_contains_values*       zRange._contains_valuercCsbt|jtst|jtrdSt|jts4t|jtr8dSt|jtsPt|jtrZtddSdSdS)uADetermine the “step” for this range, if it is a discrete one.rN)days) isinstancer+intr,rrrr=r6r6r7_get_discrete_steps zRange._get_discrete_stepstrrS)value1bound1value2bound2 only_valuesr:c Csx|dk}|dk}||kr$dkrBnn||kr4dS|rrBrNNrr>rBrFTF)rT) r5rVrWrXrYrZZvalue1_is_lower_boundZvalue2_is_lower_boundZ value1_incZ value2_incstepr6r6r7_compare_edgess^      zRange._compare_edgesotherr:c Cst|tstS|jr|jrdS|j|jkr.dS|j}|jd}|j}|jd}|j}|jd}|j}|jd} |||||dko||||| dkS)zCompare this range to the `other` taking into account bounds inclusivity, returning ``True`` if they are equal. TFrr)rRr*NotImplementedr1r+r/r,r] r5r_slowerslower_bolowerolower_bsuppersupper_boupperoupper_br6r6r7__eq__+s"       z Range.__eq__z Range[_T]c Cs|jr dS|jrdS|j}|jd}|j}|jd}|||||dkrLdS|j}|jd}|j}|jd} ||||| dkrdSdS)z7Determine whether this range is a contained by `other`.TFrr)r1r+r/r]r,rar6r6r7 contained_byFs"    zRange.contained_byzUnion[_T, Range[_T]]cCs"t|tr||S||SdS)z.Determine whether this range contains `value`.N)rRr*rkrPrOr6r6r7containscs  zRange.containsc Cs|js |jrdS|j}|jd}|j}|jd}|j}|jd}|j}|jd} |||||dkr|||||| dkr|dS|||||dkr|||||dkrdSdS)z3Determine whether this range overlaps with `other`.FrrTr1r+r/r,r] r5r_rbrcrfrgrdrerhrir6r6r7overlapsms*     zRange.overlapscCsD|js |jrdS|j}|jd}|j}|jd}|||||dkS)zBDetermine whether this range is completely to the left of `other`.Frr)r1r,r/r+r])r5r_rfrgrdrer6r6r7strictly_left_ofs   zRange.strictly_left_ofcCsD|js |jrdS|j}|jd}|j}|jd}|||||dkS)zCDetermine whether this range is completely to the right of `other`.Frrrm)r5r_rbrcrhrir6r6r7strictly_right_ofs   zRange.strictly_right_ofcCsD|js |jrdS|j}|jd}|j}|jd}|||||dkS)z>Determine whether this does not extend to the left of `other`.Fr)r1r+r/r])r5r_rbrcrdrer6r6r7not_extend_left_ofs   zRange.not_extend_left_ofcCsD|js |jrdS|j}|jd}|j}|jd}|||||dkS)z?Determine whether this does not extend to the right of `other`.Frr)r1r,r/r])r5r_rfrgrhrir6r6r7not_extend_right_ofs   zRange.not_extend_right_of)rVrWrXrYr:cCs|||||d}|dkrr|}|dkr.dS|dkrT|dkrJ|||kS||kSq|dkrd||kS|||kSn`|dkr|dkr|dks|dkr|d kr|}|dk rdS|dkr|dkp|dko|d kSdSdS) zTDetermine whether an upper bound is immediately successive to a lower bound.Tr[NFrFrBrrMrN)r]rT)r5rVrWrXrYresr\r6r6r7_upper_edge_adjacent_to_lowers>   z#Range._upper_edge_adjacent_to_lowerc Csp|js |jrdS|j}|jd}|j}|jd}|j}|jd}|j}|jd} |||||pn||| ||S)z8Determine whether this range is adjacent to the `other`.Frr)r1r+r/r,rurnr6r6r7 adjacent_tos.     zRange.adjacent_tocCs|jr |S|jr|S||s0||s0td|j}|jd}|j}|jd}|j}|jd}|j}|jd} |||||dkr|} |} n|} |} ||||| dkr|} |} n|} | } t| | t t | | dS)zCompute 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 implementedrrrC) r1rorv ValueErrorr+r/r,r]r*rr.r5r_rbrcrfrgrdrerhriZrlowerrlower_bZrupperrupper_br6r6r7unions>     z Range.unioncCs ||Sr;)r{r5r_r6r6r7__add__@sz Range.__add__cCs|js |jr|S|j}|jd}|j}|jd}|j}|jd}|j}|jd} |||||} ||||| } | dkr| dkrtd||||| } |||||} | dks| dkr|S| dkr| dkrtddddS| dkrZ| dkrZ| dkrZ|dkrdnd }|dkrB|d krB|||||dkrBtddddSt||tt||d S| dkr| dkr| dkr| d krd nd}|dkr|d kr|||||dkrtddddSt||tt||d Sd st d |d|dS)zCompute 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. rrz5Subtracting a strictly inner range is not implementedNTr<rBrMrFrCrNFzUnhandled case computing z - ) r1r+r/r,r]rwr*rr.AssertionError)r5r_rbrcrfrgrdrerhriZsl_vs_olZsu_vs_ouZsl_vs_ouZsu_vs_olrzryr6r6r7 differenceCsn       zRange.differencecCs ||Sr;)rr|r6r6r7__sub__sz Range.__sub__cCs|js|js||s$tddddS|j}|jd}|j}|jd}|j}|jd}|j}|jd} |||||dkr|} |} n|} |} ||||| dkr|} | } n|} |} t| | tt| | dS)zdCompute the intersection of this range with the `other`. .. versionadded:: 2.0.10 NTr<rrrC) r1ror*r+r/r,r]rr.rxr6r6r7 intersections2     zRange.intersectioncCs ||Sr;)rr|r6r6r7__mul__sz Range.__mul__cCs|Sr;) _stringifyr=r6r6r7__str__sz Range.__str__cCs^|jr dS|j|j}}|dkr$dn|}|dkr4dn|}td|j\}}||d||S)Nr1zTuple[str, str],)r1r+r,rr/)r5lrZb0Zb1r6r6r7rszRange._stringify)NN)F)0__name__ __module__ __qualname____doc__r+__annotations__r,r dataclassesfieldr/r1 dc_kwonlyr r8r>propertyr@rArDrErGrHrJrPrTr]rjrkrl __contains__rorp __lshift__rq __rshift__rrrsrurvr{r}rrrrrrr6r6r6r7r*;sn   Y /.L&r*c@s"eZdZdZeddddZdS) 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]r9cCstSr;)AbstractMultiRanger=r6r6r7rJszMultiRange.__sa_type_engine__N)rrrrrrJr6r6r6r7rsrcsxeZdZdZdZdZeddddddZed dd dd dZd dd dfd d ZGdddej e e Z Z S) AbstractRangez0Base class for single and multi Range SQL types.Tz Type[_TE]rr#)clskwr:cKsdSr;r6r5rrr6r6r7adaptszAbstractRange.adaptzType[TypeEngineMixin]zTypeEngine[Any]cKsdSr;r6rr6r6r7rsz-Type[Union[TypeEngine[Any], TypeEngineMixin]]c sLt|ttfr<||jk r<|j}t|d||jfd|iSt|SdS)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) issubclassAbstractSingleRangeImplAbstractMultiRangeImpl __class__rtypesuperr)r5rrZ visit_namerr6r7rs  c@seZdZdZddddddZddddd Zdddd d Zdddd d ZeZdddddZ e Z dddddZ dddddZ dddddZ dddddZdddddZdddddZdS)z AbstractRange.comparator_factoryz-Define comparison operations for range types.rzColumnElement[bool])r_rr:cKs|jt|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. )exproperater)r5r_rr6r6r7rlsz)AbstractRange.comparator_factory.containsr^cCs|jt|S)zsBoolean expression. Returns true if the column is contained within the right hand operand. )rrrr|r6r6r7rk#sz-AbstractRange.comparator_factory.contained_bycCs|jt|S)zBoolean expression. Returns true if the column overlaps (has points in common with) the right hand operand. )rrrr|r6r6r7ro)sz)AbstractRange.comparator_factory.overlapscCs|jt|S)zsBoolean expression. Returns true if the column is strictly left of the right hand operand. )rrrr|r6r6r7rp/sz1AbstractRange.comparator_factory.strictly_left_ofcCs|jt|S)ztBoolean expression. Returns true if the column is strictly right of the right hand operand. )rrrr|r6r6r7rq7sz2AbstractRange.comparator_factory.strictly_right_ofcCs|jt|S)zBoolean expression. Returns true if the range in the column does not extend right of the range in the operand. )rrrr|r6r6r7rs?sz4AbstractRange.comparator_factory.not_extend_right_ofcCs|jt|S)zBoolean expression. Returns true if the range in the column does not extend left of the range in the operand. )rrrr|r6r6r7rrEsz3AbstractRange.comparator_factory.not_extend_left_ofcCs|jt|S)z}Boolean expression. Returns true if the range in the column is adjacent to the range in the operand. )rrrr|r6r6r7rvKsz,AbstractRange.comparator_factory.adjacent_tocCs|jtj|SzRange expression. Returns the union of the two ranges. Will raise an exception if the resulting range is not contiguous. )rrraddr|r6r6r7r{Qsz&AbstractRange.comparator_factory.unioncCs|jtj|Sr)rrrsubr|r6r6r7rXsz+AbstractRange.comparator_factory.differencezColumnElement[Range[_T]]cCs|jtj|S)zRange expression. Returns the intersection of the two ranges. Will raise an exception if the resulting range is not contiguous. )rrrmulr|r6r6r7r_sz-AbstractRange.comparator_factory.intersectionN)rrrrrlrkrorprrqrrsrrrvr{rrr6r6r6r7comparator_factorys r)rrrrZrender_bind_cast __abstract__r rrZ Comparatorr*rr __classcell__r6r6rr7rs"rc@s$eZdZdZdZdddddZdS) rIzBase for PostgreSQL RANGE types. These are types that return a single :class:`_postgresql.Range` object. .. seealso:: `PostgreSQL range functions `_ Tz Range[Any]rrKcCs|jdk r|jn|j}t|tr6t|r.tStSnFt|ttfrJt St|t rf|j s`t St St|trvtStjSdSr;)r+r,rRrS _is_int32 INT4RANGE INT8RANGErfloatNUMRANGErtzinfoTSRANGE TSTZRANGEr DATERANGEsqltypesNULLTYPE)r5rLspecr6r6r7_resolve_for_literalts   z(AbstractSingleRange._resolve_for_literalNrrrrrrr6r6r6r7rIgs rIc@seZdZdZdS)rzQMarker for AbstractSingleRange that will apply a subclass-specific adaptationNrrrrr6r6r6r7rsrc@s$eZdZdZdZdddddZdS) rzBase for PostgreSQL MULTIRANGE types. these are types that return a sequence of :class:`_postgresql.Range` objects. TzSequence[Range[Any]]rrKcCs|s tjS|d}|jdk r"|jn|j}t|trRtdd|DrJtStSnFt|t t frft St|t r|j s|tStSt|trtStjSdS)Nrcss|]}t|VqdSr;)r).0rr6r6r7 sz:AbstractMultiRange._resolve_for_literal..)rrr+r,rRrSallINT4MULTIRANGEINT8MULTIRANGErr NUMMULTIRANGErr TSMULTIRANGETSTZMULTIRANGErDATEMULTIRANGE)r5rLfirstrr6r6r7rs   z'AbstractMultiRange._resolve_for_literalNrr6r6r6r7rsrc@seZdZdZdS)rzPMarker for AbstractMultiRange that will apply a subclass-specific adaptationNrr6r6r6r7rsrc@seZdZdZdZdS)rz(Represent the PostgreSQL INT4RANGE type.Nrrrrrr6r6r6r7rsrc@seZdZdZdZdS)rz(Represent the PostgreSQL INT8RANGE type.Nrr6r6r6r7rsrc@seZdZdZdZdS)rz'Represent the PostgreSQL NUMRANGE type.Nrr6r6r6r7rsrc@seZdZdZdZdS)rz(Represent the PostgreSQL DATERANGE type.Nrr6r6r6r7rsrc@seZdZdZdZdS)r&Represent the PostgreSQL TSRANGE type.Nrr6r6r6r7rsrc@seZdZdZdZdS)r(Represent the PostgreSQL TSTZRANGE type.Nrr6r6r6r7rsrc@seZdZdZdZdS)rz-Represent the PostgreSQL INT4MULTIRANGE type.Nrr6r6r6r7rsrc@seZdZdZdZdS)rz-Represent the PostgreSQL INT8MULTIRANGE type.Nrr6r6r6r7rsrc@seZdZdZdZdS)rz,Represent the PostgreSQL NUMMULTIRANGE type.Nrr6r6r6r7rsrc@seZdZdZdZdS)rz-Represent the PostgreSQL DATEMULTIRANGE type.Nrr6r6r6r7rsrc@seZdZdZdZdS)rrNrr6r6r6r7rsrc@seZdZdZdZdS)rrNrr6r6r6r7rsriiz Range[int]r0)rr:cCsH|jdks$t|jkotknoF|jdkpFt|jkoBtkSSr;)r+ _min_int_32 _max_int_32r,)rr6r6r7rs$"r)H __future__rrrrrdecimalrtypingrrr r r r r rrrrrrrrrrrrrrrrrZsqlZ sql.type_apirutilr Z util.typingr!Z sql.elementsr"r#r$r%r.Zdc_slotsrZ dataclassr*rrrIrrrrSrrrrrrrrrrrrrrrr6r6r6r7s                                   #$