def wrapped(self, *args, **kwargs): guid = self._generate_uuid() self._spark_events.emit_magic_execution_start_event(f.__name__, get_livy_kind(self.language), guid) try: result = f(self, *args, **kwargs) except Exception as e: self._spark_events.emit_magic_execution_end_event(f.__name__, get_livy_kind(self.language), guid, False, e.__class__.__name__, str(e)) raise else: self._spark_events.emit_magic_execution_end_event(f.__name__, get_livy_kind(self.language), guid, True, u"", u"") return result
def test_get_livy_kind_covers_all_langs(): for lang in LANGS_SUPPORTED: get_livy_kind(lang)