Your IP : 3.142.212.225
U
i�f" � @ sT d Z ddlZddlmZ ddlmZ ddlmZ G dd� de�Z G d d
� d
e �Z
dS )z2Base class for fixers (optional, but recommended).� N� )�PatternCompiler)�pygram)�does_tree_importc @ s� e Zd ZdZdZdZdZdZdZe �
d�Ze� Z
dZdZdZdZdZdZejZdd� Zd d
� Zdd� Zd
d� Zdd� Zddd�Zdd� Zddd�Zdd� Zdd� Z dd� Z!dS ) �BaseFixa Optional base class for fixers.
The subclass name must be FixFooBar where FooBar is the result of
removing underscores and capitalizing the words of the fix name.
For example, the class name for a fixer named 'has_key' should be
FixHasKey.
Nr ZpostF� c C s || _ || _| �� dS )a Initializer. Subclass may override.
Args:
options: a dict containing the options passed to RefactoringTool
that could be used to customize the fixer through the command line.
log: a list to append warnings and other messages to.
N)�options�log�compile_pattern)�selfr r � r �7/opt/alt/python38/lib64/python3.8/lib2to3/fixer_base.py�__init__/ s zBaseFix.__init__c C s, | j dk r(t� }|j| j dd�\| _| _dS )z�Compiles self.PATTERN into self.pattern.
Subclass may override if it doesn't want to use
self.{pattern,PATTERN} in .match().
NT)Z with_tree)�PATTERNr r
�pattern�pattern_tree)r ZPCr r r
r
; s
�zBaseFix.compile_patternc C s
|| _ dS )zOSet the filename.
The main refactoring tool should call this.
N)�filename)r r r r r
�set_filenameF s zBaseFix.set_filenamec C s d|i}| j �||�o|S )a Returns match for a given parse tree node.
Should return a true or false object (not necessarily a bool).
It may return a non-empty dict of matching sub-nodes as
returned by a matching pattern.
Subclass may override.
�node)r �match�r r Zresultsr r r
r M s z
BaseFix.matchc C s
t � �dS )a� Returns the transformation for a given parse tree node.
Args:
node: the root of the parse tree that matched the fixer.
results: a dict mapping symbolic names to part of the match.
Returns:
None, or a node that is a modified copy of the
argument node. The node argument may also be modified in-place to
effect the same change.
Subclass *must* override.
N)�NotImplementedErrorr r r r
� transformY s zBaseFix.transform�xxx_todo_changemec C s2 |}|| j kr"|tt| j�� }q| j �|� |S )z�Return a string suitable for use as an identifier
The new name is guaranteed not to conflict with other identifiers.
)�
used_names�str�next�numbers�add)r �template�namer r r
�new_namei s
zBaseFix.new_namec C s. | j rd| _ | 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>