Edit File: _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 (