3


 \�
�@sNdZd
dl
Z
d
dlmZ
d
dlmZmZmZmZ
dd�Z	Gdd�dej
�ZdS)	a�
Fixer for operator functions.

operator.isCallable(obj)       -> hasattr(obj, '__call__')
operator.sequenceIncludes(obj) -> operator.contains(obj)
operator.isSequenceType(obj)   -> isinstance(obj, collections.Sequence)
operator.isMappingType(obj)    -> isinstance(obj, collections.Mapping)
operator.isNumberType(obj)     -> isinstance(obj, numbers.Number)
operator.repeat(obj, n)        -> operator.mul(obj, n)
operator.irepeat(obj, n)       -> operator.imul(obj, n)
�N)
�
fixer_base)�Call�Name�String�touch_importc
s�f
d
d�}
|
S)Nc

s
�|_|S)
N)
�
invocation)
�
f)
�
s��2/usr/lib64/python3.6/lib2to3/fixes/fix_operator.py�decs

zinvocation.<locals>.decr
)r	rr
)
r	rrs
rc@s�eZ
dZd
ZdZdZdZdeeed�Zdd�Z	e
d	�
d
d��
Ze
d�
d
d��
Ze
d�
dd��
Z
e
d�
dd��
Ze
d�
dd��
Ze
d�
dd��
Ze
d�
dd��
Zdd�Zd d!�Zd"d#�Zd$S)%�FixOperatorTZprez�
              method=('isCallable'|'sequenceIncludes'
                     |'isSequenceType'|'isMappingType'|'isNumberType'
                     |'repeat'|'irepeat')
              z'(' obj=any ')'z�
              power< module='operator'
                trailer< '.' %(methods)s > trailer< %(obj)s > >
              |
              power< %(methods)s trailer< %(obj)s > >
              )�methods�objcCs"|j|
|�}|dk	r||
|�SdS)
N)
�
_check_method)�self�node�results�methodr
r
r�	transform+s


zFixOperator.transformzoperator.contains(%s)cCs|j|
|d
�S)N�contains)
�_handle_rename)rrrr
r
r�_sequenceIncludes0szFixOperator._sequenceIncludeszhasattr(%s, '__call__')cCs2|d
}|j�t
d�
t
d�
g}ttd�
||
jd�S)Nrz, z
'__call__'�hasattr)
�prefix)�clonerrrr)rrrr�argsr
r
r�_isCallable4s

zFixOperator._isCallablezoperator.mul(%s)cCs|j|
|d
�S)N�mul)
r)rrrr
r
r�_repeat:szFixOperator._repeatzoperator.imul(%s)cCs|j|
|d
�S)N�imul)
r)rrrr
r
r�_irepeat>szFixOperator._irepeatz$isinstance(%s, collections.Sequence)cCs|j|
|d
d�S)N�collections�Sequence)
�_handle_type2abc)rrrr
r
r�_isSequenceTypeBszFixOperator._isSequenceTypez#isinstance(%s, collections.Mapping)cCs|j|
|d
d�S)Nr"�Mapping)
r$)rrrr
r
r�_isMappingTypeFszFixOperator._isMappingTypezisinstance(%s, numbers.Number)cCs|j|
|d
d�S)NZnumbers�Number)
r$)rrrr
r
r�
_isNumberTypeJszFixOperator._isNumberTypecCs|d
d}||_|j
�
dS)Nrr
)�valueZchanged)rrr�namerr
r
rrNs


zFixOperator._handle_renamecCsFtd||
�
|d
}|j
�tddj||g�
�
g}ttd�
||
jd�S)Nrz, �
.�
isinstance)
r)rrr�joinrrr)rrr�module�abcrrr
r
rr$Ss



zFixOperator._handle_type2abccCs\t|d
|ddj
�}t|tj�rXd|kr0|St|d�
f
}|j|}|j|
d|�
dS)N�
_rr
r/rzYou should use '%s' here.)�getattrr*r-r"�Callable�strrZwarning)rrrr�subZinvocation_strr
r
rrYs







zFixOperator._check_methodN)�__name__�
__module__�__qualname__Z
BM_compatible�orderrr�dictZPATTERNrrrrrr!r%r'r)rr$rr
r
r
rr
s 


r
)�__doc__r"Zlib2to3rZlib2to3.fixer_utilrrrrrZBaseFixr
r
r
r
r�<module>
s