U
��,ab
� @ sN d Z ddlZddlmZ ddlmZmZmZm Z dd� Z
G dd� 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 � fdd�}|S )Nc s
� | _ | S �N)�
invocation)�f��s� �?/opt/alt/python38/lib64/python3.8/lib2to3/fixes/fix_operator.py�dec s zinvocation.<locals>.decr )r r r r
r
r s r c @ s� e Zd ZdZdZdZdZdeeed� Zdd� Z e
d �d
d � �Z e
d �d
d� �Z e
d�dd� �Z
e
d�dd� �Ze
d�dd� �Ze
d�dd� �Ze
d�d d � �Zd d � 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�objc C s" | � ||�}|d k r |||�S d S r )�
_check_method)�self�node�results�methodr r r
� transform+ s zFixOperator.transformzoperator.contains(%s)c C s | � ||d�S )N�contains��_handle_rename�r r r r r r
�_sequenceIncludes0 s z FixOperator._sequenceIncludesz callable(%s)c C s" |d }t td�|�� g|jd�S )Nr �callable��prefix)r r �cloner )r r r r r r r
� _isCallable4 s zFixOperator._isCallablezoperator.mul(%s)c C s | � ||d�S )N�mulr r r r r
�_repeat9 s zFixOperator._repeatzoperator.imul(%s)c C s | � ||d�S )N�imulr r r r r
�_irepeat= s zFixOperator._irepeatz(isinstance(%s, collections.abc.Sequence)c C s | � ||dd�S )N�collections.abc�Sequence��_handle_type2abcr r r r
�_isSequenceTypeA s zFixOperator._isSequenceTypez'isinstance(%s, collections.abc.Mapping)c C s | � ||dd�S )Nr& �Mappingr( r r r r
�_isMappingTypeE s zFixOperator._isMappingTypez isinstance(%s, numbers.Number)c C s | � ||dd�S )NZnumbers�Numberr( r r r r
�
_isNumberTypeI s zFixOperator._isNumberTypec C s |d d }||_ |�� d S )Nr r )�valueZchanged)r r r �namer r r r
r M s zFixOperator._handle_renamec C sF t d ||� |d }|�� tdd�||g� �g}ttd�||jd�S )Nr z, �.�
isinstancer )r r r �joinr r r )r r r �module�abcr �argsr r r
r) R s z FixOperator._handle_type2abcc C s^ t | d|d d j �}t|tjj�rZd|kr2|S t|d �f}|j| }| �|d| � d S )N�_r r r4 r zYou should use '%s' here.) �getattrr/ r2 � collectionsr5 �Callable�strr Zwarning)r r r r �subZinvocation_strr r r
r X s
zFixOperator._check_methodN)�__name__�
__module__� __qualname__Z
BM_compatible�orderr r �dictZPATTERNr r r r! r# r% r* r, r. r r) r r r r r
r s2
�
r )
�__doc__Zcollections.abcr9 Zlib2to3r Zlib2to3.fixer_utilr r r r r ZBaseFixr r r r r
�<module> s