Edit File: sqla_compat.cpython-37.pyc
B �M!V � @ s� d dl Z d dlmZ d dlmZmZmZ d dlmZ d dlm Z m Z d dl m Z d dl mZ d dlmZ d d lmZ d d � Zed d� e �de�D ��ZedkZedkZedkZedkZedkZedkZ edkZ edkZ edkZ edkZ edkZ edkZ!e�r d dlm"Z" n d d lm#Z" d d � Z$d d � Z%d!d"� Z&d#d$� Z'd%d&� Z(d'd(� Z)G d)d*� d*e j*�Z+ee+�d+d,� �Z,G d-d.� d.e�Z-ee-�d/d0� �Z.d1d2� Z/d3d4� Z0dS )5� N)� __version__)�ForeignKeyConstraint�CheckConstraint�Column)�types)�schema�sql)�traverse)�compiles)�_BindParamClause� )�compatc C s yt | �S | S d S )N)�int)�value� r �I/opt/alt/python37/lib/python3.7/site-packages/alembic/util/sqla_compat.py� _safe_int s r c C s g | ] }t |��qS r )r )�.0�xr r r � <listcomp> s r z (\d+|[abc]\d))r � � )r r � )r � r )r r � )r r � )r r r )r r r )r r r )r r r )r r r )r r � )� TextClause)� _TextClausec C s t | t�r| jS | jS d S )N)� isinstancer �parent�table)� constraintr r r �_table_for_constraint& s r# c C s< t | t�rdd� | jD �S t | t�r.t| j�S t| j�S d S )Nc S s g | ] }|j �qS r )r )r Zfkr r r r / s z+_columns_for_constraint.<locals>.<listcomp>)r r �elementsr � _find_columnsZsqltext�list�columns)r" r r r �_columns_for_constraint- s r( c s� t r� fdd�� jD �}ndd� � jD �}� jj}� jj}� jd jjj}� jd jjj}dd� � jD �}� j}� j }� j } � j } ||||||||| | f S )Nc s g | ]}� j | j�qS r )r' �name)r �key)r" r r r 9 s z _fk_spec.<locals>.<listcomp>c S s g | ] }|j j�qS r )r r) )r �elementr r r r < s r c S s g | ] }|j j�qS r )�columnr) )r r+ r r r r B s ) �sqla_100Z column_keysr$ r r) r r, r! �ondelete�onupdate� deferrable� initially) r" Zsource_columnsZ source_tableZ source_schemaZ target_schemaZ target_tableZtarget_columnsr. r/ r0 r1 r )r" r �_fk_spec6 s r2 c C s. t r | jS | jd k o(tt| jdd �tj�S d S )N�target)r- Z _type_boundZ _create_ruler �getattr�sqltypesZ SchemaType)r" r r r �_is_type_boundM s r6 c C s t � }t| i d|ji� |S )z2locate Column objects within the given expression.r, )�setr �add)Zclause�colsr r r r% ] s r% c C sF t |tj�r&t|tj�}| �|� |S t |t�r:t| |�S t d��dS )zAa workaround for the Index construct's severe lack of flexibilityz#String or text() construct expectedN) r r � string_typesr r5 �NULLTYPE� append_columnr �_textual_index_element� ValueError)r! Ztext_�cr r r �_textual_index_columne s r@ c @ s$ e Zd ZdZdZdd� Zdd� ZdS )r= a� Wrap around a sqlalchemy text() construct in such a way that we appear like a column-oriented SQL expression to an Index construct. The issue here is that currently the Postgresql dialect, the biggest recipient of functional indexes, keys all the index expressions to the corresponding column expressions when rendering CREATE INDEX, so the Index we create here needs to have a .columns collection that is the same length as the .expressions collection. Ultimately SQLAlchemy should support text() expressions in indexes. See https://bitbucket.org/zzzeek/sqlalchemy/issue/3174/ support-text-sent-to-indexes Z_textual_idx_elementc C s8 || _ || _|j| _t�| jjtj�| _|�| j� d S )N) r! �textr* r r r5 r; � fake_columnr< )�selfr! rA r r r �__init__� s z _textual_index_element.__init__c C s | j gS )N)rB )rC r r r � get_children� s z#_textual_index_element.get_childrenN)�__name__� __module__� __qualname__�__doc__Z__visit_name__rD rE r r r r r= q s r= c K s |j | jf|�S )N)�processrA )r+ �compiler�kwr r r � _render_textual_index_column� s rM c @ s e Zd ZdS )�_literal_bindparamN)rF rG rH r r r r rN � s rN c K s |j | f|�S )N)Zrender_literal_bindparam)r+ rK rL r r r �_render_literal_bindparam� s rO c C s t rt| j�S t| j�S d S )N)�sqla_08r& Z expressionsr' )�idxr r r �_get_index_expressions� s rR c C s dd� t | �D �S )Nc S s g | ]}t |d d��qS )r) N)r4 )r �expr r r r � s z+_get_index_column_names.<locals>.<listcomp>)rR )rQ r r r �_get_index_column_names� s rT )1�reZ sqlalchemyr Zsqlalchemy.schemar r r r r5 r r Zsqlalchemy.sql.visitorsr Zsqlalchemy.ext.compilerr Zsqlalchemy.sql.expressionr � r r �tuple�findallZ_versZsqla_07Zsqla_079rP Zsqla_083Zsqla_084Zsqla_09Zsqla_092Zsqla_094Zsqla_099r- Zsqla_105r r r# r( r2 r6 r% r@ Z ColumnElementr= rM rN rO rR rT r r r r �<module> sL