o >h @sUdZddlmZmZddlmZmZmZmZm Z ddl m Z m Z e dZ e dZe dZde d ee d e fd d Zd ee gefdee d ee fddZdee deed efddZdee egefd eee geffddZdeegefdee gefd ee geffddZee jeeee dZeee ge dfed<dee gefdegee fd e fd d!Zdeege fd ege ffd"d#ZdS)$z1 General functional-style helpers for disttrial. )partialwraps) AwaitableCallableIterableOptionalTypeVar)Deferredsucceed_A_B_CdefaultoptionalreturncCs|dur|S|S)z Get a definite value from an optional value. @param default: The value to return if the optional value is missing. @param optional: The optional value to return if it exists. N)rrrr~/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-packages/twisted/trial/_dist/functional.py fromOptionalsr conditionxsccs$|D] }||r |VqdSdS)z :return: An iterable over C{xs} that stops when C{condition} returns ``False`` based on the value of iterated C{xs}. Nr)rrxrrr takeWhiles rabcs|IdH|IdHS)z Wait for one action to complete and then another. If either action fails, failure is propagated. If the first action fails, the second action is not waited on. Nr)rrrrrsequence+s  rfcstfdd}|S)zc Create a function like another but with the order of the first two arguments flipped. cs ||SNr)rrrrrg<s zflip..grrrrrrflip6sr!fxfycs"ttfdd}|S)z Create a function that calls one function with an argument and then another function with the result of the first function. cs |Srr)rr"r#rrrIs zcompose..gr)r"r#rrr$rcomposeCsr%N discardResult predicateactioncs |IdH}||s|Sq)z Call a function repeatedly until its result fails to satisfy a predicate. @param predicate: The check to apply. @param action: The function to call. @return: The result of C{action} which did not satisfy C{predicate}. TNr)r'r(resultrrr iterateWhileXs  r*csddtffdd }|S)z Wrap a function with another that automatically passes an integer counter of the number of calls that have gone through the wrapper. rrcs"z }Wd7|Sd7w)Nr)r)counterrrrrrs   zcountingCalls..g)r r rr,r countingCallsksr.)__doc__ functoolsrrtypingrrrrrtwisted.internet.deferr r r r r rboolrrr!r% fromCoroutiner&__annotations__r*intr.rrrrs0(  * 4   (