관리-도구
편집 파일: fixtures.cpython-38.pyc
U 0?�f�$ � @ s� U d dl mZ d dlZd dlmZ d dlZd dlZd dlmZ d dlm Z d dl m Z d dl m Z d dl m Z d d l mZ d d l mZ d d l mZ d d l mZ d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dl Z ddl m Z ddl m Z ddl!m"Z" ddl#m$Z$ ddl%m&Z& ddl'm(Z( ddl'm)Z) ddl'm*Z* e�+� Z,e,�-d g� G d d � d e�ZG d d � d ee�Ze)�r�d d!lm.Z. nG d"d#� d#�Z.d$e._/d4d&d'�Z0i Z1d(e2d)< ed*d+� �Z3ed,d-� �Z4d5d0d1�Z5G d2d3� d3�Z6dS )6� )� annotationsN)�contextmanager)�Any)�Dict)�Column)�inspect)�MetaData)�String)�Table)�testing)�text)�config)�mock)�eq_)� TablesTest)�TestBase� )� _get_dialect� )�EnvironmentContext)�MigrationContext)� Operations)� sqla_compat��create_mock_engine)�sqla_14)�sqla_2ztest.cfgc @ sJ e Zd ZeZe�� dd� �Zejdd� �Zejdd� �Z ejdd� �Z d S ) r c c s& |j dd�� t|�V W 5 Q R X d S )NT)Z_per_migration)Zbegin_transactionr )�self�migration_context� r �F/opt/hc_python/lib/python3.8/site-packages/alembic/testing/fixtures.py� ops_context) s zTestBase.ops_contextc C s t j|tdd�d�S )NT)�transaction_per_migration��opts�r � configure�dict�r � connectionr r r r . s �zTestBase.migration_contextc C s t j|tddd�d�S )NT)r" �as_sqlr# r% r( r r r �as_sql_migration_context4 s �z!TestBase.as_sql_migration_contextc c s t j�� � }|V W 5 Q R X d S �N)r �db�connect)r �connr r r r) : s zTestBase.connectionN) �__name__� __module__� __qualname__r �is_sqlalchemy_futurer Zfixturer! r r+ r) r r r r r & s r c @ s e Zd ZdS )r N)r0 r1 r2 r r r r r @ s r )�FutureEngineMixinc @ s e Zd ZdZdS )r4 )Z sqlalchemy_14N)r0 r1 r2 � __requires__r r r r r4 H s r4 T� postgresql://c s$ g � � �fdd�}t | |���� fS )Nc s � � t| j�jd��� d S �N)�dialect)�append�str�compiler8 )�sqlZ multiparams�params��bufZenginer r �dumpR s zcapture_db.<locals>.dumpr )r8 r@ r r> r � capture_dbO s rA zDict[Any, Any]�_engsc + sh �� dd�rt�� }nt�� }��d|d�� tj� � �fdd�}tj� td|�� |V W 5 Q R X d S )NZbytes_ioFZsqlite)Z dialect_name� output_bufferc s |j f �� � | |�S r, ��update��arg�opt��conf�kwr r r& f s z)capture_context_buffer.<locals>.configurer& ) �pop�io�BytesIO�StringIOrE r r& r �patch�object)rK r? r& r rI r �capture_context_buffer\ s rR c + s� ddl m} ddlm} t�� � |� }|�� }|�|d�� fdd��}��d|i� t j ���fd d �}t j � t d |�� � V W 5 Q R X d S ) Nr )�_sqlite_file_dbr )�eventZbefore_cursor_executec s � � |d � d S )N� )�write)r/ �cursorZ statement� parameters�contextZ executemany�r? r r �bcey s z*capture_engine_context_buffer.<locals>.bcer) c s |j f �� � | |�S r, rD rF rI r r r& � s z0capture_engine_context_buffer.<locals>.configurer& )�envrS � sqlalchemyrT rM rO r. Z listens_forrE r r& r rP rQ )rK rS rT Zengr/ r[ r& r )r? rJ rK r � capture_engine_context_buffern s r^ �defaultFc s� i }|rt |d�|d< G � fdd�d�}|� �G �fdd�dt�}� rN� |d< |rZ||d< tsv| d krvtd ��d �_nt| ��|d k r�|�_d �_� s���fd d�}tj�|d�} n �|d< d } |�| |�} t | �t j _ | S )N)�naming_conventionZtarget_metadatac s( e Zd Zdd� Z� fdd�Zdd� ZdS )zop_fixture.<locals>.buffer_c S s g | _ d S r, ��lines�r r r r �__init__� s z$op_fixture.<locals>.buffer_.__init__c sF |� � }t�dd|�}� r6t�dd|�}t�dd|�}| j�|� d S )N�[\n\t]� z z\;\n*$)�strip�re�subrb r9 )r �msg�r* r r rV � s z!op_fixture.<locals>.buffer_.writec S s d S r, r rc r r r �flush� s z!op_fixture.<locals>.buffer_.flushN)r0 r1 r2 rd rV rl r rk r r �buffer_� s rm c s< e Zd Z� fdd�Z� fdd�Z� fdd�Z� fdd�Zd S ) zop_fixture.<locals>.ctxc s � S r, r rc rZ r r �get_buf� s z op_fixture.<locals>.ctx.get_bufc s g � j d d �< d S r, ra rc rZ r r �clear_assertions� s z(op_fixture.<locals>.ctx.clear_assertionsc s t � jdd� |D �� d S )Nc S s g | ]}t �d d|��qS )re rf )rh ri )�.0�sr r r � <listcomp>� s z3op_fixture.<locals>.ctx.assert_.<locals>.<listcomp>)r rb )r r< rZ r r �assert_� s z op_fixture.<locals>.ctx.assert_c s> � j D ] }t�dd|�|kr d S qds:td|� j f ��d S )Nre rf Fz"Could not locate fragment %r in %r)rb rh ri �AssertionError)r r< �stmtrZ r r �assert_contains� s �z'op_fixture.<locals>.ctx.assert_containsN)r0 r1 r2 rn ro rs rv r rZ r r �ctx� s rw r* � literal_bindsZmariadb�mysql)� � r ZMariaDBTc s: t | t�rt| �} | js t�t| j�d��}� �|� d S r7 )� isinstancer: r Zsupports_executionrt r; rV )ru Z multiparam�paramr< )r? � ctx_dialectr r �execute� s zop_fixture.<locals>.execute)r8 r rC ) r r r r Zserver_version_infoZsupports_native_booleanZ#non_native_boolean_check_constraintr ZMockr �alembic�op�_proxy) r8 r* r` rx Znative_booleanr$ rm rw r r) rY r )r* r? r~ r � op_fixture� s2 r� c @ s: e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d d �Zd S )�AlterColRoundTripFixture)� alter_columnc C s2 t j�� | _t�| j�| _t| j�| _t � | _ d S r, ) r r- r. r/ r r& rw r r� r �metadatarc r r r �setUp� s z AlterColRoundTripFixture.setUpc C s8 t d|�}t d|�}| jj�||�r4td||f ��d S )N�qz3Type objects %r and %r didn't compare as equivalent)r rw �implZ compare_typert )r �t1�t2�c1�c2r r r � _compare_type� s � �z&AlterColRoundTripFixture._compare_typec C sD t d||d�}t d||d�}| jj�||||�r@td||f ��d S )Nr� )�server_defaultz6server defaults %r and %r didn't compare as equivalent)r rw r� Zcompare_server_defaultrt )r r� �s1r� �s2r� r� r r r �_compare_server_default� s � �z0AlterColRoundTripFixture._compare_server_defaultc C s> t �| j� | j�� � | j�| j� W 5 Q R X | j�� d S r, )r Z%_safe_rollback_connection_transactionr/ �beginr� Zdrop_all�closerc r r r �tearDown� s z!AlterColRoundTripFixture.tearDownNc C s� t |�dd�|�dtd��|�dd�|�dd �d�}td | j|�}t�| j��� |�| j� t | j�}|� d �d }| j j d |j |j|jd k r�|jnd |jr�dnd |�dd �|�dd �|�dd �|�dd �d � W 5 Q R X t | j�}|� d �d }|d kr�|}t|d d|k�r|d n|j � | �|d |�d|d �� t|d |�d|j�� | �|d |�d d �|�d|d �d|k�r�|d jn|jd k �r�|jjjnd � d S )N�nameZcolname�typerz �nullableTr� )r� r� �xr F)Z existing_typeZexisting_server_defaultZexisting_nullabler� r� Znew_column_name�type_r_ )r �getr r r� r Z_ensure_scope_for_ddlr/ �creater Z get_columnsr� r� r� r� r� r� r r� r� r rG ) r Zfrom_Zto_�compare�column�tZinspZold_colZnew_colr r r �_run_alter_col s^ � � � � � ���z'AlterColRoundTripFixture._run_alter_col)N) r0 r1 r2 r5 r� r� r� r� r� r r r r r� � s r� )r6 )r_ FNFN)7� __future__r � configparser� contextlibr rM rh �typingr r r] r r r r r r r Zsqlalchemy.testingr r Z sqlalchemy.testing.assertionsr Zsqlalchemy.testing.fixturesr ZSQLAlchemyTablesTestr ZSQLAlchemyTestBaser� Z assertionsr � environmentr Z migrationr � operationsr �utilr Zutil.sqla_compatr r r � ConfigParserZtesting_config�readr4 r3 rA rB �__annotations__rR r^ r� r� r r r r �<module> sb � X