← Back
Editing: utils.cpython-311.pyc
� cR�a� � � � d Z ddlZddlmZ ddlmZmZ ddlm Z ddl m Z d� Z d� Z d � Zd � Zd � Zd � Zd � Zd� Zd� Zd� Zd� Zd� Zd� Zd� Ze d� � � ZdS )z+ Common function use for AST manipulation. � N��MODULES)�mangle�demangle)�reduce)�contextmanagerc �j � t | t j � � ot | j t � � S �N)� isinstance�ast�Constant�value�str��nodes �//usr/lib/python3/dist-packages/pythran/utils.py�isstrr s% � � �d�C�L� )� )� I�j���S�.I�.I�I� c �x � t | t j � � o t | j t t f� � S r )r r r r �int�boolr s r � isintegralr s; � � �d�C�L� )� )� M�j���c�FJ�FL� /M� /M� Mr c � � t | t j � � o&t | j t t t f� � S r )r r r r r �floatr r s r �isnumr s> � � �d�C�L� )� )� M�j���c�FK�FJ�FL� /M� /M� Mr c �v � t | t j � � sdS t d� | j D � � � � S )NFc 3 �J K � | ] }t |t j � � V � � d S r )r r �Slice)�.0�elts r � <genexpr>z isextslice.<locals>.<genexpr> s. � � � �?�?�c�z�#�s�y�)�)�?�?�?�?�?�?r )r r �Tuple�any�eltsr s r � isextslicer% s: � � �d�C�I� &� &� ��u� �?�?�T�Y�?�?�?� ?� ?�?r c � � t | j t j � � sdS t | j � � sdS | j j dk S )NFr )r �opr �Powr �rightr r s r �ispowir* sD � � �d�g�s�w� '� '� ��u� �d�j� !� !� ��u� �:� �q� � r c � �� �fd�� �t | � � \ }}|� � � s|dd� d|d fz }|d|z fS )z5 Compute path and final object for an attribute node c �& �� t |t j � � r/| t |j � � t |j � � ffS t |t j � � r, �| |j � � \ }}||j ||j fz fS dS )z9 Get function path and intrinsic from an ast.Attribute. N)r r �Namer �id� Attributer �attr)�modulesr0 �module�path�get_intrinsic_paths �r r4 z(attr_to_path.<locals>.get_intrinsic_path* s� �� � �d�C�H� %� %� :��8�D�G�,�,�-����0A�0A�/C�C� C� ��c�m� ,� ,� :�-�-�g�t�z�B�B�L�F�D��$�)�$�d�d�i�\�&9�9� 9� :� :r N����functor)�pythonic)r � isliteral)r �objr3 r4 s @r � attr_to_pathr: ' so �� �:� :� :� :� :� #�"�7�D�1�1�I�C�� �=�=�?�?� 1��C�R�C�y�I�t�B�x�0�0�� ���%� %�%r c � � t d� | dd� t j t | d � � t j � � dd� � � � S )aO Transform path to ast.Attribute. >>> import gast as ast >>> path = ('builtins', 'my', 'constant') >>> value = path_to_attr(path) >>> ref = ast.Attribute( ... value=ast.Attribute(value=ast.Name(id="builtins", ... ctx=ast.Load(), ... annotation=None, ... type_comment=None), ... attr="my", ctx=ast.Load()), ... attr="constant", ctx=ast.Load()) >>> ast.dump(ref) == ast.dump(value) True c �P � t j | |t j � � � � S r )r r/ �Load)�hpath�lasts r �<lambda>z path_to_attr.<locals>.<lambda>H s � �c�m�E�4�����&L�&L� r � Nr )r r r- r r= �r3 s r � path_to_attrrC 7 sL � �" �L�L��q�r�r�(�C�H�V�D��G�_�_�c�h�j�j�$�� M� M� O� O� Or c � � t | � � dk rt | d S t | dd� � � | d S )z� Retrieve a symbol in MODULES based on its path >>> path = ('math', 'pi') >>> path_to_node(path) #doctest: +ELLIPSIS <pythran.intrinsic.ConstantIntr object at 0x...> rA r Nr5 )�lenr � path_to_noderB s r rF rF L sA � � �4�y�y�A�~�~��t�A�w� � ��D��"��I�&�&�t�B�x�0�0r c �j � t | t j � � ot | j dd � � dk S )Nr0 �getattr)r r �CallrH �funcr s r �isattrrK Y s1 � � �t�S�X� &� &� :� �D�I�v�t� ,� ,� � 9�;r c �V � d}t | t j t j f� � s J |� � �t | t j � � st | � � rMt | � � r| j d } n| j } t | t j � � �>t | � � �M| S )a� Return modified variable name. >>> import gast as ast >>> ref = ast.Subscript( ... value=ast.Subscript( ... value=ast.Name('a', ast.Load(), None, None), ... slice=ast.Name('i', ast.Load(), None, None), ... ctx=ast.Load()), ... slice=ast.Name('j', ast.Load(), None, None), ... ctx=ast.Load()) >>> ast.dump(get_variable(ref)) "Name(id='a', ctx=Load(), annotation=None, type_comment=None)" z(Only name and subscript can be assigned.r )r r r- � SubscriptrK �argsr )� assignable�msgs r � get_variablerQ ^ s� � � 5�C� �j�3�8�S�]�";� <� <�A�A�c�A�A�A� �Z�� � /� /� *�6�*�3E�3E� *� �*� � � *�#� ��+�J�J�#�)�J� �Z�� � /� /� *�6�*�3E�3E� *� �r c �4 � t d d | S �N�builtins�pythranr ��names r �pythran_builtinrX w s � � �:� �y� )�$� /�/r c �: � | t d d v sJ �dd| fS rS r rV s r �pythran_builtin_pathrZ { s* � � �7�:�&�y�1� 1� 1� 1� 1� � �4� (�(r c �: � t t | � � � � S r )rC rZ rV s r �pythran_builtin_attrr\ � s � � �,�T�2�2� 3� 3�3r c �" � ddl m} | d| |v z z S )Nr )� cxx_keywords�_)�pythran.tablesr^ )rW r^ s r �cxxidra � s) � �+�+�+�+�+�+� �#�� �-�.� .�.r c # �d K � | � |� � d V � | � � � d S r )�append�pop)�l�vs r �pushpoprg � s/ � � � ��H�H�Q�K�K�K� �E�E�E��E�E�G�G�G�G�Gr )�__doc__�gastr r` r �pythran.conversionr r � functoolsr � contextlibr r r r r% r* r: rC rF rK rQ rX rZ r\ ra rg � r r �<module>rn sk �� 1� 1� � � � � "� "� "� "� "� "� /� /� /� /� /� /� /� /� � � � � � � %� %� %� %� %� %�J� J� J�M� M� M� M� M� M� @� @� @� !� !� !� &� &� &� O� O� O�* 1� 1� 1�;� ;� ;� � � �20� 0� 0�)� )� )� 4� 4� 4�/� /� /� � � � � � � � r
Save File
Cancel