← Back
Editing: locals_analysis.cpython-311.pyc
� cR�ag � �F � d Z ddlmZ ddlmZ ddlZ G d� de� � ZdS )z' Locals computes the value of locals() � )�ModuleAnalysisNc � � � e Zd ZdZ� fd�Z� fd�Zd� Zd� Zd� Zd� Z d� Z d � Z d � Z d � Z eZeZeZeZeZeZeZeZeZeZeZeZeZ� xZS ) �Localsa� Statically compute the value of locals() before each statement Yields a dictionary binding every node to the set of variable names defined *before* this node. Following snippet illustrates its behavior: >>> import gast as ast >>> from pythran import passmanager >>> pm = passmanager.PassManager('test') >>> code = ''' ... def b(n): ... m = n + 1 ... def b(n): ... return n + 1 ... return b(m)''' >>> tree = ast.parse(code) >>> l = pm.gather(Locals, tree) >>> sorted(l[tree.body[0].body[0]]) ['n'] >>> sorted(l[tree.body[0].body[1]]) ['b', 'm', 'n'] c � �� t � � | _ t � � | _ d| _ t t | � � � � � d S )Nr )�dict�result�set�locals�nesting�superr �__init__)�self� __class__s ��B/usr/lib/python3/dist-packages/pythran/analyses/locals_analysis.pyr zLocals.__init__$ sB �� ��f�f�� ��e�e�� ��� � �f�d���$�$�&�&�&�&�&� c � �� t t | � � � |� � || j vr | j | j | j |< d S d S �N)r r � generic_visitr � expr_parent)r �noder s �r r zLocals.generic_visit* sR �� � �f�d���)�)�$�/�/�/� �t�{� "� "� $� �D�,<� =�D�K�� � � � #� "r c � � || _ | j � � � | j |< | � |� � d S r )r r �copyr r �r r s r �store_and_visitzLocals.store_and_visit/ s> � � ��� �K� ,� ,� .� .�� �D�� ���4� � � � � r c �\ � || _ | j | j |< | � |� � d S r )r r r r r s r � visit_ModulezLocals.visit_Module4 s2 � � ��� �K�� �D�� ���4� � � � � r c � � | j r | j � |j � � | xj dz c_ || _ | j � � � | j |< | j � � � }|j j D ]}| � |� � �|j j D ]#}|j r| � |j � � �$|j r| � |j � � | j � d� |j j D � � � � |j D ]}| � |� � �|| _ | xj dz c_ d S )N� c 3 �$ K � | ] }|j V � � d S r )�id)�.0�args r � <genexpr>z+Locals.visit_FunctionDef.<locals>.<genexpr>H s$ � � � �<�<�c�3�6�<�<�<�<�<�<r )r r �add�namer r r �args�defaults�visit� annotation�returns�update�body)r r � parent_locals�defaultr" �stmts r �visit_FunctionDefzLocals.visit_FunctionDef9 sJ � � �<� '� �K�O�O�D�I� &� &� &� � � ��� � � ��� �K� ,� ,� .� .�� �D�� � �(�(�*�*� ��y�)� � �G� �J�J�w� � � � ��9�>� +� +�C��~� +�� � �3�>�*�*�*�� �<� %� �J�J�t�|� $� $� $� � ���<�<�T�Y�^�<�<�<�<�<�<��I� � �D� �J�J�t� � � � �#�� � � � ��� � � � r c �D � || _ | j � � � | j |< t j | |� � | � |j � � | j � d� |j D � � � � |j D ]}| � |� � �d S )Nc 3 �X K � | ]%}t |t j � � � |j V � �&d S r )� isinstance�ast�Namer )r! �ts r r# z&Locals.visit_Assign.<locals>.<genexpr>S sH � � � � 7� 7�A� (��C�H� 5� 5�7�1�4� 7� 7� 7� 7� 7� 7r ) r r r r �mdr( �valuer+ �targets)r r �targets r � visit_AssignzLocals.visit_AssignN s� � � ��� �K� ,� ,� .� .�� �D�� ���t� � � � � � �4�:� � � � � ��� 7� 7�� � 7� 7� 7� 7� 7� 7��l� � �F� �J�J�v� � � � � � r c �x � || _ | j � � � | j |< t j | |� � | � |j � � | j � |j j � � |j D ]}| � |� � �|j D ]}| � |� � �d S r ) r r r r r7 r( �iterr$ r: r r, �orelse�r r r/ s r � visit_ForzLocals.visit_ForX s� � � ��� �K� ,� ,� .� .�� �D�� ���t� � � � � � �4�9� � � � � ���� ��'�'�'��I� � �D� �J�J�t� � � � ��K� � �D� �J�J�t� � � � � � r c � � | j � � � | j |< | j � d� |j D � � � � d S )Nc 3 �$ K � | ] }|j V � � d S r �r% �r! �aliass r r# z&Locals.visit_Import.<locals>.<genexpr>e �$ � � � �>�>�%�5�:�>�>�>�>�>�>r �r r r r+ �namesr s r � visit_ImportzLocals.visit_Importc �I � � �K� ,� ,� .� .�� �D�� � ���>�>�4�:�>�>�>�>�>�>�>�>r c � � | j � � � | j |< | j � d� |j D � � � � d S )Nc 3 �$ K � | ] }|j V � � d S r rC rD s r r# z*Locals.visit_ImportFrom.<locals>.<genexpr>i rF r rG r s r �visit_ImportFromzLocals.visit_ImportFromg rJ r c �, � || _ | j � � � | j |< |j r$| j � |j j � � |j o| � |j � � |j D ]}| � |� � �d S r ) r r r r r% r$ r �typer( r, r? s r �visit_ExceptHandlerzLocals.visit_ExceptHandlerk s� � � ��� �K� ,� ,� .� .�� �D�� �9� *� �K�O�O�D�I�L� )� )� )� � �+�d�j�j���+�+���I� � �D� �J�J�t� � � � � � r ) �__name__� __module__� __qualname__�__doc__r r r r r0 r; r@ rI rM rP � visit_Return� visit_Yield� visit_Try�visit_AugAssign� visit_Print� visit_While�visit_If� visit_Raise� visit_Assert� visit_Expr� visit_Pass� visit_Break�visit_Continue� __classcell__)r s @r r r s �� � � � � �� �0'� '� '� '� '� >� >� >� >� >� !� !� !� !� !� !� � � �* � � � � � �?� ?� ?�?� ?� ?� � � � #�L�!�K� �I�%�O�!�K�!�K� �H�!�K�"�L� �J� �J�!�K�$�N�N�N�N�Nr r ) rT �pythran.passmanagerr �pythran.metadata�metadatar7 �gastr4 r � r r �<module>rh s ��� � /� .� .� .� .� .� � � � � � � � � � �v%� v%� v%� v%� v%�^� v%� v%� v%� v%� v%r
Save File
Cancel