c'oT1ã@sedZddlmZddlmZddlmZdZdd„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úE/usr/lib/python3/dist-packages/libpasteurize/fixes/fix_annotations.pyÚparam_without_annotations sr c@s4eZdZdZdd„ZdZdd„ZdS)ÚFixAnnotationsFcCs)|js%d|_|j|d|ƒdS)NTÚreason)ÚwarnedZwarning)Úselfrr rrrÚ warn_onces  zFixAnnotations.warn_oncezm funcdef< 'def' any parameters< '(' [params=any] ')' > ['->' ret=any] ':' any* > cCs"|jdƒ}|jdƒ}|dk ru|jjtjksKtdƒ‚|j|dtƒ|jjƒ|jƒ|dkr…dS|jt j kræx„|j D]>}|jt j kr¡|j|dtƒ|j t|ƒƒq¡Wn8|jt j kr|j|dtƒ|j t|ƒƒdS)zK This just strips annotations from the funcdef completely. ÚparamsÚretNzInvalid return annotationr )ÚgetZ prev_siblingÚtyperÚRARROWÚAssertionErrorrÚ warning_textÚremoverZ typedargslistrZtnameÚreplacer )r rZresultsrrZparamrrrÚ transforms" !   zFixAnnotations.transformN)Ú__name__Ú __module__Ú __qualname__r rZPATTERNrrrrrr s  r N) Ú__doc__Zlib2to3rZ lib2to3.pgen2rZlib2to3.fixer_utilrrr ZBaseFixr rrrrÚs