U
e5d" � @ sP d dl Z ddlmZ ddlmZ ddlmZ G dd� de�ZG dd � d e�Z dS )
� N� )�PatternCompiler)�pygram)�does_tree_importc @ s� e Zd ZdZdZdZdZdZe� d�Z
e � Z dZ
dZdZdZdZdZejZdd� Zdd � Zd
d � Zd d
� Zdd� Zd dd�Zdd� Z d dd�Z dd� Z dd� Z dd � Z dS ) �BaseFixNr ZpostF� c C s || _ || _| �� d S �N)�options�log�compile_pattern)�selfr r
� r
�*/usr/lib64/python3.8/lib2to3/fixer_base.py�__init__/ s zBaseFix.__init__c C s, | j d k r(t� }|j| j dd�\| _| _d S )NT)Z with_tree)�PATTERNr r �pattern� pattern_tree)r ZPCr
r
r r ; s
�zBaseFix.compile_patternc C s
|| _ d S r )�filename)r r r
r
r � set_filenameF s zBaseFix.set_filenamec C s d|i}| j �||�o|S )N�node)r �match�r r Zresultsr
r
r r M s z
BaseFix.matchc C s
t � �d S r )�NotImplementedErrorr r
r
r � transformY s zBaseFix.transform�xxx_todo_changemec C s2 |}|| j kr"|tt| j�� }q| j �|� |S r )�
used_names�str�next�numbers�add)r �template�namer
r
r �new_namei s
zBaseFix.new_namec C s. | j r d| _ | j�d| j � | j�|� d S )NFz### In file %s ###)� first_logr
�appendr )r �messager
r
r � log_messaget s zBaseFix.log_messagec C s> |� � }|�� }d|_d}| �|||f � |r:| �|� d S )N� z Line %d: could not convert: %s)�
get_linenoZclone�prefixr& )r r �reason�linenoZ
for_output�msgr
r
r �cannot_convertz s zBaseFix.cannot_convertc C s |� � }| �d||f � d S )Nz Line %d: %s)r( r& )r r r* r+ r
r
r �warning� s zBaseFix.warningc C s( |j | _ | �|� t�d�| _d| _d S )Nr T)r r � itertools�countr r# �r Ztreer r
r
r �
start_tree� s
zBaseFix.start_treec C s d S r r
r1 r
r
r � finish_tree� s zBaseFix.finish_tree)r )N)!�__name__�
__module__� __qualname__r r r r r r/ r0 r �setr �orderZexplicitZ run_orderZ _accept_typeZkeep_line_orderZ
BM_compatibler Zpython_symbolsZsymsr r r r r r" r& r- r. r2 r3 r
r
r
r r s2
r c s( e Zd ZdZ� fdd�Zdd� Z� ZS )�ConditionalFixNc s t t| �j|� d | _d S r )�superr9 r2 � _should_skip)r �args�� __class__r
r r2 � s zConditionalFix.start_treec C sJ | j d k r| j S | j�d�}|d }d�|d d� �}t|||�| _ | j S )N�.�����)r; �skip_on�split�joinr )r r Zpkgr! r
r
r � should_skip� s
zConditionalFix.should_skip)r4 r5 r6 rA r2 rD �
__classcell__r
r
r= r r9 � s r9 )
r/ Zpatcompr r' r Z
fixer_utilr �objectr r9 r
r
r
r � s