5dVA @s8dZddlZddlmZmZddlmZmZmZm Z ddlm Z m Z m Z Gddde Zdd d ZGd d d e ZGd dde ZGdddeZGdddeZGdddeZGdddeZGdddeZeeeeeeddddddZejeeedddd dd!Zejeeedd"dd#dd$Zejed%edd&dd'dd(Zejed)edd*dd+dd,Zejed-edd.dd/dd0ZejdS)1z1Implementation of the Range type and adaptation N)ProgrammingErrorInterfaceError) ISQLQuoteadaptregister_adapterb)new_typenew_array_type register_typec@s*eZdZdZd,Zdddddd Zd d Zed d ZeddZ eddZ eddZ eddZ eddZ eddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+ZdS)-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_boundsNz[)FcCs^|sC|dkr%td|n||_||_||_nd|_|_|_dS)N[)(]()[]zbound flags not valid: %r)rrrr) ValueErrorr r r)selflowerupperboundsemptyr3/tmp/pip-build-0jahl3lb/psycopg2/psycopg2/_range.py__init__-s    zRange.__init__cCsD|jdkrd|jjSd|jj|j|j|jfSdS)Nz%s(empty=True)z%s(%r, %r, %r))r __class____name__r r )rrrr__repr__8s zRange.__repr__cCs|jS)z:The lower bound of the range. `!None` if empty or unbound.)r )rrrrr?sz Range.lowercCs|jS)z:The upper bound of the range. `!None` if empty or unbound.)r )rrrrrDsz Range.uppercCs |jdkS)z`!True` if the range is empty.N)r)rrrrisemptyIsz Range.isemptycCs |jdkrdS|jdkS)z0`!True` if the range doesn't have a lower bound.NF)rr )rrrr lower_infNszRange.lower_infcCs |jdkrdS|jdkS)z1`!True` if the range doesn't have an upper bound.NF)rr )rrrr upper_infTszRange.upper_infcCs7|jdkrdS|jdkr&dS|jddkS)z4`!True` if the lower bound is included in the range.NFr[)rr )rrrr lower_incZs zRange.lower_inccCs7|jdkrdS|jdkr&dS|jddkS)z4`!True` if the upper bound is included in the range.NF])rr )rrrr upper_incas zRange.upper_inccCs|jdkrdS|jdk ra|jddkrK||jkr^dSqa||jkradSn|jdk r|jddkr||jkrdSq||jkrdSndS)NFrr"r$r%T)rr r )rxrrr __contains__hszRange.__contains__cCs |jdk S)N)r)rrrr__bool__xszRange.__bool__cCsGt|tsdS|j|jkoF|j|jkoF|j|jkS)NF) isinstancer r r r)rotherrrr__eq__{s z Range.__eq__cCs|j| S)N)r,)rr+rrr__ne__sz Range.__ne__cCst|j|j|jfS)N)hashr r r)rrrr__hash__szRange.__hash__cCst|tstSxedD]]}t||}t||}||krMq|dkr]dS|dkrmdS||kSqWdS)Nr r rTF)z_lowerz_upperz_bounds)r*r NotImplementedgetattr)rr+attrZ self_valueZ other_valuerrr__lt__s    z Range.__lt__cCs!||krdS|j|SdS)NT)r3)rr+rrr__le__s z Range.__le__cCs$t|tr|j|StSdS)N)r*r r3r0)rr+rrr__gt__s z Range.__gt__cCs!||krdS|j|SdS)NT)r5)rr+rrr__ge__s z Range.__ge__)z_lowerz_upperz_bounds)r __module__ __qualname____doc__ __slots__rrpropertyrrrr r!r#r&r(r)r,r-r/r3r4r5r6rrrrr !s(          r FcCs3tj|||}|j| r(|p+d|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_cursZgloballyZcasterrrrregister_rangesrBc@sLeZdZdZdZddZddZddZd d ZdS) RangeAdapterz`ISQLQuote` adapter for `Range` subclasses. This is an abstract class: concrete classes must set a `name` class attribute or override `getquoted()`. NcCs ||_dS)N)adapted)rrDrrrrszRangeAdapter.__init__cCs|jtkr|SdS)N)Z_protor)rprotorrr __conform__szRangeAdapter.__conform__cCs ||_dS)N)_conn)rconnrrrprepareszRangeAdapter.preparecCs|jdkrtdn|j}|jrAtd|jS|jdk rt|j}t|dr|j|j n|j }nd}|j dk rt|j }t|dr|j|j n|j }nd}t|jd|d|td|j S)NzMRangeAdapter must be subclassed overriding its name or the getquoted() methodz 'empty'::rIsNULL(s, z, '%s')) nameNotImplementedErrorrDrrrrhasattrrIrG getquotedrr)rrarrrrrrNs&   zRangeAdapter.getquoted) rr7r8r9rKrrFrIrNrrrrrCs    rCc@seZdZdZdddZddZeddZej d ej Z ej d Z dd d Z dd dZdS)r<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. NcCs||_|j|||jjp1|jjj}t|f||j|_|dk r}t |f|d|j|_ n d|_ dS)NARRAY) subtype_oid_create_rangesadapterrKrrrparse typecasterr array_typecaster)rr?r@oidrR array_oidrKrrrrs  zRangeCaster.__init__cCs2d|_t|tr?t|tfi|_||j_n@y+t|tri|tk ri||_nWntk r~YnX|jdkrtdnd|_yUt|trt|t fi|_nt|t r|t k r||_nWntk rYnX|jdkr.tdndS)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) rTr*strtyperCrK issubclass TypeErrorranger )rr?r@rrrrSs.     zRangeCaster._create_rangescCsQddlm}ddlm}||\}}|jdkrWtd|jn|j}d|kr|jdd\} } n |} d} y|jd | | fWn+tk r|j s|j nYn0X|j } ||kr |j r |j n| s#td |n| \} } }t ||d | d | d |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, (select typarray from pg_type where oid = rngtypid) 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; zPostgreSQL type '%s' not foundrXrRrY) psycopg2.extensionsr_Zpsycopg2.extrasr`Zserver_versionrstatussplitexecuteZ autocommitrollbackZfetchoner<)rrKr@rAr_r`rHZcursZ conn_statusZschematnameZrecr[subtypearrayrrrr=5s8          zRangeCaster._from_dba] ( \(|\[ ) # 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 (["\\])\1cCsg|dkrdS|dkr,|jddS|jj|}|dkr]td|n|jd}|dkr|jd}|dk r|jjd|}qn|jd}|dkr|jd}|dk r|jjd|}qn|dk r8|j|j|}|j|j|}n|jd |jd }|j|||S) NrTzfailed to parse range: '%s'z\1r$) r^ _re_rangematchrgroup _re_undoublesubcastrR)rscurmrrrrrrrU}s,        zRangeCaster.parsecCsIt|j||jdk r2t|j|nt|j|jdS)N)r rVrWrr^rT)rZscoperrrr>szRangeCaster._register)rr7r8r9rrS classmethodr=recompileVERBOSErprsrUr>rrrrr<s  "8 r<c@seZdZdZdS) 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. N)rr7r8r9rrrrr}s r}c@seZdZdZdS) DateRangez#Represents :sql:`daterange` values.N)rr7r8r9rrrrr~s r~c@seZdZdZdS) DateTimeRangez!Represents :sql:`tsrange` values.N)rr7r8r9rrrrrs rc@seZdZdZdS)DateTimeTZRangez#Represents :sql:`tstzrange` values.N)rr7r8r9rrrrrs rc@s"eZdZdZddZdS)NumberRangeAdapterz1Adapt a range if the subtype doesn't need quotes.cCs|j}|jrdS|js@t|jjjd}nd}|jspt|jjjd}nd}d|j d|||j dfj dS)Ns'empty'asciiz '%s%s,%s%s'rr$) rDrr rrrNdecoder!rrencode)rrOrrrrrrNs   ! !zNumberRangeAdapter.getquotedN)rr7r8r9rNrrrrrs rrXi@rRrYiAiViWiBiiCZ daterangeiHi:iIZtsrangeiDiZiEZ tstzrangeiFiiG)r9rzZpsycopg2._psycopgrrrcrrrrrr r objectr rBrCr<r}r~rrrZint4range_casterr>Zint8range_casterZnumrange_casterZdaterange_casterZtsrange_casterZtstzrange_casterrrrrs@ "0