�
��gu � �, � d dl mZ d dlmZ d dlZd� ZdS )� )�Basic)�pprintNc
� ���� �
� � �
��� d\ �
� d\ � �d\ }�
d\ �}d\ }}d\ }�d� � fd������ �
� � �
��f
d �� �| d
� � S ) z6Traverse a tree asking a user which branch to choose. )z[0;31mz[1;31m)z[0;32mz[1;32m)z[0;33mz[1;33m)z[0;34mz[1;34m)z[0;35mz[1;35m)z[0;36mz[1;36mz[0mc �x �� t d� t t | � � � � �z � � d S )N� )�print�join�map�str)�args�ENDs ��k/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/sympy/interactive/traversal.py�cprintz%interactive_traversal.<locals>.cprint s0 �� �
�b�g�g�c�#�t�n�n�%�%��+�,�,�,�,�,� c
�N �
� |dk rt � � �d�
|�d� � t � � � t | � � t �� � t | t � � r@| j r| � � � }nF| j r| � � � }n*| j }n"t | d� � rt | � � }n| S t |� � }|s| S t |� � D ]B\ }} ��d�
|�d� t |� � �� � t |� � t � � �C|dk rd}nd |dz
z } t d
|z � � }|d k rM ��d |z � � ��d
� � ��d� � ��d� � ��d� � �| |� � }�n |dv r| }n�|dk r �|d |dz � � }n�|dk r �|d |dz � � }n�|dk r" �t j |� � |dz � � }n� t# |� � }|dk s||k r �� d|z � � �| |� � }na �|| |dz � � }nK# t$ $ r �� d|z � � �| |� � }Y n$w xY w# t& $ r | }t � � Y nw xY w|S )Nr zCurrent expression (stage z):�__iter__�[z] � �0z0-%dz Your choice [%s,f,l,r,d,?]: �?z.%s - select subexpression with the given indexz"f - select the first subexpressionz!l - select the last subexpressionz!r - select a random subexpressionz d - done
)�dr �f�l������rzChoice must be in %s range
z$Choice must be a number in %s range
)r r �
isinstancer �is_Add�as_ordered_terms�is_Mul�as_ordered_factorsr �hasattr�list�len� enumerate�type�input�random�choice�int�
ValueError�EOFError)�expr�stager �n_args�i�arg�choicesr( �result�BCYAN�BGREEN�BLUE�BRED�BYELLOWr
�GREEN�RED�_interactive_traversalr s ����������r r: z5interactive_traversal.<locals>._interactive_traversal s{ �� � �1�9�9� �G�G�G���+�W�e�S�$�G�G�G�
�e� � � ��t� � � �
�c�
�
�
� �d�E� "� "�
��{�
!��,�,�.�.�����
!��.�.�0�0����y���
�T�:�
&�
&� ���:�:�D�D��K��T����� ��K� ��o�o� � �F�A�s� �F�5�#�v�q�%��t�T�#�Y�Y�� L� L� L� �3�K�K�K� �G�G�G�G� �Q�;�;��G�G� ���
�+�G�# Q��9�G�C�D�D�F�
��}�}���s� L� � � � � ���s� @�A�A�A���s� ?�@�@�@���s� ?�@�@�@���s�L�)�)�)�/�/��e�<�<����9�$�$� ����3� � �/�/��Q�����C�C����3� � �/�/��R��%�!�)�D�D����3� � �/�/��
�d�0C�0C�U�Q�Y�O�O��� Q� ��[�[�F� ��z�z�V�v�%5�%5� ��t�%C�g�%M�N�N�N�!7�!7��e�!D�!D���!7�!7��V� �e�a�i�!P�!P���� "� A� A� A��F�4�B�W�L�N� N� N� 3� 3�D�%� @� @�F�F�F�A�����1 � � � ��F� �G�G�G�G�G� ����F �
s$ � J �
I �%J�J�J"�!J"r � )r, �YELLOW�BBLUE�MAGENTA�BMAGENTA�CYANr3 r4 r5 r6 r7 r
r8 r9 r: r s @@@@@@@@@@r �interactive_traversalrA s� ����������� � +�I�C��.�M�E�6�0�O�F�G�,�K�D�%�2��G�X�,�K�D�%�
�C�-� -� -� -� -�I� I� I� I� I� I� I� I� I� I� I� I� I� I�V "� !�$�� *� *�*r )�sympy.core.basicr �sympy.printingr r' rA r; r r �<module>rD sP �� "� "� "� "� "� "� !� !� !� !� !� !�
�
�
�
�Y+� Y+� Y+� Y+� Y+r