Edit File: array.cpython-37.pyc
B ��4]Q+ � @ s� d dl mZ d dl mZ ddlmZ ddlmZ ddlmZ yddl m Z W n e k rd d Z Y nX ej fd d �Zej fd d �ZG dd� dej�Zejddd�Zejddd�Zejddd�ZG dd� dej�Zeeej< eed< d S )� )�colspecs)� ischema_names� )�types)� expression)� operators� )�UUIDNc C s |� | |�S )z�A synonym for the :meth:`.ARRAY.Comparator.any` method. This method is legacy and is here for backwards-compatibility. .. seealso:: :func:`.expression.any_` )�any)�other�arrexpr�operator� r �W/opt/alt/python37/lib64/python3.7/site-packages/sqlalchemy/dialects/postgresql/array.py�Any s r c C s |� | |�S )z�A synonym for the :meth:`.ARRAY.Comparator.all` method. This method is legacy and is here for backwards-compatibility. .. seealso:: :func:`.expression.all_` )�all)r r r r r r �All# s r c s8 e Zd ZdZd Z� fdd�Zd dd�Zd dd �Z� ZS ) �arraya� A PostgreSQL ARRAY literal. This is used to produce ARRAY literals in SQL expressions, e.g.:: from sqlalchemy.dialects.postgresql import array from sqlalchemy.dialects import postgresql from sqlalchemy import select, func stmt = select([ array([1,2]) + array([3,4,5]) ]) print(stmt.compile(dialect=postgresql.dialect())) Produces the SQL:: SELECT ARRAY[%(param_1)s, %(param_2)s] || ARRAY[%(param_3)s, %(param_4)s, %(param_5)s]) AS anon_1 An instance of :class:`.array` will always have the datatype :class:`.ARRAY`. The "inner" type of the array is inferred from the values present, unless the ``type_`` keyword argument is passed:: array(['foo', 'bar'], type_=CHAR) Multidimensional arrays are produced by nesting :class:`.array` constructs. The dimensionality of the final :class:`.ARRAY` type is calculated by recursively adding the dimensions of the inner :class:`.ARRAY` type:: stmt = select([ array([ array([1, 2]), array([3, 4]), array([column('q'), column('x')]) ]) ]) print(stmt.compile(dialect=postgresql.dialect())) Produces:: SELECT ARRAY[ARRAY[%(param_1)s, %(param_2)s], ARRAY[%(param_3)s, %(param_4)s], ARRAY[q, x]] AS anon_1 .. versionadded:: 1.3.6 added support for multidimensional array literals .. seealso:: :class:`.postgresql.ARRAY` c sZ t t| �j||� t| jt�rJt| jj| jjd k r>| jjd ndd�| _n t| j�| _d S )Nr � )� dimensions)�superr �__init__� isinstance�type�ARRAY� item_typer )�selfZclauses�kw)� __class__r r r f s zarray.__init__FNc sR |s� t jkr4t|t�s t�tjd |� ��jdd�S t� ��fdd�|D ��S d S )NT)Z_compared_to_operator�type_Z_compared_to_type�uniquec s g | ]}�j � |d �d��qS )T)�_assume_scalarr )� _bind_param)�.0�o)r r r r r � <listcomp>� s z%array._bind_param.<locals>.<listcomp>) r �getitemr �int�AssertionErrorr Z BindParameterr r )r r �objr! r r )r r r r r" r s zarray._bind_paramc C s& |t jt jt jfkr t�| �S | S d S )N)r Zany_opZall_opr& r ZGrouping)r Zagainstr r r � self_group� s zarray.self_group)FN)N) �__name__� __module__� __qualname__�__doc__Z__visit_name__r r" r* � __classcell__r r )r r r 1 s 1 r z@>� )� precedencez<@z&&c @ sj e Zd ZdZG dd� dejj�ZeZddd�Ze dd � �Z e d d � �Z d d � Z dd� Z dd� Zdd� ZdS )r a� PostgreSQL ARRAY type. .. versionchanged:: 1.1 The :class:`.postgresql.ARRAY` type is now a subclass of the core :class:`.types.ARRAY` type. The :class:`.postgresql.ARRAY` type is constructed in the same way as the core :class:`.types.ARRAY` type; a member type is required, and a number of dimensions is recommended if the type is to be used for more than one dimension:: from sqlalchemy.dialects import postgresql mytable = Table("mytable", metadata, Column("data", postgresql.ARRAY(Integer, dimensions=2)) ) The :class:`.postgresql.ARRAY` type provides all operations defined on the core :class:`.types.ARRAY` type, including support for "dimensions", indexed access, and simple matching such as :meth:`.types.ARRAY.Comparator.any` and :meth:`.types.ARRAY.Comparator.all`. :class:`.postgresql.ARRAY` class also provides PostgreSQL-specific methods for containment operations, including :meth:`.postgresql.ARRAY.Comparator.contains` :meth:`.postgresql.ARRAY.Comparator.contained_by`, and :meth:`.postgresql.ARRAY.Comparator.overlap`, e.g.:: mytable.c.data.contains([1, 2]) The :class:`.postgresql.ARRAY` type may not be supported on all PostgreSQL DBAPIs; it is currently known to work on psycopg2 only. Additionally, the :class:`.postgresql.ARRAY` type does not work directly in conjunction with the :class:`.ENUM` type. For a workaround, see the special type at :ref:`postgresql_array_of_enum`. .. seealso:: :class:`.types.ARRAY` - base array type :class:`.postgresql.array` - produces a literal array value. c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) zARRAY.Comparatora$ Define comparison operations for :class:`.ARRAY`. Note that these operations are in addition to those provided by the base :class:`.types.ARRAY.Comparator` class, including :meth:`.types.ARRAY.Comparator.any` and :meth:`.types.ARRAY.Comparator.all`. c K s | j t|tjd�S )zBoolean expression. Test if elements are a superset of the elements of the argument array expression. )� result_type)�operate�CONTAINS�sqltypes�Boolean)r r �kwargsr r r �contains� s zARRAY.Comparator.containsc C s | j t|tjd�S )z�Boolean expression. Test if elements are a proper subset of the elements of the argument array expression. )r2 )r3 � CONTAINED_BYr5 r6 )r r r r r � contained_by� s z ARRAY.Comparator.contained_byc C s | j t|tjd�S )zuBoolean expression. Test if array has elements in common with an argument array expression. )r2 )r3 �OVERLAPr5 r6 )r r r r r �overlap� s zARRAY.Comparator.overlapN)r+ r, r- r. r8 r: r< r r r r � Comparator� s r= FNc C s> t |t�rtd��t |t�r"|� }|| _|| _|| _|| _dS )aP Construct an ARRAY. E.g.:: Column('myarray', ARRAY(Integer)) Arguments are: :param item_type: The data type of items of this array. Note that dimensionality is irrelevant here, so multi-dimensional arrays like ``INTEGER[][]``, are constructed as ``ARRAY(Integer)``, not as ``ARRAY(ARRAY(Integer))`` or such. :param as_tuple=False: Specify whether return results should be converted to tuples from lists. DBAPIs such as psycopg2 return lists by default. When tuples are returned, the results are hashable. :param dimensions: if non-None, the ARRAY will assume a fixed number of dimensions. This will cause the DDL emitted for this ARRAY to include the exact number of bracket clauses ``[]``, and will also optimize the performance of the type overall. Note that PG arrays are always implicitly "non-dimensioned", meaning they can store any number of dimensions no matter how they were declared. :param zero_indexes=False: when True, index values will be converted between Python zero-based and PostgreSQL one-based indexes, e.g. a value of one will be added to all index values before passing to the database. .. versionadded:: 0.9.5 zUDo not nest ARRAY types; ARRAY(basetype) handles multi-dimensional arrays of basetypeN)r r � ValueErrorr r �as_tupler � zero_indexes)r r r? r r@ r r r r � s & zARRAY.__init__c C s | j S )N)r? )r r r r �hashable s zARRAY.hashablec C s t S )N)�list)r r r r � python_type s zARRAY.python_typec C s ||kS )Nr )r �x�yr r r �compare_values s zARRAY.compare_valuesc sz �d krt |�}�dks6�d krZ|r6t|d t tf�sZ�rP� �fdd�|D ��S � |�S n � � ���fdd�|D ��S d S )Nr r c 3 s | ]}� |�V qd S )Nr )r# rD )�itemprocr r � <genexpr>2 s z$ARRAY._proc_array.<locals>.<genexpr>c 3 s. | ]&}�� |��d k r �d nd � �V qd S )Nr )� _proc_array)r# rD )� collection�dimrG r r r rH 7 s )rB r �tuple)r ZarrrG rK rJ r )rJ rK rG r r rI # s zARRAY._proc_arrayc s$ �j �|��|�� � �fdd�}|S )Nc s" | d kr | S �� | � �jt�S d S )N)rI r rB )�value)� item_procr r r �processE s z%ARRAY.bind_processor.<locals>.process)r � dialect_impl�bind_processor)r �dialectrO r )rN r r rQ @ s zARRAY.bind_processorc s& �j �|��||�� � �fdd�}|S )Nc s, | d kr | S �� | � �j�jr"tnt�S d S )N)rI r r? rL rB )rM )rN r r r rO T s z'ARRAY.result_processor.<locals>.process)r rP �result_processor)r rR ZcoltyperO r )rN r r rS O s zARRAY.result_processor)FNF)r+ r, r- r. r5 r r= Zcomparator_factoryr �propertyrA rC rF rI rQ rS r r r r r � s + 1 r Z_array)�baser r � r r5 Zsqlr r �uuidr Z _python_UUID� ImportError�eqr r �Tupler Z custom_opr4 r9 r; r r r r r �<module> s$ ` K