o jgu @s,ddlmZddlmZddlZddZdS))Basic)pprintNc shd\d\d\}d\}d\}}d\}dfdd   f d d |d S) z6Traverse a tree asking a user which branch to choose. )zz)zz)zz)zz)zz)zzzcstdtt|dS)N)printjoinmapstr)args)ENDS/var/www/html/zoom/venv/lib/python3.10/site-packages/sympy/interactive/traversal.pycprintsz%interactive_traversal..cprintc sP|dkrt d|dtt|tt|tr4|jr(|}n|jr0|}n|j}n t |dr>t |}n|St |}|sH|St |D]\}} d|dt | t|tqL|dkrmd}nd |d}ztd |}Wnty|}tY|Sw|d kr d | d  d d d||}|S|dvr|}|S|dkrȈ|d|d}|S|dkr׈|d|d}|S|dkrt||d}|Szt|}Wnty d|||}Y|Sw|dks||kr d|||}|S|||d}|S)NrzCurrent expression (stage z):__iter__[z] 0z0-%dzYour 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 )drflrz$Choice must be a number in %s range zChoice must be in %s range )rr isinstanceris_Addas_ordered_termsis_Mulas_ordered_factorsr hasattrlistlen enumeratetypeinputEOFErrorrandomchoiceint ValueError) exprstager n_argsiargchoicesr%result BCYANBGREENBLUEBREDBYELLOWr GREENRED_interactive_traversalr r r r7s       !        z5interactive_traversal.._interactive_traversalrr )r(YELLOWBBLUEMAGENTABMAGENTACYANr r/r interactive_traversals  Kr=)sympy.core.basicrsympy.printingrr$r=r r r r s