B

�5�ab
�@sNdZd
dl
Zd
dlmZ
d
dlmZmZmZm	Z	
dd�Z
Gdd�dej�ZdS)	a�
Fixer for operator functions.

operator.isCallable(obj)       -> callable(obj)
operator.sequenceIncludes(obj) -> operator.contains(obj)
operator.isSequenceType(obj)   -> isinstance(obj, collections.abc.Sequence)
operator.isMappingType(obj)    -> isinstance(obj, collections.abc.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��0/usr/lib/python3.7/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"|�|
|�}|dk	r||
|�SdS)
N)
�
_check_method)�self�node�results�methodr
r
r�	transform+s


zFixOperator.transformzoperator.contains(%s)cCs|�|
|d
�S)N�contains)
�_handle_rename)rrrr
r
r�_sequenceIncludes0szFixOperator._sequenceIncludeszcallable(%s)cCs"|d
}tt
d�
|��g
|
jd�S)Nr�callable)
�prefix)rr�cloner)rrrrr
r
r�_isCallable4s
zFixOperator._isCallablezoperator.mul(%s)cCs|�|
|d
�S)N�mul)
r)rrrr
r
r�_repeat9szFixOperator._repeatzoperator.imul(%s)cCs|�|
|d
�S)N�imul)
r)rrrr
r
r�_irepeat=szFixOperator._irepeatz(isinstance(%s, collections.abc.Sequence)cCs|�|
|d
d�S)Nzcollections.abc�Sequence)
�_handle_type2abc)rrrr
r
r�_isSequenceTypeAszFixOperator._isSequenceTypez'isinstance(%s, collections.abc.Mapping)cCs|�|
|d
d�S)Nzcollections.abc�Mapping)
r")rrrr
r
r�_isMappingTypeEszFixOperator._isMappingTypezisinstance(%s, numbers.Number)cCs|�|
|d
d�S)NZnumbers�Number)
r")rrrr
r
r�
_isNumberTypeIszFixOperator._isNumberTypecCs|d
d}||_|�
�
dS)Nrr
)�valueZchanged)rrr�namerr
r
rrMs


zFixOperator._handle_renamecCsFtd||
�
|d
}|�
�tdd�||g�
�
g}ttd�
||
jd�S)Nrz, �
.�
isinstance)
r)rrr�joinrrr)rrr�module�abcr�argsr
r
rr"Rs



zFixOperator._handle_type2abccCs^t|d
|ddj
�}t|tjj�rZd|kr2|St|d�
f
}|j|}|�|
d|�
dS)N�
_rr
r-rzYou should use '%s' here.)	�getattrr(r+�collectionsr.�Callable�strrZwarning)rrrr�subZinvocation_strr
r
rrXs







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


r
)
�__doc__Zcollections.abcr2Zlib2to3rZlib2to3.fixer_utilrrrrrZBaseFixr
r
r
r
r�<module>
s