← Back
Editing: remove_lambdas.cpython-311.pyc
� cR�a� � � � d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl m Z ddl m Z ddl Z G d� d ej � � Z G d � d e� � ZdS ) z5 RemoveLambdas turns lambda into regular functions. � )�GlobalDeclarations� ImportedIds)�Transformation)�MODULES)�mangleN)�copyc �* � � e Zd Z� fd�Zd� Zd� Z� xZS )�_LambdaRemoverc �r �� t t | � � � � � || _ || _ d S �N)�superr �__init__�prefix�parent)�selfr r � __class__s ��H/usr/lib/python3/dist-packages/pythran/transformations/remove_lambdas.pyr z_LambdaRemover.__init__ s1 �� � �n�d�#�#�,�,�.�.�.� �� � �� � � � c �, � t | j |� � S r )�getattrr )r �attrs r � __getattr__z_LambdaRemover.__getattr__ s � ��t�{�D�)�)�)r c � � t d | j � � � vrst j t j dt d� � � � g� � }| j � |� � t d }|| j t d� � < | � |� � d� | j t | j � � � � }| � t |� � }|� | j � � d� t# |� � D � � }d� t# |� � D � � |j j z |j _ t j |t) |j � � t j |j � � gg d d � � }t/ j |t/ j � � � � | j � |� � || j |< t j |t j � � d d � � }|rrt j t j t j t d� � t j � � d d � � dt j � � � � |g|z g � � S |S )N� functoolsz {0}_lambda{1}c �^ � g | ]*}t j |t j � � d d � � ��+S r )�ast�Name�Load��.0�iins r � <listcomp>z/_LambdaRemover.visit_Lambda.<locals>.<listcomp>' s> � � .� .� .� � �x��S�X�Z�Z��t�<�<� .� .� .r c �^ � g | ]*}t j |t j � � d d � � ��+S r )r r �Paramr s r r" z/_LambdaRemover.visit_Lambda.<locals>.<listcomp>) s> � � 2� 2� 2� "� �8�C�����d�D�A�A� 2� 2� 2r �partial) r �global_declarations�valuesr �Import�aliasr �imports�append� generic_visit�formatr �len�lambda_functions�gatherr �difference_update�sorted�args� FunctionDefr �Return�body�metadata�add�Localr r �Call� Attribute) r �node�import_�functools_module� forged_name�ii� binded_args� forged_fdef� proxy_calls r � visit_Lambdaz_LambdaRemover.visit_Lambda sA � � �;� �t�'?�'F�'F�'H�'H� H� H��j�#�)�K�� �9L�9L�"M�"M�!N�O�O�G� �L� � �� (� (� (� &�{� 3� �<L�D� $�V�K�%8�%8� 9� ���4� � � �%�,�,� �K� ��%� &� &�(� (� � �[�[��d� +� +�� � � �T� 2�3�3�3�.� .�"(��*�*�.� .� .� �2� 2�&,�R�j�j�2� 2� 2� �)�.�)�� �� �o� � ���O�O� �Z�� � "� "� #� ��d� � � � � �[�(�.�"2�"2�3�3�3� � �$�$�[�1�1�1�0;�� ��-��X�k�3�8�:�:�t�T�B�B� � � ��8�� ��H�V�K� 0� 0�#�(�*�*�d�D�I�I� ��H�J�J�� � � �{�*��� � � � r )�__name__� __module__� __qualname__r r rD � __classcell__�r s @r r r sV �� � � � � � � � � � � *� *� *�' � ' � ' � ' � ' � ' � ' r r c �. � � e Zd ZdZ� fd�Zd� Zd� Z� xZS )� RemoveLambdasa� Turns lambda into top-level functions. >>> import gast as ast >>> from pythran import passmanager, backend >>> node = ast.parse("def foo(y): lambda x:y+x") >>> pm = passmanager.PassManager("test") >>> _, node = pm.apply(RemoveLambdas, node) >>> print(pm.dump(backend.Python, node)) import functools as __pythran_import_functools def foo(y): __pythran_import_functools.partial(foo_lambda0, y) def foo_lambda0(y, x): return (y + x) c �b �� t t | � � � t � � d S r )r rK r r )r r s �r r zRemoveLambdas.__init__T s( �� � �m�T�"�"�+�+�,>�?�?�?�?�?r c � � t � � | _ t � � | _ | � |� � | j |j z | j z |_ | xj t | j � � pt | j � � z c_ |S r )�listr/ r* r, r6 �update�bool)r r< s r � visit_ModulezRemoveLambdas.visit_ModuleW st � � $���� ��v�v�� � ���4� � � ��L�4�9�,�t�/D�D�� � � � �t�D�L�)�)�H�T�$�2G�-H�-H�H� � �� r c �b �� t | |j � � ��fd�|j D � � |_ |S )Nc �: �� g | ]}�� |� � ��S � )�visit)r �n�lrs �r r" z3RemoveLambdas.visit_FunctionDef.<locals>.<listcomp>a s# �� �4�4�4�Q�R�X�X�a�[�[�4�4�4r )r �namer6 )r r< rW s @r �visit_FunctionDefz RemoveLambdas.visit_FunctionDef_ s6 �� � �D�$�)� ,� ,��4�4�4�4�$�)�4�4�4�� �� r )rE rF rG �__doc__r rQ rY rH rI s @r rK rK B se �� � � � � �� � @� @� @� @� @�� � �� � � � � � r rK )rZ �pythran.analysesr r �pythran.passmanagerr �pythran.tablesr �pythran.conversionr �pythran.metadatar7 r �gastr �NodeTransformerr rK rT r r �<module>rb s� �� ;� ;� <� <� <� <� <� <� <� <� .� .� .� .� .� .� "� "� "� "� "� "� %� %� %� %� %� %� #� #� #� #� #� #� � � � � � � � � � �1 � 1 � 1 � 1 � 1 �S�(� 1 � 1 � 1 �h � � � � �N� � � � � r
Save File
Cancel