관리-도구
편집 파일: mypy.cpython-38.pyc
U -?�f�. � @ s� d dl mZ d dlZd dlZd dlmZ d dlZd dlZd dlZd dl Z ddl m Z ddl m Z ddlmZ d d l mZ e jjG d d � d e ��ZdS ) � )� annotationsN)�Path� )�TestBase� )�config)�eq_� )�utilc @ sz e Zd ZdZejdd�dd� �Zejdd�dd� �Zd d � Ze�� d d � �Z ejd d� �Z e dd� �Z dd� Z dd� ZdS )�MypyTest)Zno_sqlalchemy2_stubs�function)�scopec c s | � � E d H d S �N�� _cachedir��self� r �P/opt/hc_python/lib64/python3.8/site-packages/sqlalchemy/testing/fixtures/mypy.py�per_func_cachedir s zMypyTest.per_func_cachedir�classc c s | � � E d H d S r r r r r r �cachedir! s zMypyTest.cachedirc c s� d}t �� �l}tt|�d d��}|�d|� d�� W 5 Q R X tt|�d d��}|�d|� d�� W 5 Q R X |V W 5 Q R X d S ) N� �sqla_mypy_config.cfg�wz� [mypy] plugins = sqlalchemy.ext.mypy.plugin show_error_codes = True z� disable_error_code = no-untyped-call [mypy-sqlalchemy.*] ignore_errors = True �plain_mypy_config.cfgz^ [mypy] show_error_codes = True z� disable_error_code = var-annotated,no-untyped-call [mypy-sqlalchemy.*] ignore_errors = True )�tempfile�TemporaryDirectory�openr �write)r Z mypy_pathr � config_filer r r r % s. ��� ��� zMypyTest._cachedirc s ddl m� d� �fdd� }|S )Nr )�apiFc s� |d kr �}ddd|dt j�||r&dnd�g}t j�| �}t j�||�}| |kr\t�| |� |�|� t j�dd � � � |�\}}}|||fS )Nz--strictz--raise-exceptionsz --cache-dirz --config-filer r ZMYPY_FORCE_COLOR) �os�path�join�basename�shutil�copyfile�append�environ�pop�run) r# � use_pluginZ use_cachedir�args�filenameZ test_program�stdout�stderr�exitcode�r! r r r r+ N s, ��� z!MypyTest.mypy_runner.<locals>.run)FN)�mypyr! )r r r+ r r2 r � mypy_runnerJ s %zMypyTest.mypy_runnerc s d� �fdd� }|S )NFc s2 �� | �}� | |d�\}}}��| ||||� d S )N)r, )�_collect_messages� _check_output)r# r, �expected_messagesr/ r0 r1 �r4 r r r r+ w s �z)MypyTest.mypy_typecheck_file.<locals>.run)Fr )r r4 r+ r r8 r �mypy_typecheck_fileu s z MypyTest.mypy_typecheck_filec C s� t j�| �r| }n$t�� d j}t j�t j�|�| �}tt |�� d��}t j j D ],}|rPt j�|�rP|�t |�| � d�� qP|S )Nr z**/*.py)r" r# �isabs�inspect�stackr. r$ �dirname�listr �globr �optionsZmypy_extra_test_paths�isdir�extend)r= r# Z caller_path�filesZ extra_dirr r r �file_combinations� s zMypyTest.file_combinationsc s� ddl m} g }t�d�}t�d�}t|����}g }t|d�D �]�\� }|�|�} | r�| �d��d�\} } } t j t | �t | �fk r>t � d| �d� � q>|�|�} | �r�t| �d��} t| �d��}t| �d ��}t�d d | �d ��}|�rJ|�s6t�d dd� |�}t�ddd� |�}t�dd|�}t�ddd� |�}d } }d|� d�}|�r�tj�r�|d t�|�rldnddd� |dd � � }|�r�tj�r�t�d d d� |�}|�| ||�� f� q>|r>|�� fd d �|D �� g |d d �< q>W 5 Q R X |S ) Nr )�mypy_14z)\s*# EXPECTED(_MYPY)?(_RE)?(_TYPE)?: (.+)z%^#\s*PYTHON_VERSION\s?>=\s?(\d+\.\d+)r �.zRequires python >= %sr r z# noqa[:]? ?.*r � z([\[\]])c S s d| � d�� �S )N�\r ��group��mr r r �<lambda>� � z,MypyTest._collect_messages.<locals>.<lambda>z([\w_]+)c S s d| � d�� d�S )Nz (?:.*\.)?r �\*?rI rK r r r rM � rN �Listz builtins.listz\b(int|str|float|bool)\bc S s d| � d�� d�S )Nz builtins.r rO rI rK r r r rM � rN TzRevealed type is "�"z\b(List|Tuple|Dict|Set)\bz \b(List|Tuple|Dict|Set|Type)\bc S s | � d��� S )Nr )rJ �lowerrK r r r rM � rN zOptional\[(.*?)\]c S s | � d�� d�S )Nr z | NonerI rK r r r rM � rN c 3 s | ]\}}}� |||fV qd S r r )�.0�is_mypy�is_re� expected_msg��numr r � <genexpr>� s �z-MypyTest._collect_messages.<locals>.<genexpr>)Zsqlalchemy.ext.mypy.utilrE �re�compiler � enumerate�matchrJ � partition�sys� version_info�intr Z skip_test�bool�subr Zpy39Zpy310r( �striprB )r r# rE r7 Z expected_reZ py_ver_re�file_Zcurrent_assert_messages�linerL �major�_�minorrT rU Zis_typerV r rW r r5 � s� � � � �� � � � ��zMypyTest._collect_messagesc C s g }t j�|�}|�r�g }|�d�} | r�| �d�} t�d| �rN|�d| f� q$t�d| �rz| r�| �d�} t�d| �sZq�qZq$t�d| tj�r�q$t�d| �r$|�d | f� q$|D ]�\} } }}|� d d �}| s�d nd }t |�D ]t\}\}}|�r t�d|� d| � d|� d|� |� � |��rL �qZq�|� d| � d|� d|� |� �|� d d �kr� �qZq�|�|� q�||= q�|�r�d� |�}t d|dd� |�r�d� dd� |D ��}t d|dd� d�s�t d��|�r t t|�� d�� t d� dd� |D ��� d�s t d ��n&|dk�rt ||dd� t|d|d � d S ) N� r z.+\.py:\d+: error: .*�errorz/.+\.py:\d+: note: +(?:Possible overload|def ).*z.+\.py:\d+: note: +def \[.*z*.+\.py:\d+: note: .*(?:perhaps|suggestion)z.+\.py:\d+: note: .*Znote�'rQ z[SQLAlchemy Mypy plugin] r z.*z\:z\: �:z: z"Couldn't locate expected messages:)�sepc s s | ]\}}|V qd S r r �rS rh �msgr r r rY + s z)MypyTest._check_output.<locals>.<genexpr>zRemaining messages:Fz'expected messages not found, see stdoutz& messages from mypy were not consumed:c s s | ]\}}|V qd S r r ro r r r rY 1 s z&errors and/or notes remain, see stdout)rp )r" r# r% �splitr* rZ r] r( �I�replacer\ r$ �print�AssertionError�lenr )r r# r7 r/ r0 r1 Z not_locatedr. �outputZ raw_lines�e�olrX rT rU rp �prefix�idx�typ�errmsg�missing�extrar r r r6 � sr � � � �� zMypyTest._check_outputN)�__name__� __module__� __qualname__� __requires__r Zfixturer r r r4 r9 � staticmethodrD r5 r6 r r r r r s % * hr )� __future__r r; r" �pathlibr rZ r& r_ r �baser r r Z assertionsr r Z add_to_markerr3 r r r r r �<module> s