← Back
Editing: normalize_return.cpython-311.pyc
� cR�ax � �J � d Z ddlmZmZ ddlmZ ddlZ G d� de� � ZdS )z7 NormalizeReturn adds return statement where relevant. � )�CFG� YieldPoints)�TransformationNc �. � � e Zd ZdZ� fd�Zd� Zd� Z� xZS )�NormalizeReturna� Adds Return statement when they are implicit, and adds the None return value when not set >>> import gast as ast >>> from pythran import passmanager, backend >>> node = ast.parse("def foo(y): print(y)") >>> pm = passmanager.PassManager("test") >>> _, node = pm.apply(NormalizeReturn, node) >>> print(pm.dump(backend.Python, node)) def foo(y): print(y) return builtins.None c �b �� t t | � � � t � � d S )N)�superr �__init__r )�self� __class__s ��J/usr/lib/python3/dist-packages/pythran/transformations/normalize_return.pyr zNormalizeReturn.__init__ s' �� � �o�t�$�$�-�-�c�2�2�2�2�2� c � � | � t |� � | _ |j D ]}| � |� � �| j � t j � � D ]�}t |t j t j f� � s�d| _ | j r-|j � t j d � � � � nyt j t j dt j � � d d � � dt j � � � � }|j � t j |� � � � n��|S )NT�builtins�None)�gatherr � yield_points�body�visit�cfg� predecessorsr �NIL� isinstance�ast�Return�Raise�update�append� Attribute�Name�Load)r �node�stmt�n�nones r �visit_FunctionDefz!NormalizeReturn.visit_FunctionDef s � � �K�K� �T� :� :����I� � �D� �J�J�t� � � � � ��&�&�s�w�/�/� � �A� �a�#�*�c�i�!8�9�9� � "�� ��$� 7��I�$�$�S�Z��%5�%5�6�6�6�6� �=����S�X�Z�Z��t�D�D� ��� � � $� $�D� �I�$�$�S�Z��%5�%5�6�6�6��� � � r c �� � |j sb| j s[t j t j dt j � � d d � � dt j � � � � }||_ d| _ |S )Nr r T)�valuer r r r r! r )r r" r% s r � visit_Returnz NormalizeReturn.visit_Return2 s] � ��z� �$�"3� ��=���*�c�h�j�j�$��!M�!M�!'�����5� 5�D� �D�J� �D�K�� r )�__name__� __module__� __qualname__�__doc__r r& r) � __classcell__)r s @r r r s` �� � � � � � � � 3� 3� 3� 3� 3�� � �,� � � � � � r r ) r- �pythran.analysesr r �pythran.passmanagerr �gastr r � r r �<module>r3 su �� =� =� -� -� -� -� -� -� -� -� .� .� .� .� .� .� � � � �/� /� /� /� /�n� /� /� /� /� /r
Save File
Cancel