관리-도구
편집 파일: util.cpython-38.pyc
U -?�f�8 � @ s@ d dl mZ d dlmZ d dlZd dlZd dlZd dlmZ d dl Z d dl Z d dl m Z d dl Z d dl Z d dlmZ ddlmZ dd lmZ d d lmZ d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlm Z d dl mZ d dl m Z d dl m!Z! d dl m"Z" d dl m#Z# d dl m$Z$ e#�sLdd� Z%e% Z&Z'nej(Z&dd � Z'd d � Z)dPd d �Z*d!d"� Z+G d#d$� d$e,�Z-d%d&� Z.d'd(� Z/d)d*� Z0d+d,� Z1d-d.� Z2d/d0� Z3e!d1d2� �Z4d3d4� Z5d5d6� Z6d7d8� Z7dQd:d;�Z8d<d=� Z9G d>d?� d?e:�Z;d@dA� Z<dRdCdD�Z=dEdF� Z>dGdH� Z?dIdJ� Z@ejAdSdKdLdM�dNdO��ZBdS )T� )� annotations)�dequeN)�chain)� getsizeof)�Any� )�config)�mock� )�inspect)� Connection)�Column)�DropConstraint)� DropTable)�ForeignKeyConstraint)�MetaData)�Table��schema)�Integer�� decorator)� defaultdict)�has_refcount_gc)�inspect_getfullargspecc G s t �� t �� d S �N)�gc�collect)�args� r �G/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/util.py�non_refcount_gc_collect, s r! c C s d S r r r r r r �lazy_gc5 s r" c # sR t � } dd l}| �|� | D ]0� td|jd �D ]�� j� �fdd�fV q0q d S )Nr ���r c s � � | ��S r )�dumps��d�Zpickle_�protocolr r �<lambda>B � zpicklers.<locals>.<lambda>)�set�pickle�add�range�HIGHEST_PROTOCOL�loads)�picklersr, r r' r r1 9 s r1 c C s t j| |d�S )N)�k)�random�choices)� populationr2 r r r �random_choicesE s r6 c C s< t | t�rt| |�S | t�dd| � �tj�td|� S )N�1�0� )� isinstance�float�round�decimal�Decimal� to_integral� ROUND_FLOOR�pow)�value�precr r r � round_decimalI s ��rD c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )� RandomSetc C s t t�| ��}t�|� t|�S r )�listr+ �__iter__r3 �shuffle�iter)�self�lr r r rG T s zRandomSet.__iter__c C s4 t �dt| �d �}tt�| ��| }| �|� |S )Nr r )r3 �randint�lenrF r+ rG �remove)rJ �index�itemr r r �popY s z RandomSet.popc C s t t�| |��S r )rE r+ �union�rJ �otherr r r rR _ s zRandomSet.unionc C s t t�| |��S r )rE r+ � differencerS r r r rU b s zRandomSet.differencec C s t t�| |��S r )rE r+ � intersectionrS r r r rV e s zRandomSet.intersectionc C s t | �S r )rE �rJ r r r �copyh s zRandomSet.copyN) �__name__� __module__� __qualname__rG rQ rR rU rV rX r r r r rE S s rE c C s` t t�}| D ]\}}|| �|� q t|�D ].\}}||d� D ]}||| kr@ dS q@q,dS )zATrue if the given sorting conforms to the given partial ordering.NFT)r r+ r- � enumerate)�tuplesZsorted_elements�deps�parent�child�i�node�nr r r �conforms_partial_orderingl s rd c s> t t��| D ]\}}�| �|� q � �fdd�� t� |��S )Nc 3 s^ t | �dkrt| �V nB| D ]<}t| ��|g�}|��| �s � |�D ]}|g| V qFq d S )Nr )rM rF r+ rU rV )�elements�elemZsubsetZ sub_ordering��_all_orderings�edgesr r rh s z-all_partial_orderings.<locals>._all_orderings)r r+ r- rI )r] re r_ r` r rg r �all_partial_orderingsz s rj c C s> z || _ W n. tk r8 t�| j| j|| j| j�} Y nX | S )ac Return a function with a given __name__. Will assign to __name__ and return the original function if possible on the Python implementation, otherwise a new function will be constructed. This function should be phased out as much as possible in favor of @decorator. Tests that "generate" many named tests should be modernized. )rY � TypeError�types� FunctionType�__code__� __globals__� __defaults__� __closure__)�fn�namer r r �function_named� s � rt c O s` | � � }z$||f|�|�}| �ddd� |W S t�� }| j|� }|sN� n| Y S Y nX dS )a Run the given function under the given contextmanager, simulating the behavior of 'with' to support older Python versions. This is not necessary anymore as we have placed 2.6 as minimum Python version, however some tests are still using this structure. N)� __enter__�__exit__�sys�exc_info)�ctxrr �arg�kw�obj�resultrx Zraise_r r r �run_as_contextmanager� s r~ c C s dd� | D �S )z�Converts the results of sql execution into a plain set of column tuples. Useful for asserting the results of an unordered query. c S s h | ] }t |��qS r ��tuple)�.0�rowr r r � <setcomp>� s zrowset.<locals>.<setcomp>r )�resultsr r r �rowset� s r� c C s ds t | ��d S )NF)�AssertionError)�msgr r r �fail� s r� c O s� ddl m} t�� }|d }t|dd�}||_z| ||�W �S |�� |jj}|rjt ||� |� � � � n t |t j � ||_X dS )ztProvide bound MetaData for a single test, dropping afterwards. Legacy; use the "metadata" pytest fixture. r )�fixturesr �metadataN)� r� r r �getattrr� Zclose_all_sessions�baseZ _connection_fixture_connection� drop_all_tables_from_metadataZget_transaction�commitr �db)rr r r{ r� r� rJ Z prev_meta�cfcr r r �provide_metadata� s r� c sT t � � | D ]}� �|� q t� �� tj� fdd�| D �ddt� � d�� �d��S )a A facade around @testing.combinations() oriented towards boolean keyword-based arguments. Basically generates a nice looking identifier based on the keywords and also sets up the argument names. E.g.:: @testing.flag_combinations( dict(lazy=False, passive=False), dict(lazy=True, passive=False), dict(lazy=False, passive=True), dict(lazy=False, passive=True, raiseload=True), ) would result in:: @testing.combinations( ('', False, False, False), ('lazy', True, False, False), ('lazy_passive', True, True, False), ('lazy_passive', True, True, True), id_='iaaa', argnames='lazy,passive,raiseload' ) c s< g | ]4� d � � fdd��D ��ft� fdd��D �� �qS )�_c 3 s | ]}� � |d �r|V qdS �FN��get�r� r2 r% r r � <genexpr> s z/flag_combinations.<locals>.<listcomp>.<genexpr>c 3 s | ]}� � |d �V qdS r� r� r� r% r r r� s )�joinr� )r� ��keysr% r � <listcomp> s ��z%flag_combinations.<locals>.<listcomp>ra �a�,)Zid_�argnames)r+ �update�sortedr � combinationsrM r� )r� r& r r� r �flag_combinations� s ��r� c sL t ��}�dd� |d D �� }�fdd�� tj� fdd�tt|��D �|�S )Nc S s g | ] }t �� �qS r )r ZMock�r� rz r r r r� % s z'lambda_combinations.<locals>.<listcomp>r c s �� fdd�}d� |_ |S )Nc s � f | �� S r r �r{ )�lambda_arg_sets�posr r �fixture( s z<lambda_combinations.<locals>.create_fixture.<locals>.fixturez fixture_%3.3d)rY )r� r� )r� )r� r �create_fixture' s z+lambda_combinations.<locals>.create_fixturec s g | ]}� |�f�qS r r )r� ra )r� r r r� / s )r r r� r. rM )r� r{ r Zarg_setsr )r� r� r �lambda_combinations" s �r� c sJ t | �d }� fdd�|D �}t| j�}|�� � t�| j|�}|f |�S )z�Given a no-arg lambda and a namespace, return a new lambda that has all the values filled in. This is used so that we can have module-level fixtures that refer to instance-level variables using lambdas. r c s i | ]}|� � |��qS r )rQ r� r� r r � <dictcomp>= s z"resolve_lambda.<locals>.<dictcomp>)r �dictro r� rl rm rn )Z__fnr{ Zpos_argsZ pass_pos_argsZglbZnew_fnr r� r �resolve_lambda3 s r� �functionc s � fdd�}|S )z&Provide MetaData for a pytest fixture.c s � fdd�}t j�d�|�S )Nc 3 sB t �� }| _z � | |�}|�tj� |V W 5 |�tj� X d S r )r r r� �drop_allr r� Z create_all)rJ r� r} �rr r r �run_ddlH s z3metadata_fixture.<locals>.decorate.<locals>.run_ddl)�scope)r r� )rr r� ��ddlr� r �decorateG s z"metadata_fixture.<locals>.decorater )r� r� r r� r �metadata_fixtureD s r� c s t � fdd��}|S )ziForce the given table names to be dropped after test complete, isolating for foreign key cycles c s, z| ||�W �S t tjttj�� d� X d S )N)� include_names)�drop_all_tablesr r� r )rr r r{ ��namesr r �go^ s z force_drop_names.<locals>.gor )r� r� r r� r �force_drop_namesX s r� c @ s$ e Zd ZdZdd� Zdd� ZeZdS )�adictz,Dict keys available as attributes. Shadows.c C s0 z | | W S t k r* t�| |� Y S X d S r )�KeyErrorr� �__getattribute__)rJ �keyr r r r� k s zadict.__getattribute__c s t � fdd�|D ��S )Nc s g | ] }� | �qS r r )r� r� rW r r r� r s z"adict.__call__.<locals>.<listcomp>r )rJ r� r rW r �__call__q s zadict.__call__N)rY rZ r[ �__doc__r� r� �get_allr r r r r� h s r� c sN ddl m� � �fdd�}t|t�sB|�� �}||� W 5 Q R X n||� d S )Nr )�enginesc sT � j �| � | jjsFddlm} |jddd�� ��| � W 5 Q R X n ��| � d S )Nr )� assertionszCan't sort tablesF)Zassert_)Ztesting_reaperZprepare_for_drop_tables�dialect�supports_alterr� r� Zexpect_warningsr� )� connectionr� �r� r� r r r� z s �z)drop_all_tables_from_metadata.<locals>.go)r� r� r: r �begin)r� Zengine_or_connectionr� r� r r� r r� w s r� r c C s |d k rt |�}|d k r.|dks(td��|f}| �� ��}t|j|d��D ]�\}}|r�|d k rj|d |krjqH|�tt|d t� |d d��� qH|rH| j j s�qH|D ]j\}} |d k r�|d |kr�q�t|d t� t dt �t dt �|d d�} |�t t| jjg| jjg| d ��� q�qHW 5 Q R X d S ) Nr z2consider_schemas and schema are mutually exclusive)�consider_schemasr r r �x�y)rs )r+ r� r� �reversedZ%sort_tables_on_foreign_key_dependency�executer r r r� r� r r r r �cr� r� ) �engineZ inspectorr r� r� �connZ table_keyZfkcsZt_keyZfkc�tbr r r r� � sZ � �� � ��� � ����r� c s t � fdd��}|S )Nc s z| ||�W �S � � � X d S r )�_clear)rr rz r{ �� event_clsr r r� � s z!teardown_events.<locals>.decorater )r� r� r r� r �teardown_events� s r� c sL dd� }t tttttt|tttti� t� �td��� ���fdd���| �S )z�Returns the approximate memory footprint an object and all of its contents. source: https://code.activestate.com/recipes/577504/ c S s t �| �� �S r )r � from_iterable�itemsr% r r r � dict_handler� s z total_size.<locals>.dict_handlerr c sb t | ��krdS ��t | �� t| ��}� �� D ],\}}t| |�r0|tt�|| ���7 } q^q0|S )Nr )�idr- r r� r: �sum�map)�o�s�typ�handler�Z all_handlersZ default_size�seen�sizeofr r r� � s ztotal_size.<locals>.sizeof)r� rI rF r r� r+ � frozensetr )r� r� r r� r � total_size� s$ � r� c C sV | g}t � }d}|rR|�d�}||kr0|d7 }qt|t�r|rt|�|g | }q|S )z�given a cache key tuple, counts how many instances of actual tuples are found. used to alert large jumps in cache key complexity. r r )�objectrQ r: r� rF )�tup�stack�sentinelZ num_elementsrf r r r �count_cache_key_tuples� s r� r; r )�seconds�cleanupc c sN t � � }d V t � � | }|| krJz |� W 5 t�d|d�d| � d�� X d S )Nztest took too long (z.4fz seconds > �))�timer Z skip_test)r� r� �now�secr r r �skip_if_timeout s �r� )r )r� )Nr N)N)C� __future__r � collectionsr � contextlibr= r � itertoolsr r3 rw r r� rl �typingr r� r r r r� r r r r r r r r �sqlZ sql.sqltypesr �utilr r r r r! Z gc_collectr" r r1 r6 rD r+ rE rd rj rt r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� �contextmanagerr� r r r r �<module> sx +0 � 8 &