3 (h)E@sdZddlZddlmZmZddlmZmZmZddlm Z m Z m Z ddl m Z GdddeZd0d d ZGd d d eZGdddeZGdddeZGdddeZGdddeZGdddeZGdddeZeeeeeeddddZejeeeddd dZejeeed!d"d#dZejed$ed%d&d'dZejed(ed)d*d+dZejed,ed-d.d/dZejdS)1z1Implementation of the Range type and adaptation N)ProgrammingErrorInterfaceError) ISQLQuoteadaptregister_adapter)new_typenew_array_type register_type) string_typesc@seZdZdZd4Zd5dd Zd d Zd d ZeddZ eddZ eddZ eddZ eddZ eddZeddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3ZdS)6RangeaPython 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[)FcCsB|s,|dkrtd|||_||_||_nd|_|_|_dS)N[)(]()[]zbound flags not valid: %r)rrrr) ValueErrorr r r)selflowerupperboundsemptyr|jdkrdS|jdt|jdt|j|jdg}dj|S)Nrrz, )rstrr r join)ritemsrrr__str__Bs  z Range.__str__cCs|jS)z:The lower bound of the range. `!None` if empty or unbound.)r )rrrrrOsz Range.lowercCs|jS)z:The upper bound of the range. `!None` if empty or unbound.)r )rrrrrTsz Range.uppercCs |jdkS)z`!True` if the range is empty.N)r)rrrrisemptyYsz Range.isemptycCs|jdkrdS|jdkS)z0`!True` if the range doesn't have a lower bound.NF)rr )rrrr lower_inf^s zRange.lower_infcCs|jdkrdS|jdkS)z1`!True` if the range doesn't have an upper bound.NF)rr )rrrr upper_infes zRange.upper_infcCs&|jdks|jdkrdS|jddkS)z4`!True` if the lower bound is included in the range.NFr[)rr )rrrr lower_inclszRange.lower_inccCs&|jdks|jdkrdS|jddkS)z4`!True` if the upper bound is included in the range.NFr ])rr )rrrr upper_incsszRange.upper_inccCs~|jdkrdS|jdk rD|jddkr6||jkrDdSn||jkrDdS|jdk rz|jddkrl||jkrzdSn||jkrzdSdS)NFrr)r r+T)rr r )rxrrr __contains__zs       zRange.__contains__cCs |jdk S)N)r)rrrr__bool__szRange.__bool__cCst|j|S)N)typer/)rrrr __nonzero__szRange.__nonzero__cCs2t|tsdS|j|jko0|j|jko0|j|jkS)NF) isinstancer r r r)rotherrrr__eq__s    z Range.__eq__cCs |j| S)N)r4)rr3rrr__ne__sz Range.__ne__cCst|j|j|jfS)N)hashr r r)rrrr__hash__szRange.__hash__cCs^t|tstSxJdD]B}t||}t||}||kr6q|dkrBdS|dkrNdS||kSqWdS)Nr r rTF)r r r)r2r NotImplementedgetattr)rr3attrZ self_valueZ other_valuerrr__lt__s     z Range.__lt__cCs||kr dS|j|SdS)NT)r;)rr3rrr__le__sz Range.__le__cCst|tr|j|StSdS)N)r2r r;r8)rr3rrr__gt__s  z Range.__gt__cCs||kr dS|j|SdS)NT)r=)rr3rrr__ge__sz Range.__ge__csfddjDS)Ncs"i|]}t|rt||qSr)hasattrr9).0slot)rrr sz&Range.__getstate__..) __slots__)rr)rr __getstate__s zRange.__getstate__cCs&x |jD]\}}t|||q WdS)N)r$setattr)rstaterAvaluerrr __setstate__szRange.__setstate__)r r r)NNrF)r __module__ __qualname____doc__rCrrr%propertyrrr&r'r(r*r,r.r/r1r4r5r7r;r<r=r>rDrHrrrrr $s0         r FcCs&tj|||}|j| r|pd|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_rangesrSc@s4eZdZdZdZddZddZddZd 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)rrUrrrrszRangeAdapter.__init__cCs|jtkr|SdS)N)Z_protor)rprotorrr __conform__s zRangeAdapter.__conform__cCs ||_dS)N)_conn)rconnrrrprepareszRangeAdapter.preparecCs|jdkrtd|j}|jr.d|jjdS|jdk rbt|j}t|drX|j|j |j }nd}|j dk rt|j }t|dr|j|j |j }nd}|jjdd|d|d|j jdd S) NzMRangeAdapter must be subclassed overriding its name or the getquoted() methods 'empty'::utf8rZsNULL(s, s, 's')) nameNotImplementedErrorrUr&encoderrr?rZrX getquotedrr)rrarrrrrr`s&           zRangeAdapter.getquoted) rrIrJrKr]rrWrZr`rrrrrTs rTc@sZeZdZdZdddZddZeddZej d ej Z ej d Z dd d Z dd dZdS)rMa 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|||jjp"|jjj}t|f||j|_|dk rVt |f|d|j|_ nd|_ dS)NARRAY) subtype_oid_create_rangesadapterr]rrrparse typecasterrarray_typecaster)rrPrQoidrd array_oidr]rrrr)s zRangeCaster.__init__cCsd|_t|tr*t|tfi|_||j_n2yt|trD|tk rD||_Wntk rZYnX|jdkrntdd|_y6t|trt|t fi|_t|t r|t k r||_Wntk rYnX|jdkrtddS)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) rfr2r r0rTr] issubclass TypeErrorranger )rrPrQrrrre7s0        zRangeCaster._create_rangescCsddlm}ddlm}||\}}|jjdkr@td|jj|j}d|kr`|jdd\} } n|} d} y|j d | | fWn$tk r|j s|j Yn"X|j } ||kr|j r|j | std || \} } }t ||| | |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 found)rjrdrk)psycopg2.extensionsroZpsycopg2.extrasrpinfoZserver_versionrstatussplitexecuteZ autocommitrollbackZfetchonerM)rr]rQrRrorprYZcursZ conn_statusZschematnameZrecr0subtypearrayrrrrNZs8       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 (["\\])\1cCs|dkr dS|dkr |jddS|jj|}|dkr@td||jd}|dkrr|jd}|dk rr|jjd|}|jd}|dkr|jd }|dk r|jjd|}|dk r|j|j|}|j|j|}|jd |jd }|j|||S) NrT)rzfailed to parse range: '%s'z\1r ) rn _re_rangematchrgroup _re_undoublesubcastrd)rscurmrrrrrrrgs,       zRangeCaster.parsecCs4t|j||jdk r"t|j|t|j|jdS)N)r rhrirrnrf)rscoperrrrOs   zRangeCaster._register)N)N)N)rrIrJrKrre classmethodrNrecompileVERBOSErrrgrOrrrrrM"s # 8   rMc@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)rrIrJrKrrrrrsrc@seZdZdZdS) DateRangez#Represents :sql:`daterange` values.N)rrIrJrKrrrrrsrc@seZdZdZdS) DateTimeRangez!Represents :sql:`tsrange` values.N)rrIrJrKrrrrrsrc@seZdZdZdS)DateTimeTZRangez#Represents :sql:`tstzrange` values.N)rrIrJrKrrrrrsrc@seZdZdZddZdS)NumberRangeAdapterz1Adapt a range if the subtype doesn't need quotes.cCsr|j}|jrdS|js,t|jjjd}nd}|jsLt|jjjd}nd}d|j d|||j dfj dS)Ns'empty'asciir!z '%s%s,%s%s'rr ) rUr&r'rrr`decoder(rrr_)rrarrrrrr`szNumberRangeAdapter.getquotedN)rrIrJrKr`rrrrrsri@iA)rjrdrkiViWiBiiCZ daterangeiHi:iIZtsrangeiDiZiEZ tstzrangeiFiiG)F)rKrZpsycopg2._psycopgrrrsrrrrrr Zpsycopg2.compatr objectr rSrTrMrrrrrZint4range_casterrOZint8range_casterZnumrange_casterZdaterange_casterZtsrange_casterZtstzrange_casterrrrrsF 0 0(