3
\" � @ 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 ZdZdZdZdZdZej 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 || _ || _| j� d S )N)�options�log�compile_pattern)�selfr r � r �*/usr/lib64/python3.6/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 �PCr r r
r
; s
zBaseFix.compile_patternc C s
|| _ d S )N)�filename)r r r r r
� set_filenameF s zBaseFix.set_filenamec C s d|i}| j j||�o|S )N�node)r �match)r r �resultsr r r
r M s z
BaseFix.matchc C s
t � �d S )N)�NotImplementedError)r r r r r r
� transformY s zBaseFix.transform�xxx_todo_changemec C s6 |}x || j kr$|tt| j�� }qW | j j|� |S )N)�
used_names�str�next�numbers�add)r �template�namer r r
�new_namei s
zBaseFix.new_namec C s. | j r d| _ | jjd| j � | jj|� d S )NFz### In file %s ###)� first_logr �appendr )r �messager r r
� log_messaget s zBaseFix.log_messagec C s> |j � }|j� }d|_d}| j|||f � |r:| j|� 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 |j � }| jd||f � d S )Nz Line %d: %s)r( r& )r r r* r+ r r r
�warning� s zBaseFix.warningc C s( |j | _ | j|� tjd�| _d| _d S )Nr T)r r � itertools�countr r# )r �treer r r r
�
start_tree� s
zBaseFix.start_treec C s d S )Nr )r r1 r 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 )N)�superr9 r2 � _should_skip)r �args)� __class__r r
r2 � s zConditionalFix.start_treec C sJ | j d k r| j S | jjd�}|d }dj|d d� �}t|||�| _ | j S )N�.r �����r? )r; �skip_on�split�joinr )r r Zpkgr! r r r
� should_skip� s
zConditionalFix.should_skip)r4 r5 r6 r@ r2 rC �
__classcell__r r )r= r
r9 � s r9 )
r/ Zpatcompr r' r Z
fixer_utilr �objectr r9 r r r r
� s