|
Server : Apache System : Linux iad1-shared-b8-43 6.6.49-grsec-jammy+ #10 SMP Thu Sep 12 23:23:08 UTC 2024 x86_64 User : dh_edsupp ( 6597262) PHP Version : 8.2.26 Disable Function : NONE Directory : /lib/python3/dist-packages/psycopg2/__pycache__/ |
Upload File : |
o
��a�D � @ sr d Z ddlZddlmZmZ ddlmZmZmZ ddlm Z m
Z
m Z G dd� d�Z d/d d
�Z
G d d � d �ZG d
d� d�ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de �ZG dd� de�Zeee� eeedddd �Ze�� eeed d d d �Ze�� eeed d!d"d �Ze�� ed#ed$d%d&d �Ze�� ed'ed(d)d*d �Ze�� ed+ed,d-d.d �Ze�� dS )0z1Implementation of the Range type and adaptation
� N)�ProgrammingError�InterfaceError)� ISQLQuote�adapt�register_adapter)�new_type�new_array_type�
register_typec @ 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 sF |s|dvr
t d|����|| _|| _|| _d S d | _ | _| _d S )N)r z(]z()z[]zbound flags not valid: )�
ValueErrorr r
r )�self�lower�upper�bounds�empty� r �1/usr/lib/python3/dist-packages/psycopg2/_range.py�__init__/ s
zRange.__init__c C s2 | j d u r | jj� d�S d�| jj| j| j| j �S )Nz (empty=True)z{}({!r}, {!r}, {!r}))r � __class__�__name__�formatr r
�r r r r �__repr__: s
�zRange.__repr__c C s> | j d u rdS | 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 � | j S )z:The lower bound of the range. `!None` if empty or unbound.)r r r r r r N � z Range.lowerc C r$ )z:The upper bound of the range. `!None` if empty or unbound.)r
r r r r r S r% z Range.upperc C s
| j du S )z `!True` if the range is empty.N�r r r r r �isemptyX s
z
Range.isemptyc C � | j du rdS | jdu S )z0`!True` if the range doesn't have a lower bound.NF�r r r r r r � lower_inf] �
zRange.lower_infc C r( )z1`!True` if the range doesn't have an upper bound.NF�r r
r r r r � upper_infd r+ zRange.upper_infc C �& | j du s
| jdu r dS | j d dkS )z4`!True` if the lower bound is included in the range.NFr �[r) r r r r � lower_inck � zRange.lower_incc C r. )z4`!True` if the upper bound is included in the range.NFr �]r, r r r r � upper_incr r1 zRange.upper_incc C s� | j d u rdS | jd ur"| j d dkr|| jk rdS n|| jkr"dS | jd ur>| j d dkr7|| jkr5dS dS || jkr>dS dS )NFr r/ r r2 T)r r r
)r �xr r r � __contains__y s"
�
�zRange.__contains__c C s
| j d uS �Nr& r r r r �__bool__� �
zRange.__bool__c C s t | ��| �S r6 )�typer7 r r r r � __nonzero__� s zRange.__nonzero__c C s2 t |t�sdS | j|jko| j|jko| j|jkS )NF)�
isinstancer
r r
r �r �otherr r r �__eq__� s
�
�z Range.__eq__c C s | � |� S r6 )r> r< r r r �__ne__� s z Range.__ne__c C s t | j| j| jf�S r6 )�hashr r
r r r r r �__hash__� s zRange.__hash__c C s` t |t�stS dD ]$}t| |�}t||�}||krq |d u r! dS |d u r( dS ||k S dS )Nr TF)r; r
�NotImplemented�getattr)r r= �attr�
self_value� other_valuer r r �__lt__� s
z Range.__lt__c C � | |krdS | � |�S �NT)rG r< r r r �__le__� �
z Range.__le__c C s t |t�r
|�| �S tS r6 )r; r
rG rB r< r r r �__gt__� s
z Range.__gt__c C rH rI )rL r< r r r �__ge__� rK z Range.__ge__c s � fdd�� j D �S )Nc s" i | ]
}t � |�r|t� |��qS r )�hasattrrC )�.0�slotr r r �
<dictcomp>� s
�z&Range.__getstate__.<locals>.<dictcomp>)� __slots__r r r r � __getstate__� s
�zRange.__getstate__c C s" |� � D ]
\}}t| ||� qd S r6 )r" �setattr)r �staterP �valuer r r � __setstate__� s �zRange.__setstate__)NNr F) r �
__module__� __qualname__�__doc__rR r r r# �propertyr r r' r* r- r0 r3 r5 r7 r: r> r? rA rG rJ rL rM rS rW r r r r r
# s@
r
Fc C s& t �| ||�}|�| r|pd� |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_curs�globally�casterr r r �register_range� s rd 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 �
|| _ d S r6 )�adapted)r rg r r r r � r8 zRangeAdapter.__init__c C s | j tu r| S d S r6 )�_protor )r �protor r r � __conform__� s
�zRangeAdapter.__conform__c C rf r6 )�_conn)r �connr r r �prepare r8 zRangeAdapter.preparec C s� | j d u r td��| j}|jrd| j �d� S |jd ur1t|j�}t|d�r,|�| j � |�
� }nd}|j d urMt|j �}t|d�rH|�| j � |�
� }nd}| j �d�d | d | d |j �d� d S )
NzMRangeAdapter must be subclassed overriding its name or the getquoted() methods 'empty'::�utf8rm s NULL� (s , s , 's '))
�name�NotImplementedErrorrg r' �encoder r rN rm rk � getquotedr r )r �r�ar r r r r rs s4
�
�
��zRangeAdapter.getquoted) r rX rY rZ rp r rj rm rs r r r r re � s re c @ sZ e Zd ZdZddd�Zdd� Zedd� �Ze� d ej
�Z e� d
�Z dd d �Z
dd
d�ZdS )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.
Nc C sb || _ | �||� | jjp| jjj}t|f|| j�| _|d ur,t |f|d | j�| _
d S d | _
d S )N�ARRAY) � subtype_oid�_create_ranges�adapterrp r r r �parse�
typecasterr �array_typecaster)r r_ r` �oidrw � array_oidrp r r r r ( s
�
zRangeCaster.__init__c C s� d| _ t|t�rt|tfi �| _ || j _nzt|t�r"|tur"|| _ W n ty, Y nw | j du r6td��d| _zt|t�rGt|t fi �| _t|t �rS|t urS|| _W n ty] Y nw | jdu rgtd��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)
ry r; r r9 re rp �
issubclass� TypeError�ranger
)r r_ r` r r r rx 6 s>
� �
�
� �
��zRangeCaster._create_rangesc C s� ddl m} ddlm} ||�\}}|jjdk r td|jj ��|j}d|v r0|�dd�\} }
n|}
d} z
|� d |
| f� W n tyM |j
sL|� � � w |� � } ||kr]|j
s]|� � | sgtd
|� d ���| \} }
}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;
zPostgreSQL type 'z ' not found�r} rw r~ )�psycopg2.extensionsr� �psycopg2.extrasr� �info�server_versionr �status�split�execute�
autocommit�rollback�fetchoner\ )r rp r` ra r� r� rl �curs� conn_status�schema�tname�recr9 �subtype�arrayr r r r] Y sD �� ��
�
�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 u rd S |dkr| j dd�S | j�|�}|d u r"td|� d���|�d�}|d u r;|�d�}|d ur;| j�d|�}|�d �}|d u rT|�d
�}|d urT| j�d|�}|d urf|�| j|�}|�| j|�}|�d �|�d � }| � |||�S )
Nr T)r zfailed to parse range: '�'� � z\1� � r � ) r� � _re_range�matchr �group� _re_undouble�sub�castrw )r �s�cur�mr r r r r r rz � s,
zRangeCaster.parsec C s4 t | j|� | jd urt | j|� t| j| j� d S r6 )r r{ r| r r� ry )r �scoper r r r^ � s
zRangeCaster._registerr6 )r rX rY rZ r rx � classmethodr] �re�compile�VERBOSEr� r� rz r^ r r r r r\ ! s
#
7 �
r\ c @ � 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 rX rY rZ r r r r r� � s r� c @ r� )� DateRangez#Represents :sql:`daterange` values.Nr� r r r r r� � � r� c @ r� )�
DateTimeRangez!Represents :sql:`tsrange` values.Nr� r r r r r� � r� r� c @ r� )�DateTimeTZRangez#Represents :sql:`tstzrange` values.Nr� r r r r r� � r� r� c @ s e Zd ZdZdd� ZdS )�NumberRangeAdapterz1Adapt a range if the subtype doesn't need quotes.c C s| | j }|jrdS |jst|j��� �d�}nd}|js&t|j��� �d�}nd}d|j d � |� d|� |j d � d��
d�S )Ns 'empty'�asciir r� r �,r ) rg r' r* r r rs �decoder- r r rr )r rt r r r r r rs � s ,z NumberRangeAdapter.getquotedN)r rX rY rZ rs r r r r r� � s r� i@ � iA r� iV � iW iB i� iC � daterangeiH i: iI �tsrangeiD iZ iE � tstzrangeiF i� iG )F) rZ r� �psycopg2._psycopgr r r� r r r r r r r
rd re r\ r� r� r� r� r� �int4range_casterr^ �int8range_caster�numrange_caster�daterange_caster�tsrange_caster�tstzrange_casterr r r r �<module> sR
0 0 (
������