ó jÜxec@ s_ddlmZddlmZdefd„ƒYZdd d„ƒYZd„Zd„Zd S( i˙˙˙˙(tdivision(twrapstLogomakerErrorcB s eZdZd„Zd„ZRS(s Class used by Logomaker to handle errors. parameters ---------- message: (str) The message passed to check(). This only gets passed to the LogomakerError constructor when the condition passed to check() is False. cC s ||_dS(N(tmessage(tselfR((su/oak/stanford/groups/akundaje/marinovg/programs/Anaconda2/lib/python2.7/site-packages/logomaker/src/error_handling.pyt__init__scC s|jS(N(R(R((su/oak/stanford/groups/akundaje/marinovg/programs/Anaconda2/lib/python2.7/site-packages/logomaker/src/error_handling.pyt__str__s(t__name__t __module__t__doc__RR(((su/oak/stanford/groups/akundaje/marinovg/programs/Anaconda2/lib/python2.7/site-packages/logomaker/src/error_handling.pyRs  t DebugResultcB seZdZd„ZRS(s0 Container class for debugging results. cC s||_||_dS(N(tresulttmistake(RR R ((su/oak/stanford/groups/akundaje/marinovg/programs/Anaconda2/lib/python2.7/site-packages/logomaker/src/error_handling.pyRs (RRR R(((su/oak/stanford/groups/akundaje/marinovg/programs/Anaconda2/lib/python2.7/site-packages/logomaker/src/error_handling.pyR scC s|st|ƒ‚ndS(sp Checks a condition; raises a LogomakerError with message if condition evaluates to False parameters ---------- condition: (bool) A condition that, if false, halts Logomaker execution and raises a clean error to user message: (str) The string to show user if condition is False. returns ------- None N(R(t conditionR((su/oak/stanford/groups/akundaje/marinovg/programs/Anaconda2/lib/python2.7/site-packages/logomaker/src/error_handling.pytcheck"sc stˆƒ‡fd†ƒ}|S(să Decorator function to handle Logomaker errors. This decorator allows the user to pass the keyword argument 'should_fail' to any wrapped function. If should_fail is None (or is not set by user), the function executes normally, and can be called as result = func(*args, **kwargs) In particular, Python execution will halt if any errors are raised. However, if the user specifies should_fail=True or should_fail=False, then Python will not halt even in the presence of an error. Moreover, the function will return a tuple, e.g., result, mistake = func(*args, should_fail=True, **kwargs) with mistake flagging whether or not the function failed or succeeded as expected. c s.|jddƒ}t|ttdfkd|ƒd}d}yGˆ||Ž}|tkrjdGHt}n|tkr„dGHt}nWnetk rě}|tkr˝d|jƒfGHt}qí|tkrăd|jƒfGHt}qí|‚nX|dkrý|Sˆjdkr||d_dSt ||ƒSdS( Nt should_fails+FATAL: should_fail = %s is not bool or NonesUNEXPECTED SUCCESS.sExpected success.sExpected error:sUNEXPECTED ERROR:Ri( tpoptNoneRtTruetFalseRRRR R (targstkwargsRR R te(tfunc(su/oak/stanford/groups/akundaje/marinovg/programs/Anaconda2/lib/python2.7/site-packages/logomaker/src/error_handling.pyt wrapped_funcSs<            (R(RR((Rsu/oak/stanford/groups/akundaje/marinovg/programs/Anaconda2/lib/python2.7/site-packages/logomaker/src/error_handling.pyt handle_errors;sKN(( t __future__Rt functoolsRt ExceptionRR RR(((su/oak/stanford/groups/akundaje/marinovg/programs/Anaconda2/lib/python2.7/site-packages/logomaker/src/error_handling.pyts