U Zʹju @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)ENDZ/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/interactive/traversal.pycprintsz%interactive_traversal..cprintc sR|dkrt d|dtt|tt|trh|jrP|}q|jr`|}q|j}nt |dr|t |}n|St |}|s|St |D]2\}} d|dt | t|tq|dkrd}n d |d}ztd |}Wn"tk r|}tYn6X|d krf d | d  d d d||}n|dkrv|}n|dkr|d|d}n|dkr|d|d}n|dkr҈t||d}n|z t|}Wn.tk r  d|||}YnBX|dks"||kr< d|||}n|||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 isinstancerZis_AddZas_ordered_termsZis_MulZas_ordered_factorsr hasattrlistlen enumeratetypeinputEOFErrorrandomchoiceint ValueError) exprZstager Zn_argsiargchoicesr!result ZBCYANZBGREENZBLUEZBREDZBYELLOWr ZGREENZRED_interactive_traversalr r r r*sv                  z5interactive_traversal.._interactive_traversalrr )r$ZYELLOWZBBLUEZMAGENTAZBMAGENTAZCYANr r)r interactive_traversals Kr+)Zsympy.core.basicrZsympy.printingrr r+r r r r s