+h>HhdZddlZddlmZmZddlmZmZmZddlm Z m Z m Z GddZ d-dZ Gd d ZGd d ZGd de ZGdde ZGdde ZGdde ZGddeZeeeeeedddZej-eeedddZej-eeeddd Zej-ed!ed"d#d$Zej-ed%ed&d'd(Zej-ed)ed*d+d,Zej-y).z1Implementation of the Range type and adaptation N)ProgrammingErrorInterfaceError) ISQLQuoteadaptregister_adapter)new_typenew_array_type register_typeceZdZdZdZddZdZdZedZ edZ ed Z ed Z ed Z ed Zed ZdZdZdZdZdZdZdZdZdZdZdZy)RangeaPython 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_boundsNc|s(|dvrtd|||_||_||_ydx|_x|_|_y)N)[)z(]z()z[]zbound flags not valid: ) ValueErrorrrr)selflowerupperboundsemptys B/var/www/html/venv/lib/python3.12/site-packages/psycopg2/_range.py__init__zRange.__init__/sK55 #:6*!EFFDKDK!DL7; ;DK ;$+ c|j|jjdSdj|jj|j|j |jS)Nz (empty=True)z{}({!r}, {!r}, {!r}))r __class____name__formatrrrs r__repr__zRange.__repr__:sV << nn--.l; ;)001H1H T[[$,,8 8rc|jy|jdt|jdt|j|jdg}dj |S)Nrrz, )rstrrrjoin)ritemss r__str__z Range.__str__AsY <<  LLO      LLO  wwu~rc|jS)z:The lower bound of the range. `!None` if empty or unbound.)rr s rrz Range.lowerN{{rc|jS)z:The upper bound of the range. `!None` if empty or unbound.)rr s rrz Range.upperSr*rc|jduS)z`!True` if the range is empty.Nrr s risemptyz Range.isemptyXs||t##rc8|jy|jduS)z0`!True` if the range doesn't have a lower bound.NFrrr s r lower_infzRange.lower_inf]  << {{d""rc8|jy|jduS)z1`!True` if the range doesn't have an upper bound.NFrrr s r upper_infzRange.upper_infdr2rcX|j |jy|jddk(S)z4`!True` if the lower bound is included in the range.Fr[r0r s r lower_inczRange.lower_inck- << 4;;#6||A#%%rcX|j |jy|jddk(S)z4`!True` if the upper bound is included in the range.Fr#]r4r s r upper_inczRange.upper_incrr9rc|jy|j2|jddk(r||jkry||jkry|j3|jddk(r||jkDryy||jk\ryy)NFrr7r#r;T)rrr)rxs r __contains__zRange.__contains__ys <<  ;; "||A#%t{{?  # ;; "||A#%t{{?  # rc|jduSNr-r s r__bool__zRange.__bool__s||4''rct|tsy|j|jk(xr4|j|jk(xr|j|jk(S)NF) isinstancer rrrrothers r__eq__z Range.__eq__sN%' u||+. u||+.  - /rc&|j| SrA)rGrEs r__ne__z Range.__ne__s;;u%%%rcZt|j|j|jfSrA)hashrrrr s r__hash__zRange.__hash__s T[[$++t||<==rct|tstSdD].}t||}t||}||k(r!|y|y||kcSy)Nr TF)rDr NotImplementedgetattr)rrFattr self_value other_values r__lt__z Range.__lt__sf%'! !3 0D t,J!%.K[(#$!K// 0rc0||k(ry|j|SNT)rSrEs r__le__z Range.__le__ 5=;;u% %rcPt|tr|j|StSrA)rDr rSrNrEs r__gt__z Range.__gt__s! eU #<<% %! !rc0||k(ry|j|SrU)rYrEs r__ge__z Range.__ge__rWrcn|jDcic]}t||r |t||c}Scc}wrA) __slots__hasattrrO)rslots r __getstate__zRange.__getstate__s;?'$*=gdD))? ??s 2cN|jD]\}}t|||yrA)r'setattr)rstater_values r __setstate__zRange.__setstate__s' ;;= 'KD% D$ & 'r)NNrF)r __module__ __qualname____doc__r]rr!r(propertyrrr.r1r5r8r<r?rBrGrIrLrSrVrYr[r`rerrr r #s0I <8 $$## ## && && ,(/&> & " & ?'rr chtj|||}|j| xr|xsd|S)aCreate 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 rregister_rangerts74 ! !'7L AF \2l:d; Mrc,eZdZdZdZdZdZdZdZy) RangeAdapterz`ISQLQuote` adapter for `Range` subclasses. This is an abstract class: concrete classes must set a `name` class attribute or override `getquoted()`. Nc||_yrA)adapted)rrxs rrzRangeAdapter.__init__s  rc,|jtur|SyrA)_protor)rprotos r __conform__zRangeAdapter.__conform__s ;;) #K $rc||_yrA)_conn)rconns rpreparezRangeAdapter.prepares  rc|j td|j}|jrd|jj dzS|j Mt |j }t|dr|j|j|j}nd}|jMt |j}t|dr|j|j|j}nd}|jj ddz|zdz|zdz|jj dzd zS) NzMRangeAdapter must be subclassed overriding its name or the getquoted() methods 'empty'::utf8rsNULL(s, s, 's')) nameNotImplementedErrorrxr.encoderrr^rr~ getquotedrr)rrarrs rrzRangeAdapter.getquoteds 99 %,- - LL 99$))"2"26":: : 77 aggAq)$ $**%KKMEE 77 aggAq)$ $**%KKMEEyy'$.6>Fyy''/0278 8r) rrfrgrhrrr|rrrjrrrvrvs" D8rrvceZdZdZd dZdZedZejdejZ ejdZ d dZ d d Zy) rla 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. Nc4||_|j|||jjxs |jjj }t |f||j|_|!t|f|dz|j|_ yd|_ y)NARRAY) subtype_oid_create_rangesadapterrrrrparse typecasterr array_typecaster)rrorpoidr array_oidrs rrzRangeCaster.__init__$s& GW-||  CDLL$:$:$C$C"C64<  $2 dWndoo%?D !%)D !rcd|_t|tr)t|tfi|_||j_n t |tr|tur||_|j tdd|_ t|trt|tfi|_t |tr|tur||_|j tdy#t$rYwxYw#t$rY2wxYw)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) rrDr%typervr issubclass TypeErrorranger )rrorps rrzRangeCaster._create_ranges2s  gs #,"=DL 'DLL  g|4#<7#*DL << LN N  '3'!'E8R8 '5)gU.B$  :: CE E !     s%CAC- C*)C*- C98C9cddlm}ddlm}||\}}|jj dkr"t d|jj z|j}d|vr|jdd\} } n|} d} |jd | | f|j} | sg d } |j|k(r|jd d } |jd |f|j} | r | dd\} }  r|jd ||k7r|js|j| st d|d| dd\} }}t||| ||S#t $rYjwxYw# r|jdwwxYw)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 foundrrr)psycopg2.extensionsrpsycopg2.extrasrinfoserver_versionrstatussplitexecutefetchone autocommitrollbackrl)rrrprqrrrcurs conn_statusschematnamerec savepointrsubtypearrays rrmzRangeCaster._from_dbUs >4%l3 d 99 # #e +"#L))**$+, ,kk  $; JJsA.MFEEF   V_  mmo H! ;;"77LL!:; $I   X mmo$'GME6LL!FG / / MMO"$TF+68 8 #2Awgu4'U< <)$  LL!FGs*$7EE$ E!E$ E!!E$$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 (["\\])\1cv|y|dk(r|jdS|jj|}|td|d|j d}|/|j d}||j j d|}|j d }|/|j d }||j j d|}|8|j|j|}|j|j|}|j d |j d z}|j|||S) NrT)rzfailed to parse range: ''rz\1r#) r _re_rangematchrgroup _re_undoublesubcastr)rscurmrrrs rrzRangeCaster.parses( 9 <::D:) ) NN  # 9 #;A3a!@A A  =GGAJE ))--eU;  =GGAJE ))--eU; ?HHT--u5EHHT--u5Eaggaj(zz%//rct|j||jt|j|t|j|j yrA)r rrrrr)rscopes rrnzRangeCaster._registers>doou-  , $// 7T\\2rrA)rrfrgrhrr classmethodrmrecompileVERBOSErrrrnrjrrrlrlsf )!EFJ<J3rrlceZdZdZy) 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. Nrrfrgrhrjrrrrs   rrceZdZdZy) DateRangez#Represents :sql:`daterange` values.Nrrjrrrr-rrceZdZdZy) DateTimeRangez!Represents :sql:`tsrange` values.Nrrjrrrrs+rrceZdZdZy)DateTimeTZRangez#Represents :sql:`tstzrange` values.NrrjrrrrrrrceZdZdZdZy)NumberRangeAdapterz1Adapt a range if the subtype doesn't need quotes.c|j}|jry|js3t|jj j d}nd}|js3t|jj j d}nd}d|jd|d||jddjdS)Ns'empty'asciir$rr,r#) rxr.r1rrrdecoder5rrr)rrrrs rrzNumberRangeAdapter.getquoteds LL 99{{ !''N,,.55g>EE{{!''N,,.55g>EEAIIaL>%%1a@HHQQrN)rrfrgrhrrjrrrrs ;Rrri@iAriViWiBiiC daterangeiHi:iItsrangeiDiZiE tstzrangeiFiiG)F)rhrpsycopg2._psycopgrrrrrrrr r r rtrvrlrrrrrint4range_casterrnint8range_casternumrange_casterdaterange_castertsrange_castertstzrange_casterrjrrrsb6 >BBGGh'h'V>-8-8`y3y3x 5    E  e RR212 1< ".1< ".0, $$0{I $$0Y  $$0{O $$0r