xH d Z ddlZddlmZmZ ddlmZmZmZ ddlm Z m
Z
m Z G d d Z d.d Z
G d
d Z G d d
Z G d de Z G d de Z G d de Z G d de Z G d de Z eee eeeddd Ze eeed d d Ze eeed d d! Ze ed"ed#d$d% Ze ed&ed'd(d) Ze ed*ed+d,d- Ze dS )/z1Implementation of the Range type and adaptation
N)ProgrammingErrorInterfaceError) ISQLQuoteadaptregister_adapter)new_typenew_array_type
register_typec e Zd ZdZdZddZd Zd Zed Z ed
Z
ed Z ed Z ed
Z
ed Zed Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd Zd ZdS ) Rangea Python representation for a PostgreSQL |range|_ type.
:param lower: lower bound for the range. `!None` means unbound
:param upper: upper bound for the range. `!None` means unbound
:param bounds: one of the literal strings ``()``, ``[)``, ``(]``, ``[]``,
representing whether the lower or upper bounds are included
:param empty: if `!True`, the range is empty
_lower_upper_boundsN[)Fc |s-|dvrt d| || _ || _ || _ d S d x| _ x| _ | _ d S )N)r z(]z()z[]zbound flags not valid: )
ValueErrorr r r )selflowerupperboundsemptys c/builddir/build/BUILD/cloudlinux-venv-1.0.12/venv/lib64/python3.11/site-packages/psycopg2/_range.py__init__zRange.__init__/ s_ < 555 !E6!E!EFFF DK DK!DLLL7; ;DK ;$+ c | j | j j dS d | j j | j | j | j S )Nz (empty=True)z{}({!r}, {!r}, {!r}))r __class____name__formatr r r s r __repr__zRange.__repr__: sJ < n-;;; ;)00 1H T[$,8 8
8r c | j dS | j d t | j dt | j | j d g}d | S )Nr r z, )r strr r join)r itemss r __str__z
Range.__str__A sY < 7
LO LO
wwu~~ r c | j S )z:The lower bound of the range. `!None` if empty or unbound.)r r s r r z Range.lowerN {r c | j S )z:The upper bound of the range. `!None` if empty or unbound.)r r s r r z Range.upperS r* r c | j du S )z `!True` if the range is empty.Nr r s r isemptyz
Range.isemptyX s |t##r c & | j dS | j du S )z0`!True` if the range doesn't have a lower bound.NFr r r s r lower_infzRange.lower_inf] < 5{d""r c & | j dS | j du S )z1`!True` if the range doesn't have an upper bound.NFr r r s r upper_infzRange.upper_infd r2 r c D | j | j dS | j d dk S )z4`!True` if the lower bound is included in the range.NFr [r0 r s r lower_inczRange.lower_inck ) < 4;#65|A#%%r c D | j | j dS | j d dk S )z4`!True` if the upper bound is included in the range.NFr# ]r4 r s r upper_inczRange.upper_incr r9 r c | j dS | j ,| j d dk r|| j k rdS n
|| j k rdS | j ,| j d dk r|| j k rdS n
|| j k rdS dS )NFr r7 r# r; T)r r r )r xs r __contains__zRange.__contains__y s < 5 ; "|A#%%t{?? 5 # ## 5 ; "|A#%%t{?? 5 # ## 5tr c | j d uS Nr- r s r __bool__zRange.__bool__ s |4''r c t |t sdS | j |j k o | j |j k o| j |j k S )NF)
isinstancer r r r r others r __eq__z Range.__eq__ sL %'' 5 u|+ . u|+.
- /r c . | | S rA )rG rE s r __ne__z Range.__ne__ s ;;u%%%%r c D t | j | j | j f S rA )hashr r r r s r __hash__zRange.__hash__ s T[$+t|<===r c t |t st S dD ]:}t | | }t || }||k r)| dS | dS ||k c S dS )Nr
TF)rD r NotImplementedgetattr)r rF attr
self_value other_values r __lt__z Range.__lt__ s %'' "! !3
0
0D t,,J!%..K[((#tt $ uu!K////ur c < | |k rdS | | S NT)rS rE s r __le__z Range.__le__ " 5==4;;u%% %r c d t |t r| | S t S rA )rD r rS rN rE s r __gt__z Range.__gt__ s, eU # # "<<%% %! !r c < | |k rdS | | S rU )rY rE s r __ge__z Range.__ge__ rW r c * fd j D S )Nc R i | ]#}t | |t | $S )hasattrrO ).0slotr s r
z&Range.__getstate__.. sC ? ? ?'$*=*=?gdD)) ? ? ?r ) __slots__r s `r __getstate__zRange.__getstate__ s0 ? ? ? ? ? ? ? ?r c \ | D ]\ }}t | || d S rA )r' setattr)r statera values r __setstate__zRange.__setstate__ s< ;;== ' 'KD% D$ & & & & ' 'r )NNr F)r
__module__ __qualname____doc__rc r r! r( propertyr r r. r1 r5 r8 r< r? rB rG rI rL rS rV rY r[ rd ri r^ r r r r # s 0I < < < <8 8 8 X X $ $ X$ # # X# # # X# & & X& & & X& ,( ( (/ / /& & &> > > & & & " " " & & & ? ? ?' ' ' ' 'r r Fc r t | || }| | r|pd |S )a Create and register an adapter and the typecasters to convert between
a PostgreSQL |range|_ type and a PostgreSQL `Range` subclass.
:param pgrange: the name of the PostgreSQL |range| type. Can be
schema-qualified
:param pyrange: a `Range` strict subclass, or just a name to give to a new
class
:param conn_or_curs: a connection or cursor used to find the oid of the
range and its subtype; the typecaster is registered in a scope limited
to this object, unless *globally* is set to `!True`
:param globally: if `!False` (default) register the typecaster only on
*conn_or_curs*, otherwise register it globally
:return: `RangeCaster` instance responsible for the conversion
If a string is passed to *pyrange*, a new `Range` subclass is created
with such name and will be available as the `~RangeCaster.range` attribute
of the returned `RangeCaster` object.
The function queries the database on *conn_or_curs* to inspect the
*pgrange* type and raises `~psycopg2.ProgrammingError` if the type is not
found. If querying the database is not advisable, use directly the
`RangeCaster` class and register the adapter and typecasters using the
provided functions.
N) RangeCaster_from_db _register)pgrangepyrange conn_or_cursgloballycasters r register_rangerw s@ 4
!
!'7L
A
AF
\2l:d;;; Mr c . e Zd ZdZdZd Zd Zd Zd ZdS ) RangeAdapterz`ISQLQuote` adapter for `Range` subclasses.
This is an abstract class: concrete classes must set a `name` class
attribute or override `getquoted()`.
Nc || _ d S rA )adapted)r r{ s r r zRangeAdapter.__init__ s
r c & | j t u r| S d S rA )_protor )r protos r __conform__zRangeAdapter.__conform__ s ;) # #K $ #r c || _ d S rA )_conn)r conns r preparezRangeAdapter.prepare s
r c | j t d | j }|j r d| j d z S |j St
|j }t |d r| | j |
}nd}|j St
|j }t |d r| | j |
}nd}| j d dz |z dz |z dz |j d z d z S )
NzMRangeAdapter must be subclassed overriding its name or the getquoted() methods 'empty'::utf8r s NULL (s , s , 's '))
nameNotImplementedErrorr{ r. encoder r r_ r r getquotedr r )r rar r s r r zRangeAdapter.getquoted sA 9 %,- -
-
L 9 ; $)"2"26":":: : 7 agAq)$$
& $*%%%KKMMEEE 7 agAq)$$
& $*%%%KKMMEEEy ''$.6>Fy''//0278 8r ) r rj rk rl r r r r r r^ r r ry ry sa
D 8 8 8 8 8r ry c e Zd ZdZd
dZd Zed Z ej dej
Z ej d Z d
dZ
d
d ZdS ) ro a Helper class to convert between `Range` and PostgreSQL range types.
Objects of this class are usually created by `register_range()`. Manual
creation could be useful if querying the database is not advisable: in
this case the oids must be provided.
Nc || _ | || | j j p| j j j }t
|f|| j | _ |!t |f|dz | j | _
d S d | _
d S )NARRAY) subtype_oid_create_rangesadapterr r r r parse
typecasterr array_typecaster)r rr rs oidr array_oidr s r r zRangeCaster.__init__$ s & GW---| CDL$:$C"C64<< $2 dWndo%? %?D ! ! ! %)D ! ! !r c 4 d| _ t |t r)t |t fi | _ || j _ n7 t
|t r|t ur|| _ n# t $ r Y nw xY w| j t d d| _ t |t r t |t fi | _ t
|t r|t ur|| _ n# t $ r Y nw xY w| j t d dS )z0Create Range and RangeAdapter classes if needed.Nz:pgrange must be a string or a RangeAdapter strict subclassz1pyrange must be a type or a Range strict subclass)
r rD r% typery r
issubclass TypeErrorranger )r rr rs s r r zRangeCaster._create_ranges2 sN gs # # , "==DL 'DL
g|44 + #< 7 7#*DL
< LN N
N
'3''
9 !'E8R 8 8
'5))
%gU.B.B $
D : CE E
E s% %A- -
A:9A:AC2 2
C?>C?c 6 ddl m} ddlm} || \ }}|j j dk r t
d|j j z |j }d|v r| dd \ } }
n|}
d} | d |
| f |
} | s d
} |j |k r| d d } | d
|f |
} | r
| dd \ }
} n# t $ r Y nw xY w| r| d n # | r| d w w xY w||k r|j s| | st
d| d | dd \ }
}}t |||
|| S )z|Return a `RangeCaster` instance for the type *pgrange*.
Raise `ProgrammingError` if the type is not found.
r )STATUS_IN_TRANSACTION)_solve_conn_cursiX` z'range types not available in version %s.r# publiczselect rngtypid, rngsubtype, typarray
from pg_range r
join pg_type t on t.oid = rngtypid
join pg_namespace ns on ns.oid = typnamespace
where typname = %s and ns.nspname = %s;
FzSAVEPOINT register_typeTzSELECT rngtypid, rngsubtype, typarray, typname, nspname
from pg_range r
join pg_type t on t.oid = rngtypid
join pg_namespace ns on ns.oid = typnamespace
WHERE t.oid = %s::regtype
Nz#ROLLBACK TO SAVEPOINT register_typezPostgreSQL range 'z ' not foundr r r )psycopg2.extensionsr psycopg2.extrasr infoserver_versionr statussplitexecutefetchone
autocommitrollbackro )r r rs rt r r r curs conn_statusschematnamerec savepointr subtypearrays r rp zRangeCaster._from_dbU sO >=====444444%%l33
d 9 #e + +"#L)*$+ , ,
, k $;; JJsA . .MFEEE F
V_
mmoo H
H ! ;"777LL!:;;; $I X
mmoo ,$'GME6 $
HLL!FGGG HLL!FGGGGH / / / / MMOOO 8"6T6668 8
8 #2A2w gu4'U< < < $D& >
D D&
D D& &E a]
( \(|\[ ) # lower bound flag
(?: # lower bound:
" ( (?: [^"] | "")* ) " # - a quoted string
| ( [^",]+ ) # - or an unquoted string
)? # - or empty (not catched)
,
(?: # upper bound:
" ( (?: [^"] | "")* ) " # - a quoted string
| ( [^"\)\]]+ ) # - or an unquoted string
)? # - or empty (not catched)
( \)|\] ) # upper bound flag
z (["\\])\1c |d S |dk r| d S | j | }|t d| d | d }|2| d }|| j d| }| d }|2| d
}|| j d| }|6| | j | }| | j | }| d | d z }| ||| S )
Nr T)r zfailed to parse range: ''r z\1 r# ) r _re_rangematchr group _re_undoublesubcastr )r scurmr r r s r r zRangeCaster.parse sD 94 <<::D:)) ) N # # 9 !@A!@!@!@AA A
=GGAJJE )--eU;;
=GGAJJE )--eU;; ?HHT -u55EHHT -u55Eaggajj(zz%///r c t | j | | j t | j | t | j | j d S rA )r
r r r r r )r scopes r rq zRangeCaster._register sJ dou--- , $/ 7 7 7T\22222r rA )r rj rk rl r r classmethodrp recompileVERBOSEr r r rq r^ r r ro ro s ) ) ) ) !E !E !EF J< J< [J3 3 3 3 3 3r ro c e Zd ZdZdS ) NumericRangezA `Range` suitable to pass Python numeric types to a PostgreSQL range.
PostgreSQL types :sql:`int4range`, :sql:`int8range`, :sql:`numrange` are
casted into `!NumericRange` instances.
Nr rj rk rl r^ r r r r s
Dr r c e Zd ZdZdS ) DateRangez#Represents :sql:`daterange` values.Nr r^ r r r r --Dr r c e Zd ZdZdS )
DateTimeRangez!Represents :sql:`tsrange` values.Nr r^ r r r r s ++Dr r c e Zd ZdZdS )DateTimeTZRangez#Represents :sql:`tstzrange` values.Nr r^ r r r r r r r c e Zd ZdZd ZdS )NumberRangeAdapterz1Adapt a range if the subtype doesn't need quotes.c | j }|j rdS |j s:t |j d }nd}|j s:t |j d }nd}d|j d | d| |j d d
d S )Ns 'empty'asciir$ r r ,r# ) r{ r. r1 r r r decoder5 r r r )r r r r s r r z NumberRangeAdapter.getquoted s L 9 :{
!'NN,,..55g>>EEE{ !'NN,,..55g>>EEE@AIaL@%@@%@1 @@@HHQQQr N)r rj rk rl r r^ r r r r s. ;;R R R R Rr r i@ iA r iV iW iB i iC daterangeiH i: iI tsrangeiD iZ iE tstzrangeiF i iG )F) rl r psycopg2._psycopgr r r r r r r r r
r rw ry ro r r r r r int4range_casterrq int8range_casternumrange_casterdaterange_castertsrange_castertstzrange_casterr^ r r r s> 6
> > > > > > > > B B B B B B B B B B G G G G G G G G G Gh' h' h' h' h' h' h' h'V >-8 -8 -8 -8 -8 -8 -8 -8`y3 y3 y3 y3 y3 y3 y3 y3x 5
E
e R R R R R R R R2 1 2 2 2
; 1< ". . . ; 1< ". . . + 0, $$0 0 0 ;{I $$0 0 0 Y
$$0 0 0 ;{O $$0 0 0 r