관리-도구
편집 파일: sqla_compat.cpython-38.pyc
U 0?�f@L � @ s d dl mZ d dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d d lm Z d d lm Z d d lm Z d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dlm Z d dlm Z d dl m Z d dl!m"Z" d dl#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* d d!l+m,Z, e�rJd d"lm-Z- d d#lm.Z. d d$lm/Z/ d d%lm0Z0 d d&lm1Z1 d d'lm2Z2 d d(l3m4Z4 d d)l!m5Z5 d d*l6m7Z7 d d+l8m9Z9 d d,l#m:Z: d d-l;m<Z< d d.l;m=Z= d d/l>m?Z? d d0l>m@Z@ ed1ed2 d3�ZAG d4d5� d5e �ZBd6d7d8�d9d:�ZCeDd;d<� e�Ed=e�D ��ZFeFd>kZGeFd?kZHeFd@kZIeFdAkZJeFdBkZKeZLzd dClMmNZN W n" eOk �r� d dCl#mNZN Y nX G dDdE� dE�ZPe�r dFd6dGdH�dIdJ�ZQn d dKlRmQZQ zd dLlmSZS W n4 eOk �rl e�s`G dMdN� dNeP�ZSdOZTdOZUY nX dPZTeFdQkZUzd dRlmVZV W n0 eOk �r� e�s�G dSdT� dTeP�ZVdOZWY n$X eXeVe"�ZYd�dUdVdWdX�dYdZ�ZZdPZWeK�r�d d[l!m[Z[ n d d\l\m]Z[ ede^e[f Z_ee^e[f Z`d]d^d_�d`da�Zad]dbd_�dcdd�Zbd]ded_�dfdg�ZcdhZdejedidjdk�dldm��Zfd�dndo�Zgdpdqdk�drds�Zhdpdtdk�dudv�Zidpdtdk�dwdx�ZjdidVdk�dydz�Zkd{d|d}�d~d�Zld1d1d��d�d��Zmdpd�dk�d�d��Znd�d��d�d��Zodpd6d�dVd��d�d��Zpd�d�� Zqd�d�� ZrdVd��d�d��ZsdVd��d�d��Ztd�d�d��d�d��Zud�d�� Zvd�d�dtd��d�d��Zwd�d�� Zxeyejzd���r|d�d�� Z{d�d�� Z|nd�d�� Z{d�d�� Z|d�d�d��d�d��Z}d�dVd��d�d��Z~d�dVd��d�d��Zd�d�� Z�d�d�dtd��d�d��Z�d�d�d�d��d�d��Z�d1d�d1d��d�d��Z�G d�d d�ej:�Z�eQe��d�d�d6dĜd�dƄ�Z�G d�dȄ d�e$�Z�eQe��d�d�d6dĜd�dʄ�Z�d�d�d͜d�dτZ�d�d�dedҜd�dԄZ�d�d�dVdҜd�dׄZ�d�dVdٜd�dۄZ�d�d݄ Z�d�d�d�d�d�Z�eH�r�d d�lm�Z� d d�lm�Z� e�Z�n"d d�lm�Z� d�d� Z�d�d��d�d�Z�d{dVd�d�d�Z�d�dVd�d�d�Z�dS )�� )� annotationsN)�Any)�Callable)�Dict)�Iterable)�Iterator)�Mapping)�Optional)�Protocol)�Set)�Type)� TYPE_CHECKING)�TypeVar)�Union)� __version__)�inspect��schema)�sql)�types)�url)�CheckConstraint)�Column)�ForeignKeyConstraint)�visitors)� DialectKWArgs)� BindParameter)� ColumnClause)� quoted_name)� TextClause)�UnaryExpression)�traverse)� TypeGuard)� ClauseElement)�Index)�Table)� Connection)�Dialect)� Transaction)� Inspector)�ColumnCollection)� SQLCompiler)�Insert)� ColumnElement)� Constraint)� SchemaItem)�Select)� TableClause�_CE)zColumnElement[Any]r/ )�boundc @ s e Zd Zddddd�dd�ZdS )�_CompilerProtocolr �str)�element�compiler�kw�returnc K s d S �N� )�selfr6 r7 r8 r; r; �F/opt/hc_python/lib/python3.8/site-packages/alembic/util/sqla_compat.py�__call__> � z_CompilerProtocol.__call__N)�__name__� __module__� __qualname__r> r; r; r; r= r4 = s r4 r5 zUnion[int, str])�valuer9 c C s z t | �W S | Y S X d S r: )�int)rC r; r; r= � _safe_intA s rE c C s g | ] }t |��qS r; )rE )�.0�xr; r; r= � <listcomp>I s rH z (\d+|[abc]\d))� � )rI � )rI rK � )rI rK � )� )� _NONE_NAMEc @ s e Zd ZdZdS )� _Unsupportedz.Placeholder for unsupported SQLAlchemy classesN)r@ rA rB �__doc__r; r; r; r= rP Y s rP zType[ClauseElement]z0Callable[[_CompilerProtocol], _CompilerProtocol])r6 �dialectsr9 c G s d S r: r; )r6 rR r; r; r= �compiles_ s rS )rS )�Computedc @ s e Zd ZdS )rT N�r@ rA rB r; r; r; r= rT k s rT FT)rI rJ � )�Identityc @ s e Zd ZdS )rW NrU r; r; r; r= rW y s rW z&Union[Identity, schema.Sequence, None]�boolzDict[str, Any])�identity�dialect_kwargsr9 c s� � d kr i S t r8� �� }|r�t� t�s*t�|�� j� nHi }t� t�rd� j|d< � j d k rd� j |d< d}|�� fdd�|D �� |S )N�always�on_null) �start� incrementZminvalueZmaxvalueZ nominvalueZ nomaxvalue�cycle�cache�orderc s* i | ]"}t � |d �d k r|t � |d ��qS r: )�getattr�rF �key�rY r; r= � <dictcomp>� s � z._get_identity_options_dict.<locals>.<dictcomp>) �identity_has_dialect_kwargsZ_as_dict� isinstancer �AssertionError�updaterZ rW r[ r\ )rY rZ �as_dict�attrsr; re r= �_get_identity_options_dict� s& ��rm )� _NoneName)�symbol�_ConstraintNamez!TypeGuard[_ConstraintNameDefined])�namer9 c C s | t kpt| ttf�S r: )rO rh r5 rn �rq r; r; r= �constraint_name_defined� s rs zTypeGuard[str]c C s t | t�S r: )rh r5 rr r; r; r= �constraint_name_string� s rt z Optional[str]c C s t | �r | S d S r: )rt rr r; r; r= �constraint_name_or_none� s ru �autozOptional[Connection]zIterator[None])� connectionr9 c c sl z | j }W n tk r" d }Y nX |d kr4d V n4|� sb| d k sFt�| �� � d V W 5 Q R X nd V d S r: )�in_transaction�AttributeErrorri �begin�rw rx r; r; r= �_ensure_scope_for_ddl� s r| c C s t r| j|d�S | j|d�S d S )N)� hide_password)�sqla_14Zrender_as_stringZ __to_string__)r r} r; r; r= �url_render_as_string� s r r& r( c C s t | �}|r|S | �� S d S r: )�_get_connection_transactionrz �rw Z transactionr; r; r= �"_safe_begin_connection_transaction� s r� �Nonec C s t | �}|r|�� d S r: )r� �commitr� r; r; r= �#_safe_commit_connection_transaction� s r� c C s t | �}|r|�� d S r: )r� �rollbackr� r; r; r= �%_safe_rollback_connection_transaction� s r� c C s, z | j }W n tk r Y dS X |� S d S )NF)rx ry r{ r; r; r= � _get_connection_in_transaction s r� r$ z Iterable[ColumnElement[Any]])�idxr9 c C s | j S r: )� expressions)r� r; r; r= � _idx_table_bound_expressions s r� )� schema_itemr9 c K s& t | d�r| jf |�S | jf |�S d S )N�_copy)�hasattrr� �copy)r� r8 r; r; r= r� s r� zOptional[Transaction]c C s t r | �� S | j}|jS d S r: )r~ Zget_transaction�_rootZ_Connection__transaction)rw �rr; r; r= r� s r� zurl.URL)r9 c O s* t tjd�rtjj| |�S tj| |�S d S )N�create)r� r �URLr� )�argr8 r; r; r= � _create_url s r� zUnion[str, None])� connectable� tablename� schemanamer9 c C s( t rt| ��||�S | j�| ||�S d S r: )r~ r Z has_table�dialect)r� r� r� r; r; r= �_connectable_has_table' s �r� c K s@ t r.| �� �}|�||�W 5 Q R � S Q R X n| j�||�S d S r: )r~ Z_operation_context�execute�bind)� inspectorZ statement�params�connr; r; r= �_exec_on_inspector2 s r� c C s& t s | jS ddlm} | j|jkS d S )Nr r )r~ Znullable�sqlalchemy.sqlr Z_user_defined_nullableZNULL_UNSPECIFIED)Zmetadata_columnr r; r; r= �_nullability_might_be_unset: s �r� c G s t sdS tdd� | D ��S d S )NFc s s | ]}t |t�V qd S r: )rh rT �rF �sdr; r; r= � <genexpr>I s z._server_default_is_computed.<locals>.<genexpr>)� has_computed�any�Zserver_defaultr; r; r= �_server_default_is_computedE s r� c G s t sdS tdd� | D ��S d S )NFc s s | ]}t |t�V qd S r: )rh rW r� r; r; r= r� P s z._server_default_is_identity.<locals>.<genexpr>)r~ r� r� r; r; r= �_server_default_is_identityL s r� r. r% )� constraintr9 c C s* t | t�r | j}|d k s t�|S | jS d S r: )rh r �parentri �table)r� r� r; r; r= �_table_for_constraintS 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� )rF Zfkr; r; r= rH ^ s z+_columns_for_constraint.<locals>.<listcomp>)rh r �elementsr � _find_columnsZsqltext�list�columns�r� r; r; r= �_columns_for_constraint\ s r� r) )r� r� r9 c C s t r| �|d �S | �|d �S d S r: )r~ Z reflect_tableZ reflecttable)r� r� r; r; r= �_reflect_tablee s �r� c C s* t | �r"t| �\}}|�|j|�S | S d S r: )�_type_has_variants�_get_variant_mapping�getrq )�type_r� Z base_type�mappingr; r; r= �_resolve_for_variantn s r� �_variant_mappingc C s t | j�S r: )rX r� �r� r; r; r= r� x s r� c C s | | j fS r: )r� r� r; r; r= r� { s r� c C s t | �tjkS r: )�type�sqltypesZVariantr� r; r; r= r� � s c C s | j | jfS r: )�implr� r� r; r; r= r� � s r r c s� t r0� jd k st�� jd k s t�t� jt�s0t�� fdd�� jD �}� jj}� jj }� jd j j j }� jd j j j}dd� � jD �}� j }� j }� j} � j} ||||||||| | f S )Nc s g | ]}� j | j�qS r; )r� rq rc r� r; r= rH � s z _fk_spec.<locals>.<listcomp>r c S s g | ] }|j j�qS r; )�columnrq )rF r6 r; r; r= rH � s )r r� ri r� rh r� r% Z column_keysrq r r� r� �ondelete�onupdate� deferrable� initially) r� Zsource_columnsZ source_tableZ source_schemaZ target_schema� target_tableZtarget_columnsr� r� r� r� r; r� r= �_fk_spec� s6 ��r� c C sF | j d �� }|�d�}|�d� d�|�}| jd k s:t�|| jjkS )Nr �.���)r� Z _get_colspec�split�pop�joinr� ri rd )r� �spec�tokensZtablekeyr; r; r= �_fk_is_self_referential� s r� c C s | j S r: )Z _type_boundr� r; r; r= �_is_type_bound� s r� 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� r* z%Union[Column[Any], ColumnClause[Any]])� collectionr� r9 c C sH |j dk st�| |j }t| d�s,t| d�r:| j�|� n | �|� dS )z(remove a column from a ColumnCollection.NZ _immutableZ _readonly)rd ri r� Z_parent�remove)r� r� Z to_remover; r; r= � _remove_column_from_collection� s r� z*Union[str, TextClause, ColumnElement[Any]]z&Union[ColumnElement[Any], Column[Any]])r� �text_r9 c C sp t |t�r$t|tj�}| �|� |S t |t�r8t| |�S t |t�rNt| |j �S t |t j �rdt || �S t d��dS )zAa workaround for the Index construct's severe lack of flexibilityz#String or text() construct expectedN)rh r5 r r� �NULLTYPE� append_columnr �_textual_index_element�_textual_index_column�textr r- �_copy_expression� ValueError)r� r� �cr; r; r= r� � s r� )� expressionr� r9 c s � fdd�}t �| i |�S )Nc sV t | t�rN| jd k rN| j� k rN| j� jkr6� j| j S t| �}� �|� |S nd S d S r: )rh r r� rq r� r� r� )�colr� �r� r; r= �replace� s ��� z!_copy_expression.<locals>.replace)r Zreplacement_traverse)r� r� r� r; r� r= r� � s �r� c @ s. e Zd ZdZdZdddd�dd�Zd d � Zd S ) r� aQ 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 SQLAlchemy issue 3174. Z_textual_idx_elementr% r r� )r� r� r9 c C s8 || _ || _|j| _t�| jjtj�| _|�| j� d S r: ) r� r� rd r r r� r� � fake_columnr� )r< r� r� r; r; r= �__init__ s z _textual_index_element.__init__c C s | j gS r: )r� )r< r; r; r= � get_children s z#_textual_index_element.get_childrenN)r@ rA rB rQ Z__visit_name__r� r� r; r; r; r= r� � s r� r+ )r6 r7 r9 c K s |j | jf|�S r: )�processr� �r6 r7 r8 r; r; r= � _render_textual_index_column s r� c @ s e Zd ZdS )�_literal_bindparamNrU r; r; r; r= r� s r� c K s |j | f|�S r: )Zrender_literal_bindparamr� r; r; r= �_render_literal_bindparam! s r� r r )r� r9 c C s t r | jS i S d S r: )�sqla_13�kwargs)r� r; r; r= �_column_kwargs( s r� zUnion[Index, Constraint]zOptional[Dialect])r� r� r9 c C s� | j d krd S |d k st�tr.|jj| dd�S t| j d�rFt| j �}nt}|t| j �dd�}| j |d�} t | t j �r�|� |d �}|�| �S |j�| �S d S )NF�Z_alembic_quote�quote)r� rr )rq ri r~ �identifier_preparer�format_constraintr� r� r r5 � __class__rh r r$ Z ddl_compilerZ_prepared_index_name)r� r� Zquoted_name_cls�new_name�dr; r; r= �_get_constraint_final_name/ s" � r� zUnion[Constraint, Index]c C sD t r6| jd krdS |d k s t�|jj| dd�}|d k S | jd k S d S )NFr� )r~ rq ri r� r� )r� r� rq r; r; r= �_constraint_is_namedT s �r� r' )� mysql_dialectr9 c C s t r | jS t| jo| j�S d S r: )r~ Z is_mariadbrX Zserver_version_info� _is_mariadb�r� r; r; r= r� c s �r� c C s | j S r: )� _mariadb_normalized_version_infor� r; r; r= r� m s r� zUnion[TableClause, Table]r, )r� r9 c C s t r| �� �� S | jdd�S d S )NT)�inline)r~ �insertr� )r� r; r; r= �_insert_inlineq s r� )�create_mock_engine)�select�� create_enginec K s t dd|d�S )Nz postgresql://Zmock)Zstrategy�executorr )r r r8 r; r; r= r� � s �r� r0 c O s t jt| �f|�S r: )r r r� )r� r8 r; r; r= �_select� s r )�indexr9 c C s | j D ]}t|�r dS qdS �NTF)r� � is_expression)r �exprr; r; r= �is_expression_index� s r )r r9 c C s* t | t�r| j} q t | t�r"| jr&dS dS r )rh r r6 r Z is_literal)r r; r; r= r � s r )F)T)�� __future__r � contextlib�re�typingr r r r r r r r r r r r r Z sqlalchemyr r r r r r� Zsqlalchemy.enginer Zsqlalchemy.schemar r r r� r Zsqlalchemy.sql.baser Zsqlalchemy.sql.elementsr r r r r Zsqlalchemy.sql.visitorsr! �typing_extensionsr" r# r$ r% r& r' r( Z sqlalchemy.engine.reflectionr) r* Zsqlalchemy.sql.compilerr+ Zsqlalchemy.sql.dmlr, r- Zsqlalchemy.sql.schemar. r/ Zsqlalchemy.sql.selectabler0 r1 r2 r4 rE �tuple�findallZ_versr� r~ Z sqla_14_18Z sqla_14_26Zsqla_2Zsqlalchemy_versionZsqlalchemy.sql.namingrO � ImportErrorrP rS Zsqlalchemy.ext.compilerrT r� Zhas_computed_reflectionrW Z has_identity� issubclassrg rm rn Zsqlalchemy.utilro r5 rp Z_ConstraintNameDefinedrs rt ru ZAUTOINCREMENT_DEFAULT�contextmanagerr| r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� Z TypeEnginer� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r Z _sa_selectr r r r r; r; r; r= �<module> s, � �' ! %