File: //opt/alt/python37/lib64/python3.7/site-packages/psycopg2/__pycache__/_range.cpython-37.pyc
B
*��]E � @ s� d Z ddlZddlmZmZ ddlmZmZmZ ddlm Z m
Z
m Z ddl m
Z
G dd� de�Zd0d
d �ZG d d
� d
e�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Zeee� eeeddd d �Ze�� eeed d d d �Ze�� eeed!d"d#d �Ze�� ed$ed%d&d'd �Z e �� ed(ed)d*d+d �Z e �� ed,ed-d.d/d �Z e �� dS )1z1Implementation of the Range type and adaptation
� N)�ProgrammingError�InterfaceError)� ISQLQuote�adapt�register_adapter)�new_type�new_array_type�
register_type)� string_typesc @ s� e Zd ZdZdZd2dd�Zdd � Zd
d � Zed d
� �Z edd� �Z
edd� �Z edd� �Z edd� �Z
edd� �Zedd� �Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� ZdS )3�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 C sB |s,|dkrt d| ��|| _|| _|| _nd | _ | _| _d S )N)z[)z(]z()z[]zbound flags not valid: %r)�
ValueErrorr r
r )�self�lower�upper�bounds�empty� r �B/opt/alt/python37/lib64/python3.7/site-packages/psycopg2/_range.py�__init__/ s zRange.__init__c C s4 | j d krd| jj S d| jj| j| j| j f S d S )Nz%s(empty=True)z%s(%r, %r, %r))r � __class__�__name__r r
)r r r r �__repr__: s
zRange.__repr__c C s> | j d krdS | j d t| j�dt| j�| j d g}d�|�S )Nr r z, � � )r �strr r
�join)r �itemsr r r �__str__A s
z
Range.__str__c C s | j S )z:The lower bound of the range. `!None` if empty or unbound.)r )r r r r r N s z Range.lowerc C s | j S )z:The upper bound of the range. `!None` if empty or unbound.)r
)r r r r r S s z Range.upperc C s
| j dkS )z `!True` if the range is empty.N)r )r r r r �isemptyX s z
Range.isemptyc C s | j dkrdS | jdkS )z0`!True` if the range doesn't have a lower bound.NF)r r )r r r r � lower_inf] s
zRange.lower_infc C s | j dkrdS | jdkS )z1`!True` if the range doesn't have an upper bound.NF)r r
)r r r r � upper_infd s
zRange.upper_infc C s& | j dks| jdkrdS | j d dkS )z4`!True` if the lower bound is included in the range.NFr �[)r r )r r r r � lower_inck s zRange.lower_incc C s& | j dks| jdkrdS | j d dkS )z4`!True` if the upper bound is included in the range.NFr �])r r
)r r r r � upper_incr s zRange.upper_incc C s~ | j d krdS | jd k rD| j d dkr6|| jk rDdS n|| jkrDdS | jd k rz| j d dkrl|| jkrzdS n|| jkrzdS dS )NFr r% r r' T)r r r
)r �xr r r � __contains__y s
zRange.__contains__c C s
| j d k S )N)r )r r r r �__bool__� s zRange.__bool__c C s t | ��| �S )N)�typer+ )r r r r � __nonzero__� s zRange.__nonzero__c C s2 t |t�sdS | j|jko0| j|jko0| j|jkS )NF)�
isinstancer r r
r )r �otherr r r �__eq__� s
z Range.__eq__c C s | � |� S )N)r0 )r r/ r r r �__ne__� s z Range.__ne__c C s t | j| j| jf�S )N)�hashr r
r )r r r r �__hash__� s zRange.__hash__c C s^ t |t�stS xJdD ]B}t| |�}t||�}||kr6q|d krBdS |d krNdS ||k S qW dS )N)r r
r TF)r. r �NotImplemented�getattr)r r/ �attrZ
self_valueZ other_valuer r r �__lt__� s
z Range.__lt__c C s | |kr dS | � |�S d S )NT)r7 )r r/ r r r �__le__� s z Range.__le__c C s t |t�r|�| �S tS d S )N)r. r r7 r4 )r r/ r r r �__gt__� s
z Range.__gt__c C s | |kr dS | � |�S d S )NT)r9 )r r/ r r r �__ge__� s z Range.__ge__c s � fdd�� j D �S )Nc s" i | ]}t � |�rt� |�|�qS r )�hasattrr5 )�.0�slot)r r r �
<dictcomp>� s z&Range.__getstate__.<locals>.<dictcomp>)� __slots__)r r )r r � __getstate__� s
zRange.__getstate__c C s& x |� � D ]\}}t| ||� q
W d S )N)r �setattr)r �stater= �valuer r r � __setstate__� s zRange.__setstate__)NNr F) r �
__module__� __qualname__�__doc__r? r r r! �propertyr r r" r# r$ r& r( r* r+ r- r0 r1 r3 r7 r8 r9 r: r@ rD r r r r r # s0
r Fc C s& t �| ||�}|�| r |p d� |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)�pgrange�pyrange� conn_or_cursZgloballyZcasterr r r �register_range� s rO c @ s4 e Zd ZdZdZdd� Zdd� Zdd� Zd d
� ZdS ) � RangeAdapterz�`ISQLQuote` adapter for `Range` subclasses.
This is an abstract class: concrete classes must set a `name` class
attribute or override `getquoted()`.
Nc C s
|| _ d S )N)�adapted)r rQ r r r r � s zRangeAdapter.__init__c C s | j tkr| S d S )N)Z_protor )r �protor r r � __conform__� s
zRangeAdapter.__conform__c C s
|| _ d S )N)�_conn)r �connr r r �prepare s zRangeAdapter.preparec C s� | j d krtd��| j}|jr.d| j �d� S |jd k rbt|j�}t|d�rX|�| j � |�
� }nd}|j d k r�t|j �}t|d�r�|�| j � |�
� }nd}| j �d�d | d | d |j �d� d S )
NzMRangeAdapter must be subclassed overriding its name or the getquoted() methods 'empty'::�utf8rV s NULL� (s , s , 's '))
�name�NotImplementedErrorrQ r" �encoder r r; rV rT � getquotedr r )r �r�ar r r r r r\ s&
zRangeAdapter.getquoted) r rE rF rG rY r rS rV r\ r r r r rP � s rP c @ sZ e Zd ZdZddd�Zdd� Zedd� �Ze� d ej
�Z e� d
�Z dd d �Z
dd
d�ZdS )rI 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 C s` || _ | �||� | jjp"| jjj}t|f|| j�| _|d k rVt |f|d | j�| _
nd | _
d S )N�ARRAY) � subtype_oid�_create_ranges�adapterrY r r r �parse�
typecasterr �array_typecaster)r rL rM �oidr` � array_oidrY r r r r ( s zRangeCaster.__init__c C s� d| _ t|t�r*t|tfi �| _ || j _n2y t|t�rD|tk rD|| _ W n tk
rZ Y nX | j dkrntd��d| _y6t|t�r�t|t fi �| _t|t �r�|t k r�|| _W n tk
r� Y nX | jdkr�td��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)
rb r. r
r, rP rY �
issubclass� TypeError�ranger )r rL rM r r r ra 6 s0
zRangeCaster._create_rangesc C s� ddl m} ddlm} ||�\}}|jjdk r@td|jj ��|j}d|kr`|�dd�\} }
n|}
d} y|� d |
| f� W n$ tk
r� |j
s�|� � � Y n X |� � } ||kr�|j
s�|� � | s�td
| ��| \} }
}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 �publicz�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;
z PostgreSQL type '%s' not found)rf r` rg )�psycopg2.extensionsrk Zpsycopg2.extrasrl �infoZserver_versionr �status�split�executeZ
autocommitZrollbackZfetchonerI )r rY rM rN rk rl rU ZcursZ conn_statusZschemaZtnameZrecr, �subtype�arrayr r r rJ Y s8
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 (["\\])\1c C s� |d kr d S |dkr | j dd�S | j�|�}|d kr@td| ��|�d�}|d krr|�d�}|d k rr| j�d|�}|�d�}|d kr�|�d �}|d k r�| j�d|�}|d k r�|�| j|�}|�| j|�}|�d
�|�d � }| � |||�S ) Nr T)r zfailed to parse range: '%s'� � z\1� � r � ) rj � _re_range�matchr �group� _re_undouble�sub�castr` )r �s�cur�mr r r r r r rc � s,
zRangeCaster.parsec C s4 t | j|� | jd k r"t | j|� t| j| j� d S )N)r rd re r rj rb )r Zscoper r r rK � s
zRangeCaster._register)N)N)N)r rE rF rG r ra � classmethodrJ �re�compile�VERBOSEr{ r~ rc rK r r r r rI ! s
# 8
rI c @ s e Zd ZdZdS )� NumericRangez�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.
N)r rE rF rG r r r r r� � s r� c @ s e Zd ZdZdS )� DateRangez#Represents :sql:`daterange` values.N)r rE rF rG r r r r r� � s r� c @ s e Zd ZdZdS )�
DateTimeRangez!Represents :sql:`tsrange` values.N)r rE rF rG r r r r r� � s r� c @ s e Zd ZdZdS )�DateTimeTZRangez#Represents :sql:`tstzrange` values.N)r rE rF rG r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )�NumberRangeAdapterz1Adapt a range if the subtype doesn't need quotes.c C sr | j }|jrdS |js,t|j��� �d�}nd}|jsLt|j��� �d�}nd}d|j d |||j d f �
d�S )Ns 'empty'�asciir z '%s%s,%s%s'r r ) rQ r" r# r r r\ �decoder$ r r r[ )r r] r r r r r r\ � s z NumberRangeAdapter.getquotedN)r rE rF rG r\ r r r r r� � s r� i@ � iA )rf r` rg iV � iW iB i� iC Z daterangeiH i: iI ZtsrangeiD iZ iE Z tstzrangeiF i� iG )F) rG r� Zpsycopg2._psycopgr r ro r r r r r r Zpsycopg2.compatr
�objectr rO rP rI r� r� r� r� r� Zint4range_casterrK Zint8range_casterZnumrange_casterZdaterange_casterZtsrange_casterZtstzrange_casterr r r r �<module> sF 0
0 (