3
\" @ sP d dl Z ddlmZ ddlmZ ddlmZ G dd deZG dd d eZ dS )
N )PatternCompiler)pygram)does_tree_importc @ s e Zd ZdZdZdZdZdZej dZ
e Z dZ
dZdZdZdZdZejZdd Zdd Zd
d Zd d
Zdd Zd ddZdd Z d ddZ dd Z dd Z dd Z dS ) BaseFixNr ZpostF c C s || _ || _| j d S )N)optionslogcompile_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 )Nnode)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.transformxxx_todo_changemec C s6 |}x || j kr$|tt| j }qW | j j| |S )N)
used_namesstrnextnumbersadd)r templatenamer 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_linenoZcloneprefixr& )r r reasonlinenoZ
for_outputmsgr 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 itertoolscountr 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 fddZdd 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_onsplitjoinr )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