JezK
Edit File: array.pyc
� 1��fc @ s d d l Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z d d l m Z e j d � Z e j d � Z d e j e j f d � � YZ e j d d d d e �Z e j d d d d e �Z e j d d d d e �Z d e j f d � � YZ d � Z d S( i����Ni ( t types( t util( t coercions( t expression( t operators( t rolesc C s | j | | � S( sj A synonym for the ARRAY-level :meth:`.ARRAY.Comparator.any` method. See that method for details. ( t any( t othert arrexprt operator( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt Any s c C s | j | | � S( sj A synonym for the ARRAY-level :meth:`.ARRAY.Comparator.all` method. See that method for details. ( t all( R R R ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt All s t arrayc B sS e Z d Z d Z d Z e Z d � Z e d � � Z e d d � Z d d � Z RS( s� 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:`_types.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:`_types.ARRAY` type is calculated by recursively adding the dimensions of the inner :class:`_types.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` R t postgresqlc K s� g | D] } t j t j | � ^ q } t t | � j | | � g | D] } | j ^ qH | _ | j d | j r | j d n t j � } t | t � r� t | j d | j d k r� | j d n d �| _ n t | � | _ d S( Nt type_i t dimensionsi i ( R t expectR t ExpressionElementRolet superR t __init__t typet _type_tuplet popt sqltypest NULLTYPEt isinstancet ARRAYt item_typeR t None( t selft clausest kwt ct argt main_type( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyR \ s ( " c C s | f S( N( ( R ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt _select_iterables s c C sy | s | t j k r@ t j d | d | d | d | j d t �St g | D]$ } | j | | d t d | �^ qJ � Sd S( Nt _compared_to_operatorR t _compared_to_typet uniquet _assume_scalar( R t getitemR t BindParameterR R t TrueR t _bind_param( R R t objR( R t o( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyR, w s c C s3 | t j t j t j f k r+ t j | � S| Sd S( N( R t any_opt all_opR) R t Grouping( R t against( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt self_group� s N( t __name__t __module__t __doc__t __visit_name__t stringify_dialectR+ t inherit_cacheR t propertyR$ t FalseR R, R3 ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyR $ s 1 s @>t precedencei t is_comparisons <@s &&R c B s� e Z d Z d e j j f d � � YZ e Z e d e d � Z e d � � Z e d � � Z d � Z d � Z e j d � � Z d � Z d � Z d � Z RS( s� 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`. .. container:: topic **Detecting Changes in ARRAY columns when using the ORM** The :class:`_postgresql.ARRAY` type, when used with the SQLAlchemy ORM, does not detect in-place mutations to the array. In order to detect these, the :mod:`sqlalchemy.ext.mutable` extension must be used, using the :class:`.MutableList` class:: from sqlalchemy.dialects.postgresql import ARRAY from sqlalchemy.ext.mutable import MutableList class SomeOrmClass(Base): # ... data = Column(MutableList.as_mutable(ARRAY(Integer))) This extension will allow "in-place" changes such to the array such as ``.append()`` to produce events which will be detected by the unit of work. Note that changes to elements **inside** the array, including subarrays that are mutated in place, are **not** detected. Alternatively, assigning a new array value to an ORM element that replaces the old one will always trigger a change event. .. seealso:: :class:`_types.ARRAY` - base array type :class:`_postgresql.array` - produces a literal array value. t Comparatorc B s) e Z d Z d � Z d � Z d � Z RS( s* Define comparison operations for :class:`_types.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 | d t j �S( s� Boolean expression. Test if elements are a superset of the elements of the argument array expression. kwargs may be ignored by this operator but are required for API conformance. t result_type( t operatet CONTAINSR t Boolean( R R t kwargs( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt contains� s c C s | j t | d t j �S( s� Boolean expression. Test if elements are a proper subset of the elements of the argument array expression. R? ( R@ t CONTAINED_BYR RB ( R R ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt contained_by� s c C s | j t | d t j �S( su Boolean expression. Test if array has elements in common with an argument array expression. R? ( R@ t OVERLAPR RB ( R R ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt overlap� s ( R4 R5 R6 RD RF RH ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyR> � s c C sa t | t � r t d � � n t | t � r9 | � } n | | _ | | _ | | _ | | _ d S( sP 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 sU Do not nest ARRAY types; ARRAY(basetype) handles multi-dimensional arrays of basetypeN( R R t ValueErrorR R t as_tupleR t zero_indexes( R R RJ R RK ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyR s & c C s | j S( N( RJ ( R ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt hashable8 s c C s t S( N( t list( R ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt python_type< s c C s | | k S( N( ( R t xt y( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt compare_values@ s c s� � d k r t | � } n � d k sT � d k r� | sT t | d t t f � r� � rt � � f d � | D� � S� | � Sn# � � � � � f d � | D� � Sd S( Ni i c 3 s | ] } � | � Vq d S( N( ( t .0RO ( t itemproc( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pys <genexpr>R s c 3 s= | ]3 } � j | � � d k r+ � d n d � � Vq d S( i N( t _proc_arrayR ( RR RO ( t collectiont dimRS R ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pys <genexpr>W s ( R RM R t tuple( R t arrRS RV RU ( ( RU RV RS R sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyRT C s c C s t | j t j � o | j j S( N( R R R t Enumt native_enum( R ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt _against_native_enum` s c C s | S( N( ( R t bindvalue( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt bind_expressiong s c s1 � j j | � j | � � � � f d � } | S( Nc s- | d k r | S� j | � � j t � Sd S( N( R RT R RM ( t value( t item_procR ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt processo s ( R t dialect_implt bind_processor( R t dialectR` ( ( R_ R sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyRb j s c sv � j j | � j | | � � � � f d � } � j rr | � t j d � � � f d � � � � f d � } n | S( Nc s< | d k r | S� j | � � j � j r1 t n t � Sd S( N( R RT R RJ RW RM ( R^ ( R_ R ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyR` ~ s s ^{(.*)}$c s" � j | � j d � } t | � S( Ni ( t matcht groupt _split_enum_values( R^ t inner( t pattern( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt handle_raw_string� s c s8 | d k r | S� t | t j � r1 � | � n | � S( N( R R R t string_types( R^ ( Ri t super_rp( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyR` � s ( R Ra t result_processorR[ t ret compile( R Rc t coltypeR` ( ( Ri R_ Rh R Rk sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyRl y s N( R4 R5 R6 R R R> t comparator_factoryR; R R R: RL RN RQ RT R t memoized_propertyR[ R] Rb Rl ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyR � s F "1 c C s� d | k r# | r | j d � Sg S| j d d � } | j d d � } g } t j d | � } t } x_ | D]W } | d k r� | } ql | r� | j | j d d � � ql | j t j d | � � ql W| S( Nt "t ,s \"s _$ESC_QUOTE$_s \\s \s (")s ([^\s,]+),?( t splitt replaceRm R; t appendt extendt findall( t array_stringt textt resultt on_quotest in_quotest tok( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyRf � s ( Rm t R R R t sqlR R R R t eqR R t ClauseListt ColumnElementR t custom_opR+ RA RE RG R Rf ( ( ( sW /opt/alt/python27/lib64/python2.7/site-packages/sqlalchemy/dialects/postgresql/array.pyt <module> s o�