ó º9—^c @s?dZddlZddlmZmZddlmZmZmZddlm Z m Z m Z ddl m Z defd„ƒYZed „Zd efd „ƒYZd efd „ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZeeeƒeeeddddddƒZejƒeeedddddd ƒZejƒeeedd!dd"dd#ƒZejƒed$edd%dd&dd'ƒZejƒed(edd)dd*dd+ƒZejƒed,edd-dd.dd/ƒZejƒdS(0s1Implementation of the Range type and adaptation iÿÿÿÿN(tProgrammingErrortInterfaceError(t ISQLQuotetadapttregister_adapter(tnew_typetnew_array_typet register_type(t string_typestRangecBseZdZdZddded„Zd„Zd„Ze d„ƒZ e d „ƒZ e d „ƒZ e d „ƒZ e d „ƒZe d „ƒZe d„ƒZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„ZRS(s…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 t_lowert_uppert_boundss[)cCs^|sC|dkr%td|ƒ‚n||_||_||_nd|_|_|_dS(Ns[)s(]s()s[]sbound flags not valid: %r(s[)s(]s()s[](t ValueErrorR R R tNone(tselftlowertuppertboundstempty((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt__init__/s    cCsD|jdkrd|jjSd|jj|j|j|jfSdS(Ns%s(empty=True)s%s(%r, %r, %r)(R Rt __class__t__name__R R (R((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt__repr__:s cCsU|jdkrdS|jdt|jƒdt|jƒ|jdg}dj|ƒS(NRis, it(R RtstrR R tjoin(Rtitems((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt__str__As   cCs|jS(s:The lower bound of the range. `!None` if empty or unbound.(R (R((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyRNscCs|jS(s:The upper bound of the range. `!None` if empty or unbound.(R (R((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyRSscCs |jdkS(s`!True` if the range is empty.N(R R(R((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pytisemptyXscCs |jdkrtS|jdkS(s0`!True` if the range doesn't have a lower bound.N(R RtFalseR (R((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt lower_inf]scCs |jdkrtS|jdkS(s1`!True` if the range doesn't have an upper bound.N(R RRR (R((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt upper_infdscCs3|jdks|jdkr"tS|jddkS(s4`!True` if the lower bound is included in the range.it[N(R RR R(R((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt lower_inckscCs3|jdks|jdkr"tS|jddkS(s4`!True` if the upper bound is included in the range.it]N(R RR R(R((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt upper_incrscCs³|jdkrtS|jdk ra|jddkrK||jkr^tSqa||jkratSn|jdk r¯|jddkr™||jkr¬tSq¯||jkr¯tSntS(NiR!iR#(R RRR R tTrue(Rtx((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt __contains__yscCs |jdk S(N(R R(R((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt__bool__scCst|ƒj|ƒS(N(ttypeR((R((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt __nonzero__’scCsGt|tƒstS|j|jkoF|j|jkoF|j|jkS(N(t isinstanceR RR R R (Rtother((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt__eq__–s cCs|j|ƒ S(N(R-(RR,((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt__ne__scCst|j|j|jfƒS(N(thashR R R (R((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt__hash__ scCst|tƒstSxedD]]}t||ƒ}t||ƒ}||krMq|dkr]tS|dkrmtS||kSqWtS(NR R R (R R R (R+R tNotImplementedtgetattrRR%R(RR,tattrt self_valuet other_value((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt__lt__§s    cCs!||krtS|j|ƒSdS(N(R%R6(RR,((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt__le__·s cCs$t|tƒr|j|ƒStSdS(N(R+R R6R1(RR,((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt__gt__½s cCs!||krtS|j|ƒSdS(N(R%R8(RR,((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt__ge__Ãs cs‡fd†ˆjDƒS(Ncs1i|]'}tˆ|ƒrtˆ|ƒ|“qS((thasattrR2(t.0tslot(R(s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pys Ês (t __slots__(R((Rs1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt __getstate__És cCs1x*|jƒD]\}}t|||ƒq WdS(N(Rtsetattr(RtstateR<tvalue((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt __setstate__Ís(R R R N(Rt __module__t__doc__R=RRRRRtpropertyRRRRR R"R$R'R(R*R-R.R0R6R7R8R9R>RB(((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyR #s0              cCs3tj|||ƒ}|j| r(|p+dƒ|S(sé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(t RangeCastert_from_dbt _registerR(tpgrangetpyranget conn_or_curstgloballytcaster((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pytregister_rangeÒst RangeAdaptercBs8eZdZdZd„Zd„Zd„Zd„ZRS(s¢`ISQLQuote` adapter for `Range` subclasses. This is an abstract class: concrete classes must set a `name` class attribute or override `getquoted()`. cCs ||_dS(N(tadapted(RRP((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyRùscCs|jtkr|SdS(N(t_protoR(Rtproto((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyt __conform__üscCs ||_dS(N(t_conn(Rtconn((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pytpreparescCs&|jdkrtdƒ‚n|j}|jrDd|jjdƒS|jdk r“t|jƒ}t|dƒr„|j |j ƒn|j ƒ}nd}|j dk rèt|j ƒ}t|dƒrÙ|j |j ƒn|j ƒ}nd}|jjdƒd|d|d|j jdƒd S( NsMRangeAdapter must be subclassed overriding its name or the getquoted() methods 'empty'::tutf8RVtNULLt(s, s, 's')(tnameRtNotImplementedErrorRPRtencodeRRR:RVRTt getquotedRR (RtrtaRR((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyR]s&   N( RRCRDRRZRRSRVR](((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyROñs    RFcBsneZdZdd„Zd„Zed„ƒZej dej ƒZ ej dƒZ dd„Z dd„ZRS( s 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. cCsŠ||_|j||ƒ|jjp1|jjj}t|f||jƒ|_|dk r}t |f|d|jƒ|_ n d|_ dS(NtARRAY( t subtype_oidt_create_rangestadapterRZRRRtparset typecasterRRtarray_typecaster(RRIRJtoidRat array_oidRZ((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyR(s  cCs0d|_t|tƒr?t|tfiƒ|_||j_n?y+t|tƒri|tk ri||_nWntk r}nX|jdkrœtdƒ‚nd|_ yUt|tƒrÒt|t fiƒ|_ nt|t ƒrù|t k rù||_ nWntk r nX|j dkr,tdƒ‚ndS(s0Create Range and RangeAdapter classes if needed.s:pgrange must be a string or a RangeAdapter strict subclasss1pyrange must be a type or a Range strict subclassN( RRcR+RR)RORZt issubclasst TypeErrortrangeR (RRIRJ((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyRb6s0      c CsVddlm}ddlm}||ƒ\}}|jjdkr]td|jjƒ‚n|j}d|kr|jddƒ\} } n |} d} y|j d | | fƒWn*tk rß|j sÙ|j ƒn‚n0X|j ƒ} ||kr|j r|j ƒn| s(td |ƒ‚n| \} } }t ||d | d | d |ƒS(s|Return a `RangeCaster` instance for the type *pgrange*. Raise `ProgrammingError` if the type is not found. iÿÿÿÿ(tSTATUS_IN_TRANSACTION(t_solve_conn_cursiX`s'range types not available in version %st.itpublicsÞselect 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; sPostgreSQL type '%s' not foundRgRaRh(tpsycopg2.extensionsRltpsycopg2.extrasRmtinfotserver_versionRtstatustsplittexecutet autocommittrollbacktfetchoneRF(RRZRJRKRlRmRUtcurst conn_statustschemattnametrecR)tsubtypetarray((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyRGYs8          s] ( \(|\[ ) # 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 s (["\\])\1cCsg|dkrdS|dkr,|jdtƒS|jj|ƒ}|dkr]td|ƒ‚n|jdƒ}|dkr®|jdƒ}|dk r®|jjd|ƒ}q®n|jdƒ}|dkrÿ|jdƒ}|dk rÿ|jjd|ƒ}qÿn|dk r8|j |j |ƒ}|j |j |ƒ}n|jdƒ|jd ƒ}|j|||ƒS( NRsfailed to parse range: '%s'iis\1iiii( RRkR%t _re_rangetmatchRtgroupt _re_undoubletsubtcastRa(RtstcurtmRRR((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyRd¡s,        cCsIt|j|ƒ|jdk r2t|j|ƒnt|j|jƒdS(N(RReRfRRRkRc(Rtscope((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyRHÀsN(RRCRDRRRbt classmethodRGtretcompiletVERBOSERR„RdRH(((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyRF!s  #8  t NumericRangecBseZdZRS(sÄA `Range` suitable to pass Python numeric types to a PostgreSQL range. PostgreSQL types :sql:`int4range`, :sql:`int8range`, :sql:`numrange` are casted into `!NumericRange` instances. (RRCRD(((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyRÈst DateRangecBseZdZRS(s#Represents :sql:`daterange` values.(RRCRD(((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyRÑst DateTimeRangecBseZdZRS(s!Represents :sql:`tsrange` values.(RRCRD(((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyR‘ÖstDateTimeTZRangecBseZdZRS(s#Represents :sql:`tstzrange` values.(RRCRD(((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyR’ÛstNumberRangeAdaptercBseZdZd„ZRS(s1Adapt a range if the subtype doesn't need quotes.cCs¡|j}|jrdS|js@t|jƒjƒjdƒ}nd}|jspt|jƒjƒjdƒ}nd}d|j d|||j dfj dƒS(Ns'empty'tasciiRs '%s%s,%s%s'ii( RPRRRRR]tdecodeR RR R\(RR^RR((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyR]çs   ! !(RRCRDR](((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pyR“åsRgi@RaiRhiAiViiWiBi¤iCt daterangeiHi:iIttsrangeiDiZiEt tstzrangeiFi iG( RDRŒtpsycopg2._psycopgRRRpRRRRRRtpsycopg2.compatRtobjectR RRNRORFRRR‘R’R“tint4range_casterRHtint8range_castertnumrange_castertdaterange_casterttsrange_casterttstzrange_caster(((s1/tmp/pip-unpacked-wheel-WyN9Ij/psycopg2/_range.pytsB ¯ 0§