관리-도구
편집 파일: schemaobj.cpython-38.pyc
U 0?�f�$ � @ sh d dl mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d d lm Z d d l m Z d d lmZ d d lmZ d d lmZ d dlmZ d dlmZ ddlmZ ddlmZ e �rVd dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d dlm Z d dlm Z d dlm Z d dlm!Z! d d l"m#Z# dd l$m%Z% G d d � d �Z&d S )!� )� annotations)�Any)�Dict)�List)�Optional)�Sequence)�Tuple)� TYPE_CHECKING)�Union��schema)�Column)� Constraint)�Index)�Integer)�NULLTYPE� )�util)� sqla_compat)� ColumnElement)� TextClause)�CheckConstraint)� ForeignKey)�ForeignKeyConstraint)�MetaData)�PrimaryKeyConstraint)�Table)�UniqueConstraint)� TypeEngine)�MigrationContextc @ s e Zd Zd<ddd�dd�Zd=ddd d d d �d d�Zd>dddddd d dd d d d dd� dd�Zd?ddd d dd�dd�Zd@dddd dd�d d �ZdAddd d d d �d d!�Zd"d#�d$d%�Z dd&d'�d(d)�Z dd*d+d,�d-d.�Z dBd d d/d d0d1�d2d3�Z dd4d5�d6d7�Z d"d8dd9�d:d;�ZdS )C� SchemaObjectsNzOptional[MigrationContext]�None)�migration_context�returnc C s || _ d S �N)r" )�selfr" � r& �J/opt/hc_python/lib/python3.8/site-packages/alembic/operations/schemaobj.py�__init__) s zSchemaObjects.__init__z,Optional[sqla_compat._ConstraintNameDefined]�strz Sequence[str]z Optional[str]r )�name� table_name�colsr r# c sT | � � }dd� |D �}tj||f|�d|i�� tj� fdd�|D �d|i|��}|S )Nc S s g | ]}t �|t��qS r& �� sa_schemar r ��.0�nr& r& r' � <listcomp>7 s z8SchemaObjects.primary_key_constraint.<locals>.<listcomp>r c s g | ]}� j | �qS r& ��cr/ ��tr& r' r2 < s r* )�metadatar. r r ) r% r* r+ r, r � dialect_kw�m�columns�pr& r5 r' �primary_key_constraint. s ���z$SchemaObjects.primary_key_constraintz List[str]zOptional[bool]r ) r* �source�referent� local_cols� remote_cols�onupdate�ondelete� deferrable� source_schema�referent_schema� initially�matchr# c s� | � � }||kr"| | kr"|| }n&|}tj||fdd� |D ��d| i� tj||fdd� t�|�D ��d| i�}| r�d| |f n|� | | d< tj|� fdd�|D �f||||| d�| ��}|�|� |S ) Nc S s g | ]}t �|t��qS r& r- r/ r& r& r' r2 X s z8SchemaObjects.foreign_key_constraint.<locals>.<listcomp>r c S s g | ]}t �|t��qS r& r- r/ r& r& r' r2 _ s ��%s.%srG c s g | ]}d � |f �qS )rH r& r/ ��tnamer& r' r2 p s )r* rA rB rC rF )r7 r. r r Z unique_listr �append_constraint)r% r* r= r> r? r@ rA rB rC rD rE rF rG r8 r9 Zt1_cols�t1�fr& rI r' �foreign_key_constraint@ sR � ������ ����� z$SchemaObjects.foreign_key_constraintr )r* r= r? r r# c 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' r2 � s z3SchemaObjects.unique_constraint.<locals>.<listcomp>r r* c s g | ]}� j | �qS r& r3 r/ r5 r& r' r2 � s )r. r r7 r rK )r% r* r= r? r �kwZuqr& r5 r' �unique_constraint| s � �� z SchemaObjects.unique_constraintz*Union[str, TextClause, ColumnElement[Any]]zUnion[CheckConstraint])r* r= � conditionr r# c K sB t j|| �� t �dt�|d�}t j|fd|i|��}|�|� |S )N�xr r* )r. r r7 r r r rK )r% r* r= rQ r rO r6 Zckr& r& r' �check_constraint� s � z SchemaObjects.check_constraintr )r* r+ �type_r r# c K s� | j ||d�}dd� tjtjdd� tjd�}z || }W nD tk r| } z&tdd�tdd � |D ��� �| �W 5 d } ~ X Y nX ||d �}|� |� |S d S ) Nr c S s t jg g | d�S )N�r* )r. r rU r& r& r' �<lambda>� s �z2SchemaObjects.generic_constraint.<locals>.<lambda>c S s t jd| d�S )N� rU )r. r rU r& r& r' rV � � )Z foreignkey�primary�unique�checkNz'type' can be one of %sz, c s s | ]}t |�V qd S r$ )�repr)r0 rR r& r& r' � <genexpr>� s z3SchemaObjects.generic_constraint.<locals>.<genexpr>rU ) �tabler. r r r �KeyError� TypeError�join�sortedrK ) r% r* r+ rT r rO r6 �types�constZker& r& r' �generic_constraint� s( � ��� z SchemaObjects.generic_constraintr )r# c C sF i }| j d k r:d| j jkr:| j jd }t|d�r:|j|d< tjf |�S )NZtarget_metadata�naming_convention)r" �opts�hasattrrf r. r )r% rO �mtr& r& r' r7 � s � � zSchemaObjects.metadatar )r* r# c s� | � � }dd� |D �}|�dd�}|r<|D ]}d |_|_q*tj||f|�|�� � fdd�|D �}|D ]} � �| � qf� jD ]} | �|| � q|� S )Nc S s. g | ]&}t |t�r|jd k r&t�|�n|�qS r$ )� isinstancer r^ r �_copy)r0 r4 r& r& r' r2 � s �z'SchemaObjects.table.<locals>.<listcomp>�_constraints_includedFc sL g | ]D}t |ttf�rt|d d�� k rDt|d d�dk rDtj|� d�n|�qS )�parentN)Z target_table)rj r r �getattrr rk )r0 �elemr5 r& r' r2 � s ���) r7 �poprZ �indexr. r rK Z foreign_keys�_ensure_table_for_fk) r% r* r: rO r9 r, rl r4 � constraintsrd rM r& r5 r' r^ � s" � � zSchemaObjects.tabler r )r* rT r# c K s t j||f|�S r$ )r. r )r% r* rT rO r& r& r' �column� s zSchemaObjects.columnz4Sequence[Union[str, TextClause, ColumnElement[Any]]]r )r* � tablenamer: r r# c sD t j|p d| �� |d�� � |d< t j|f� fdd�|D ��|�}|S )NZno_tabler �_tablec s g | ]}t j�� |��qS r& )r r Z_textual_index_columnr/ r5 r& r' r2 s z'SchemaObjects.index.<locals>.<listcomp>)r. r r7 r )r% r* ru r: r rO �idxr& r5 r' rq � s ����zSchemaObjects.indexzTuple[Optional[str], str])� table_keyr# c C s> d|kr.|� d�}d�|dd� �}|d }n|}d }||fS )N�.r ���)�splitra )r% rx �tokens�snamerJ r& r& r' �_parse_table_key s z SchemaObjects._parse_table_keyr )r7 �fkr# c 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. ry � r N) rj Z_colspecr) �rsplitr~ Ztablesr. r r4 � append_columnr r )r% r7 r rx �cnamer} rJ Zrel_tr& r& r' rr s z"SchemaObjects._ensure_table_for_fk)N)N)NNNNNNN)N)N)N)N)�__name__� __module__� __qualname__r( r< rN rP rS re r7 r^ rt rq r~ rr r& r& r& r' r ( s2 � � �(A � � � ' � r N)'� __future__r �typingr r r r r r r r Z sqlalchemyr r. Zsqlalchemy.sql.schemar r r Zsqlalchemy.typesr r rW r r Zsqlalchemy.sql.elementsr r r r r r r r r Zsqlalchemy.sql.type_apir Zruntime.migrationr r r& r& r& r' �<module> s: