관리-도구
편집 파일: _autogen_fixtures.cpython-38.pyc
U 0?�f�& � @ s� U d dl mZ d dlmZ d 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 ddlmZ ddlmZ ddlmZ ddlmZ ddl m Z ddl m Z ddl m!Z! dd l m"Z" dd l#m$Z$ dd l#m%Z% e&� Z'd e(d < e �)ed!�d"d#� �Z*d$d%� Z+e+Z,d&e(d'< d(Z-d&e(d)< G d*d+� d+�Z.G d,d-� d-�Z/G d.d/� d/e/�Z0G d0d1� d1e/�Z1d(S )2� )� annotations)�Any)�Dict)�Set)�CHAR)�CheckConstraint)�Column)�event)� ForeignKey)�Index)�inspect)�Integer)�MetaData)�Numeric)�String)�Table)�Text)�text)�UniqueConstraint� )� autogenerate)�util)�api)�_fk_spec)�MigrationContext��ops)�config)�eq_)�clear_staging_env)� staging_envzSet[Any]�names_in_this_testZafter_parent_attachc C s t �| j� d S �N)r! �add�name)�table�parent� r' �U/opt/hc_python/lib/python3.8/site-packages/alembic/testing/suite/_autogen_fixtures.py� new_table% s r) c C s |dkr|t kS dS d S )Nr% T)r! )�objr$ �type_Z reflectedZ compare_tor' r' r( �_default_include_object* s r, r �_default_object_filtersN�_default_name_filtersc @ s6 e Zd ZU dZdZded< edd� �Zedd� �ZdS ) �ModelOne)Z unique_constraint_reflectionNr �schemac C s� | j }t|d�}td|tdtdd�tdtd��tdt�td td��td d �� td |tdtdd�td td �dd�� td|tdtdd�tdtdd�dt d�d�t ddd�� td|tdt �td tt d ��� |S ) N�r0 �user�idT�Z primary_keyr$ �2 �a1�pwZpw_idx�address� email_address�d F��nullable�order�order_id�amount� � �0�r< �server_defaultz amount >= 0�ck_order_amount�r$ �extra�x�uid�user.id) r0 r r r r r r r r r r r r ��clsr0 �mr' r' r( �_get_db_schema; sH � � � � �zModelOne._get_db_schemac C s� | j }t|d�}td|tdtdd�tdtd�dd �td td d �� td |tdtdd�tdtd�dd �tdtd��tddd�� td|tdtdd�tdtdd�dt d�d�tdtt d��t d d d�� td |tdtdd�td td��tdtt d ��t d!�� |S )"Nr1 r2 r3 Tr4 r$ r5 Fr; r6 rH )rD r8 r9 r: ZstreetZuq_emailrF r= r> r? � rA rB rC Zuser_idrJ z amount > -1rE �item� descriptionzorder.order_idzlen(description) > 5) r0 r r r r r r r r r r r rK r' r' r( �_get_model_schemah sN � � � � �zModelOne._get_model_schema) �__name__� __module__� __qualname__� __requires__r0 �__annotations__� classmethodrN rR r' r' r' r( r/ 6 s ,r/ c @ s e Zd Zddd�ZdS )� _ComparesFKsNc C s� t |d �\ }}}}}}}}}}t|d |� t||� t||� t||� t|| � t|| � t|| � t|| � t|| � tdd� |d jD �|� |d k r�|dkr�t| j��|�}|d d }t|d j|� q�t|d j|� nt|d j|� d S )N� r c S s g | ] }|j j�qS r' )�columnr$ )�.0�elemr' r' r( � <listcomp>� s z0_ComparesFKs._assert_fk_diff.<locals>.<listcomp>Zservergeneratedr$ )r r �elementsr �bindZget_foreign_keysr$ )�self�diffr+ Z source_tableZsource_columnsZ target_tableZtarget_columnsr$ Zconditional_nameZ source_schemaZonupdateZondeleteZ initiallyZ deferrableZfk_source_schemaZfk_source_tableZfk_source_columnsZfk_target_schemaZfk_target_tableZfk_target_columnsZ fk_onupdateZ fk_ondeleteZ fk_deferrableZ fk_initiallyZfksZserver_fk_namer' r' r( �_assert_fk_diff� s: � z _ComparesFKs._assert_fk_diff)NNNNNNN)rS rT rU rc r' r' r' r( rY � s �rY c @ s` e Zd ZU dd� Zedd� �Zi Zded< edd� �Zed d � �Z d d � Z d d� Z ddd�Z dS )� AutogenTestc c s0 |D ]&}t |t�r$| �|�E d H q|V qd S r" )� isinstance�list�_flatten_diffs)ra Zdiffs�dr' r' r( rg � s zAutogenTest._flatten_diffsc C s t jS r" )r �db�rL r' r' r( � _get_bind� s zAutogenTest._get_bindzDict[Any, Any]�configure_optsc C s6 t � | �� | _| �� | _| j�| j� | �� | _d S r" )r rk r` rN �m1� create_allrR �m2rj r' r' r( � setup_class� s zAutogenTest.setup_classc C s | j �| j� t� d S r" )rm �drop_allr` r rj r' r' r( �teardown_class� s zAutogenTest.teardown_classc C sd | j �� | _}dd| jddddttd� }| jr<|�| j� tj ||d� | _ }t � || j�| _ d S )NT�upgrades� downgrades�op.�sa.) � compare_type�compare_server_default�target_metadata� upgrade_token�downgrade_token�alembic_module_prefix�sqlalchemy_module_prefix�include_object� include_name�� connection�opts)r` �connect�connro r- r. rl �updater � configure�contextr �AutogenContext�autogen_context)ra r� �ctx_optsr� r' r' r( �setUp� s$ � � zAutogenTest.setUpc C s | j �� d S r" )r� �close�ra r' r' r( �tearDown� s zAutogenTest.tearDownNc C s> |d k r|| j jd< |d k r&|g| j _|d k r8|g| j _| j S )N�include_schemas)r� r� Z_object_filtersZ _name_filters)ra �object_filters� name_filtersr� r' r' r( �_update_context s zAutogenTest._update_context)NNN) rS rT rU rg rX rk rl rW rp rr r� r� r� r' r' r' r( rd � s �rd c @ s2 e Zd Zddeeddfdd�Zdd� Zdd� ZdS ) �AutogenFixtureTestFNc C s |r | j j} | j} | | _| _z�|| | _} t�| j�D ]} | �| j � q8| j �� ��} dd| dddd|||d� }|r~|� |� t j | |d� | _ }t �|| �}tjg d�}t�||� |r�|W 5 Q R � W �(S |�� W 5 Q R � W �S W 5 Q R X W 5 |�r| j j} | | _| _X d S ) NTrs rt ru rv ) rw rx ry rz r{ r| r} r~ r r� r� r )r` �dialect�max_identifier_lengthZ#_user_defined_max_identifier_length�_alembic_metadatar �to_listrn r� r� r r� r� r r� r Z UpgradeOpsr Z_produce_net_changesZas_diffs)ra rm ro r� r� r� r� Z return_opsr� r� Zexisting_lengthZmodel_metadatarM r� r� r� r� Zuor' r' r( �_fixture sP � � � &��zAutogenFixtureTest._fixturec C s t � tj| _d S r" )r r ri r` r� r' r' r( r� G s zAutogenFixtureTest.setUpc C s2 t | d�r(t�| j�D ]}|�| j� qt� d S )Nr� )�hasattrr r� r� rq r` r )ra rM r' r' r( r� K s zAutogenFixtureTest.tearDown)rS rT rU r- r. r� r� r� r' r' r' r( r� s � 8r� )2� __future__r �typingr r r Z sqlalchemyr r r r r r r r r r r r r r r � r r r Zddl.baser Z migrationr � operationsr Ztestingr r Z testing.envr r �setr! rW Z listens_forr) r, r- r. r/ rY rd r� r' r' r' r( �<module> sN b7?