3 •m¯\&ã@sJddlmZddlmZGdd„deƒZGdd„dƒZdd„Zd d „Zd S) é)Údivision)Úwrapsc@s eZdZdZdd„Zdd„ZdS)ÚLogomakerErrora 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. cCs ||_dS)N)Úmessage)Úselfr©rú%../../logomaker/src/error_handling.pyÚ__init__szLogomakerError.__init__cCs|jS)N)r)rrrrÚ__str__szLogomakerError.__str__N)Ú__name__Ú __module__Ú __qualname__Ú__doc__r r rrrrrs rc@seZdZdZdd„ZdS)Ú DebugResultz0 Container class for debugging results. cCs||_||_dS)N)ÚresultÚmistake)rrrrrrr szDebugResult.__init__N)r r r rr rrrrrsrcCs|s t|ƒ‚dS)ap 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)Ú conditionrrrrÚcheck"srcstˆƒ‡fdd„ƒ}|S)aã 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. csò|jddƒ}t|d kd|ƒd}d}y:ˆ||Ž}|dkrHtdƒd}n|dkr^tdƒd}nWn^tk r¾}zB|dkrŽtd|jƒƒd}n |dkrªtd|jƒƒd}n|‚WYdd}~XnX|dkrÌ|Sˆjd krä||d _dSt||ƒSdS) NÚ should_failTFz+FATAL: should_fail = %s is not bool or NonezUNEXPECTED SUCCESS.zExpected success.zExpected error:zUNEXPECTED ERROR:r r)TFN)ÚpoprÚprintrr r rr)ÚargsÚkwargsrrrÚe)ÚfuncrrÚ wrapped_funcSs<       z#handle_errors..wrapped_func)r)rrr)rrÚ handle_errors;sKrN) Ú __future__rÚ functoolsrÚ ExceptionrrrrrrrrÚs