B �A�[-�@sJdZddlmZddlmZddlmZdZdd�ZGdd �d ej �Z d S)z& Fixer to remove function annotations �)� fixer_base)�token)�symsz)Removing function annotations completely.cCs |jdS)Nr)�children)�node�r�T/opt/alt/python37/lib/python3.7/site-packages/libpasteurize/fixes/fix_annotations.py�param_without_annotationssr c@s$eZdZdZdd�ZdZdd�ZdS)�FixAnnotationsFcCs|jsd|_|j||d�dS)NT)�reason)�warnedZwarning)�selfrrrrr� warn_onceszFixAnnotations.warn_oncezm funcdef< 'def' any parameters< '(' [params=any] ')' > ['->' ret=any] ':' any* > cCs�|�d�}|�d�}|dk rR|jjtjks2td��|j|td�|j��|��|dkr^dS|jt j kr�x`|jD],}|jt jkrr|j|td�|� t|��qrWn(|jt jkr�|j|td�|� t|��dS)zK This just strips annotations from the funcdef completely. �params�retNzInvalid return annotation)r)�getZprev_sibling�typer�RARROW�AssertionErrorr�warning_text�removerZ typedargslistrZtname�replacer )r rZresultsrrZparamrrr� transforms" zFixAnnotations.transformN)�__name__� __module__�__qualname__rrZPATTERNrrrrrr sr N)�__doc__Zlib2to3rZ lib2to3.pgen2rZlib2to3.fixer_utilrrr ZBaseFixr rrrr�s