U Ùf&ã@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 úC/tmp/pip-target-lpfmz8o1/lib/python/logomaker/src/error_handling.pyÚ__init__szLogomakerError.__init__cCs|jSrr)r r r r Ú__str__szLogomakerError.__str__N)Ú__name__Ú __module__Ú __qualname__Ú__doc__r r r r r r rs rc@seZdZdZdd„ZdS)Ú DebugResultz0 Container class for debugging results. cCs||_||_dSr)ÚresultÚmistake)r rrr r r r szDebugResult.__init__N)rrrrr r r r r rsrcCs|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)Ú conditionrr r r Ú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. c sò| dd¡}t|dkd|ƒd}d}z:ˆ||Ž}|dkrHtdƒd}n|dkr^tdƒd}nWn^tk r¾}z@|dkrŽtd| ¡ƒd}n |dkrªtd | ¡ƒd}n|‚W5d}~XYnX|dkrÌ|Sˆjd krä||d _dSt||ƒSdS) NÚ should_fail)TFNz+FATAL: should_fail = %s is not bool or NoneTzUNEXPECTED SUCCESS.FzExpected success.zExpected error:zUNEXPECTED ERROR:r r)ÚpoprÚprintrr rrr)ÚargsÚkwargsrrrÚe©Úfuncr r Ú wrapped_funcSsB ÿ ÿÿ  z#handle_errors..wrapped_funcr)rrr rr Ú handle_errors;sJr N) Ú __future__rÚ functoolsrÚ Exceptionrrrr r r r r Ús