Edit File: schemaobj.cpython-37.opt-1.pyc
B �M!V� � @ sH d dl mZ d dlmZmZ ddlmZ ddlm Z G dd� de �Z dS ) � )�schema)�NULLTYPE�Integer� )� string_types)�utilc @ sz e Zd Zddd�Zddd�Zd dd�Zd dd �Zd d d �Zd d d �Zdd� Z dd� Z dd� Z d dd�Z dd� Z dd� ZdS )!� SchemaObjectsNc C s || _ d S )N)�migration_context)�selfr � r �M/opt/alt/python37/lib/python3.7/site-packages/alembic/operations/schemaobj.py�__init__ s zSchemaObjects.__init__c sZ | � � }dd� |D �}tj||f|�d|i�� tj� fdd�|D �d|i�}� �|� |S )Nc S s g | ]}t �|t��qS r )� sa_schema�Columnr )�.0�nr r r � <listcomp> s z8SchemaObjects.primary_key_constraint.<locals>.<listcomp>r c s g | ]}� j | �qS r )�c)r r )�tr r r s �name)�metadatar �Table�PrimaryKeyConstraint�append_constraint)r r � table_nameZcolsr �m�columns�pr )r r �primary_key_constraint s z$SchemaObjects.primary_key_constraintc s� | � � }||kr|| }n&|}tj||fdd� |D ��d| i� tj||fdd� |D ��d| i�}| rrd| |f n|� tjr�| | d< tj|� fdd�|D �f||||| d�| ��}|�|� |S ) Nc S s g | ]}t �|t��qS r )r r r )r r r r r r & s z8SchemaObjects.foreign_key_constraint.<locals>.<listcomp>r c S s g | ]}t �|t��qS r )r r r )r r r r r r + s z%s.%s�matchc s g | ]}d � |f �qS )z%s.%sr )r r )�tnamer r r 6 s )r �onupdate�ondelete� deferrable� initially)r r r r Zsqla_08�ForeignKeyConstraintr )r r �sourceZreferent� local_colsZ remote_colsr! r" r# Z source_schemaZreferent_schemar$ r Z dialect_kwr Zt1_colsZt1�fr )r r �foreign_key_constraint s6 z$SchemaObjects.foreign_key_constraintc sV t j|| �� fdd� |D ��d|i�� ||d< t j� fdd�|D �|�}� �|� |S )Nc S s g | ]}t �|t��qS r )r r r )r r r r r r F s z3SchemaObjects.unique_constraint.<locals>.<listcomp>r r c s g | ]}� j | �qS r )r )r r )r r r r I s )r r r �UniqueConstraintr )r r r& r' r �kwZuqr )r r �unique_constraintC s z SchemaObjects.unique_constraintc K sB t j|| �� t �dt�|d�}t j|fd|i|��}|�|� |S )N�x)r r )r r r r r �CheckConstraintr )r r r& Z conditionr r+ r Zckr r r �check_constraintO s z SchemaObjects.check_constraintc K s� | j ||d�}dd� tjtjdd� tjd�}y || }W n4 tk rl tdd�tdd � |D ��� ��Y nX ||d �}|� |� |S d S ) N)r c S s t jg g | d�S )N)r )r r% )r r r r �<lambda>Y s z2SchemaObjects.generic_constraint.<locals>.<lambda>c S s t jd| d�S )N� )r )r r. )r r r r r0 ] � )Z foreignkeyZprimary�unique�checkNz'type' can be one of %sz, c s s | ]}t |�V qd S )N)�repr)r r- r r r � <genexpr>d s z3SchemaObjects.generic_constraint.<locals>.<genexpr>)r ) �tabler r r* Z Constraint�KeyError� TypeError�join�sortedr ) r r r �type_r r+ r �typesZconstr r r �generic_constraintV s " z SchemaObjects.generic_constraintc C sF i }| j d k r:d| j jkr:| j jd }t|d�r:|j|d< tjf |�S )NZtarget_metadata�naming_convention)r Zopts�hasattrr? r ZMetaData)r r+ Zmtr r r r j s zSchemaObjects.metadatac O s< | � � }tj||f|�|�}x|jD ]}| �||� q$W |S )N)r r r Z foreign_keys�_ensure_table_for_fk)r r r r+ r r r( r r r r7 s s zSchemaObjects.tablec K s t j||f|�S )N)r r )r r r< r+ r r r �columnz s zSchemaObjects.columnc s< t j|p d| �� |d�� t j|f� fdd�|D ��|�}|S )NZno_table)r c s g | ]}t j�� |��qS r )r Z sqla_compatZ_textual_index_column)r r )r r r r � s z'SchemaObjects.index.<locals>.<listcomp>)r r r ZIndex)r r Z tablenamer r r+ �idxr )r r �index} s zSchemaObjects.indexc C s> d|kr.|� d�}d�|dd� �}|d }n|}d }||fS )N�.r ���)�splitr: )r � table_key�tokens�snamer r r r �_parse_table_key� s z SchemaObjects._parse_table_keyc C sr t |jt�rn|j�dd�\}}| �|�\}}||jkrHtj|||d�}n |j| }||jkrn|� t� |t �� dS )zUcreate a placeholder Table object for the referent of a ForeignKey. rE � )r N) � isinstanceZ_colspecr �rsplitrK Ztablesr r r Z append_columnr r )r r ZfkrH ZcnamerJ r Zrel_tr r r rA � s z"SchemaObjects._ensure_table_for_fk)N)N)NNNNNNN)N)N)N)N)�__name__� __module__� __qualname__r r r) r, r/ r> r r7 rB rD rK rA r r r r r s % r N) Z sqlalchemyr r Zsqlalchemy.typesr r Z util.compatr r1 r �objectr r r r r �<module> s