a XC?hsk�@s dZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z m Z ddlmZddlmZmZddlmZd d d �ZGd d �d e�Zdd�Zdd�Zdd�Zdd�Zdd�ZGdd�de�ZGdd�de�ZGdd�de�Z Gdd�de�Z!dS)!z#Guido van Rossum �N)�chain�)�driver�tokenize�token)� find_root)�pytree�pygram)� btm_matcherTcCsTt|ggdg�}g}t�|j�D].\}}}|�d�r |rD|dd�}|�|�q |S)N�*�fix_�)� __import__�pkgutilZ iter_modules�__path__� startswith�append)Z fixer_pkgZ remove_prefixZpkgZ fix_names�finder�nameZispkg�r�5/opt/alt/python39/lib64/python3.9/lib2to3/refactor.py�get_all_fix_namess   rc@s eZdZdS)� _EveryNodeN��__name__� __module__� __qualname__rrrrr+srcCs�t|tjtjf�r(|jdur t�|jhSt|tj�rH|jrDt|j�St�t|tj �r�t �}|jD]}|D]}|� t|��qhq`|St d|��dS)Nz$Oh no! I don't understand pattern %s) � isinstancerZ NodePatternZ LeafPattern�typerZNegatedPatternZcontent�_get_head_typesZWildcardPattern�set�update� Exception)Zpat�r�p�xrrrr/s     rc Cs�t�t�}g}|D]v}|jrbzt|j�}WntyF|�|�Yq�0|D]}||�|�qLq|jdur~||j�|�q|�|�qtt j j � �t j j �D]}||�|�q�t|�S�N)� collections� defaultdict�list�patternrrrZ _accept_typerr �python_grammarZ symbol2number�values�tokens�extend�dict)Z fixer_listZ head_nodesZevery�fixerZheadsZ node_typerrr�_get_headnode_dictKs$     �r1cs�fdd�t�d�D�S)Ncsg|]}�d|�qS��.r)�.0�fix_name�Zpkg_namerr� hs�z+get_fixers_from_package..F)rr6rr6r�get_fixers_from_packageds �r8cCs|Sr&r)�objrrr� _identityksr:csVd}t�t�|�j���fdd�}ttjtjtj h�}t �}z�|�\}}||vrTq>q>|tj krl|rf�q6d}q>|tj k�r6|dk�r6|�\}}|tj ks�|dkr��q6|�\}}|tj ks�|dkrq6|�\}}|tj kr�|dkr�|�\}}|tj k�r4|�|�|�\}}|tj k�s.|d k�r"�q4|�\}}q�q>�q6q>Wnt�yLYn0t|�S) NFcst��}|d|dfS)Nrr)�next)�tok��genrr�advancersz(_detect_future_features..advanceT�fromZ __future__�import�(�,)r�generate_tokens�io�StringIO�readline� frozensetr�NEWLINE�NL�COMMENTr �STRING�NAME�OP�add� StopIteration)�sourceZhave_docstringr?�ignore�features�tp�valuerr=r�_detect_future_featuresosB           rVc@s eZdZdS)� FixerErrorNrrrrrrW�srWc@s�eZdZdddd�ZdZdZd4dd�Zdd �Zd d �Zd d �Z dd�Z dd�Z d5dd�Z d6dd�Z dd�Zd7dd�Zdd�Zd8dd�Zdd�Zd d!�Zd9d"d#�Zd:d$d%�Zd&Zd'Zd(d)�Zd*d+�Zd,d-�Zd.d/�Zd0d1�Zd2d3�ZdS);�RefactoringToolF)�print_function� exec_function�write_unchanged_filesZFixr NcCsJ||_|p g|_|j��|_|dur0|j�|�tj��|_|jdrR|jj d=n|jdrf|jj d=|j� d�|_ g|_ t �d�|_g|_d|_tj|jtj|jd�|_|��\|_|_g|_t��|_g|_g|_t|j|j�D]H}|j r�|j�!|�q�||jv�r|j�"|�q�||jvr�|j�"|�q�t#|j�|_$t#|j�|_%dS) NrY�printrZ�execr[rXF)�convert�logger)&�fixers�explicit�_default_options�copy�optionsr!r r+�grammar�keywords�getr[�errors�loggingZ getLoggerr_� fixer_log�wroterZDriverrr^� get_fixers� pre_order� post_order�files�bmZ BottomMatcher�BMZ bmi_pre_orderZbmi_post_orderrZ BM_compatibleZ add_fixerrr1�bmi_pre_order_heads�bmi_post_order_heads)�selfZ fixer_namesrdrar0rrr�__init__�sB         �    zRefactoringTool.__init__c CsVg}g}|jD�]}t|iidg�}|�dd�d}|�|j�rR|t|j�d�}|�d�}|jd�dd�|D��}zt ||�}Wn$t y�t d ||f�d�Yn0||j |j �} | jr�|jd ur�||jvr�|�d |�q|�d |�| jd k�r|�| �q| jdk�r|�| �qt d| j��qt�d�} |j| d�|j| d�||fS)Nr r3r������_�cSsg|] }|���qSr)�title)r4r$rrrr7��z.RefactoringTool.get_fixers..zCan't find %s.%sTzSkipping optional fixer: %szAdding transformation: %sZpreZpostzIllegal fixer order: %rZ run_order��key)r`r�rsplitr� FILE_PREFIX�len�split� CLASS_PREFIX�join�getattr�AttributeErrorrWrdrjra� log_message� log_debug�orderr�operator� attrgetter�sort) rtZpre_order_fixersZpost_order_fixersZ fix_mod_path�modr5�parts� class_nameZ fix_classr0Zkey_funcrrrrl�s:    �         zRefactoringTool.get_fixerscOs�dSr&r)rt�msg�args�kwdsrrr� log_errorszRefactoringTool.log_errorcGs|r ||}|j�|�dSr&)r_�info�rtr�r�rrrr�szRefactoringTool.log_messagecGs|r ||}|j�|�dSr&)r_�debugr�rrrr� szRefactoringTool.log_debugcCsdSr&r)rt�old_text�new_text�filename�equalrrr� print_outputszRefactoringTool.print_outputcCs8|D].}tj�|�r$|�|||�q|�|||�qdSr&)�os�path�isdir� refactor_dir� refactor_file)rt�items�write� doctests_onlyZ dir_or_filerrr�refactors zRefactoringTool.refactorc Cs�tjd}t�|�D]�\}}}|�d|�|��|��|D]>}|�d�s>tj�|�d|kr>tj�||�} |� | ||�q>dd�|D�|dd�<qdS)N�pyzDescending into %sr3rcSsg|]}|�d�s|�qSr2)r)r4Zdnrrrr72rzz0RefactoringTool.refactor_dir..) r��extsep�walkr�r�rr��splitextr�r�) rtZdir_namer�r�Zpy_ext�dirpathZdirnames� filenamesr�fullnamerrrr� s   �zRefactoringTool.refactor_dirc Cs�zt|d�}Wn4tyB}z|�d||�WYd}~dSd}~00zt�|j�d}W|��n |��0tj|d|dd��}|��|fWd�S1s�0YdS)N�rbzCan't open %s: %s)NNrr#rx��encoding�newline) �open�OSErrorr�r�detect_encodingrG�closerE�read)rtr��f�errr�rrr�_read_python_source4sz#RefactoringTool._read_python_sourcecCs�|�|�\}}|durdS|d7}|rn|�d|�|�||�}|jsL||kr`|�|||||�q�|�d|�nH|�||�}|js�|r�|jr�|jt|�dd�|||d�n |�d|�dS)N� zRefactoring doctests in %szNo doctest changes in %srv)r�r�zNo changes in %s)r�r��refactor_docstringr[�processed_file�refactor_string� was_changed�str)rtr�r�r��inputr��output�treerrrr�Ds    �zRefactoringTool.refactor_filec Cs�t|�}d|vrtj|j_zfz|j�|�}WnFtyr}z.|�d||jj |�WYd}~W|j|j_dSd}~00W|j|j_n |j|j_0||_ |� d|�|� ||�|S)NrYzCan't parse %s: %s: %szRefactoring %s) rVr Z!python_grammar_no_print_statementrreZ parse_stringr"r�� __class__r�future_featuresr�� refactor_tree)rt�datarrSr�r�rrrr�[s"   �  �  zRefactoringTool.refactor_stringcCs�tj��}|rN|�d�|�|d�}|js2||krB|�|d|�q�|�d�n:|�|d�}|jsj|r~|jr~|�t |�d|�n |�d�dS)NzRefactoring doctests in stdinzzNo doctest changes in stdinzNo changes in stdin) �sys�stdinr�r�r�r[r�r�r�r�)rtr�r�r�r�rrr�refactor_stdinvs     zRefactoringTool.refactor_stdinc Cs�t|j|j�D]}|�||�q|�|j|���|�|j|���|j�|� ��}t |� ���r�|jj D�]B}||vrj||rj||j tjjdd�|jr�||j tjjd�t||�D]�}|||vr�||�|�z t|�Wnt�y�Yq�Yn0|j�r||jv�rq�|�|�}|r�|�||�}|dur�|�|�|��D] }|j�s\g|_|j�|��qJ|j�|� ��}|D]*} | |v�r�g|| <|| �|| ��q�q�qjqTt|j|j�D]}|�||��q�|jS)NT)r|�reverser{)rrmrnZ start_tree� traverse_byrrrsrq�runZleaves�anyr,r`r�rZBaseZdepthZkeep_line_orderZ get_linenor)�remover� ValueErrorZfixers_applied�match� transform�replacerr.Z finish_treer�) rtr�rr0Z match_set�node�results�newZ new_matchesZfxrrrrr��sJ        zRefactoringTool.refactor_treecCsV|sdS|D]D}||jD]4}|�|�}|r|�||�}|dur|�|�|}qq dSr&)rr�r�r�)rtr`Z traversalr�r0r�r�rrrr��s    zRefactoringTool.traverse_bycCs�|j�|�|dur.|�|�d}|dur.dS||k}|�||||�|r`|�d|�|js`dS|rv|�||||�n |�d|�dS)NrzNo changes to %szNot writing changes to %s)rorr�r�r�r[� write_file)rtr�r�r�r�r�r�rrrr��s  zRefactoringTool.processed_filec Cs�ztj|d|dd�}Wn4tyJ}z|�d||�WYd}~dSd}~00|�Rz|�|�Wn2ty�}z|�d||�WYd}~n d}~00Wd�n1s�0Y|�d|�d|_dS)N�wrxr�zCan't create %s: %szCan't write %s: %szWrote changes to %sT)rEr�r�r�r�r�rk)rtr�r�r�r��fpr�rrrr�sB zRefactoringTool.write_filez>>> z... c Csg}d}d}d}d}|jdd�D]�}|d7}|���|j�r~|durZ|�|�||||��|}|g}|�|j�} |d| �}q |dur�|�||j�s�|||j��dkr�|� |�q |dur�|�|�||||��d}d}|� |�q |du�r |�|�||||��d� |�S)NrT��keependsrr�rx) � splitlines�lstripr�PS1r.�refactor_doctest�find�PS2�rstriprr�) rtr�r��result�blockZ block_lineno�indent�lineno�line�irrrr�sD  � ��  �   �z"RefactoringTool.refactor_docstringc sz��||��}Wnhtyz}zP�j�tj�rL|D]}��d|�d��q4��d|||j j |�|WYd}~Sd}~00�� ||��rt |�j dd�}|d|d�||dd�} }|d�d�s�|dd7<��j|�d�g}|�r|��fd d �|D�7}|S) Nz Source: %sr�z+Can't parse docstring in %s line %s: %s: %sTr�rrvrcsg|]}��j|�qSr)r�)r4r��r�rtrrr7^rzz4RefactoringTool.refactor_doctest..)� parse_blockr"r_Z isEnabledForri�DEBUGr�r�r�r�rr�r�r��endswithr��pop) rtr�r�r�r�r�r�r�r�Zclippedrr�rr�Ds& �"z RefactoringTool.refactor_doctestcCs�|jr d}nd}|js$|�d|�n"|�d|�|jD]}|�|�q6|jrl|�d�|jD]}|�|�q\|jr�t|j�dkr�|�d�n|�dt|j��|jD]"\}}}|j|g|�Ri|��q�dS) N�werez need to bezNo files %s modified.zFiles that %s modified:z$Warnings/messages while refactoring:rzThere was 1 error:zThere were %d errors:)rkror�rjrhr)rtr��file�messager�r�r�rrr� summarizeas$       zRefactoringTool.summarizecCs"|j�|�|||��}t�|_|Sr&)rZ parse_tokens� wrap_toksrHr�)rtr�r�r�r�rrrr�xszRefactoringTool.parse_blockc csdt�|�||�j�}|D]F\}}\}}\} } } ||d7}| |d7} ||||f| | f| fVqdS)Nr)rrD� gen_lines�__next__) rtr�r�r�r-rrUZline0Zcol0Zline1Zcol1Z line_textrrrr��s   zRefactoringTool.wrap_toksccsx||j}||j}|}|D]N}|�|�r>|t|�d�Vn(||��dkrVdVntd||f��|}qdVqldS)Nr�zline=%r, prefix=%rrx)r�r�rrr��AssertionError)rtr�r��prefix1Zprefix2�prefixr�rrrr��s   zRefactoringTool.gen_lines)NN)FF)FF)FF)F)NFN)N)rrrrbr�r~rurlr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rrrrrX�s@� 7(   O�  + rXc@s eZdZdS)�MultiprocessingUnsupportedNrrrrrr��sr�csBeZdZ�fdd�Zd �fdd� Z�fdd�Z�fd d �Z�ZS) �MultiprocessRefactoringToolcs&tt|�j|i|��d|_d|_dSr&)�superr�ru�queue� output_lock�rtr��kwargs�r�rrru�sz$MultiprocessRefactoringTool.__init__Frc s>|dkrtt���|||�Sz ddl�Wnty>t�Yn0�jdurRtd������_�� ��_ ��fdd�t |�D�}zn|D] }|� �q�tt���|||�W�j� �t |�D]}�j�d�q�|D]}|��r�|� �q�d�_nL�j� �t |�D]}�j�d��q|D]}|���r|� ��qd�_0dS)Nrrz already doing multiple processescsg|]}�j�jd��qS))�target)ZProcess�_child)r4r���multiprocessingrtrrr7�s�z8MultiprocessRefactoringTool.refactor..)r�r�r�r�� ImportErrorr�r�� RuntimeErrorZ JoinableQueueZLockr��range�startr��putZis_alive)rtr�r�r�Z num_processesZ processesr$r�r�r�rr��sF �       � �   �    z$MultiprocessRefactoringTool.refactorcs\|j��}|durX|\}}z$tt|�j|i|��W|j��n |j��0|j��}q dSr&)r�rgr�r�r�Z task_done)rtZtaskr�r�r�rrr��s  ��z"MultiprocessRefactoringTool._childcs6|jdur|j�||f�ntt|�j|i|��SdSr&)r�rr�r�r�r�r�rrr��s  ��z)MultiprocessRefactoringTool.refactor_file)FFr)rrrrur�r�r�� __classcell__rrr�rr��s  � r�)T)"� __author__rEr�rr�rir�r'� itertoolsrZpgen2rrrZ fixer_utilrrxrr r rprr"rrr1r8r:rVrW�objectrXr�r�rrrr� s6    (