Я пытаюсь обернуть все методы существующего класса (не моего создания) в пакет try/except. Это может быть любой класс, но я буду использовать здесь класс pandas.DataFrame в качестве практического примера.
Поэтому, если вызванный метод завершается успешно, мы просто идем дальше. Но если он должен генерировать исключение, оно добавляется в список для последующей проверки/обнаружения (хотя в приведенном ниже примере для простоты просто выдается оператор печати).
(Обратите внимание, что типы исключений, связанных с данными, которые могут возникнуть при вызове метода экземпляра, еще не известны; и это причина этого упражнения: обнаружение).
Этот пост был весьма полезен (особенно @martineau Python-3 ответ), но у меня возникли проблемы с его адаптацией. Ниже я ожидал, что второй вызов (обернутого) метода info() выдаст вывод на печать, но, к сожалению, этого не происходит.
#!/usr/bin/env python3
import functools, types, pandas
def method_wrapper(method):
@functools.wraps(method)
def wrapper(*args, **kwargs): #Note: args[0] points to 'self'.
try:
print('Calling: {}.{}()... '.format(args[0].__class__.__name__,
method.__name__))
return method(*args, **kwargs)
except Exception:
print('Exception: %r' % sys.exc_info()) # Something trivial.
#<Actual code would append that exception info to a list>.
return wrapper
class MetaClass(type):
def __new__(mcs, class_name, base_classes, classDict):
newClassDict = {}
for attributeName, attribute in classDict.items():
if type(attribute) == types.FunctionType: # Replace it with a
attribute = method_wrapper(attribute) # decorated version.
newClassDict[attributeName] = attribute
return type.__new__(mcs, class_name, base_classes, newClassDict)
class WrappedDataFrame2(MetaClass('WrappedDataFrame',
(pandas.DataFrame, object,), {}),
metaclass=type):
pass
print('Unwrapped pandas.DataFrame().info():')
pandas.DataFrame().info()
print('\n\nWrapped pandas.DataFrame().info():')
WrappedDataFrame2().info()
print()
Это выводит:
Unwrapped pandas.DataFrame().info():
<class 'pandas.core.frame.DataFrame'>
Index: 0 entries
Empty DataFrame
Wrapped pandas.DataFrame().info(): <-- Missing print statement after this line.
<class '__main__.WrappedDataFrame2'>
Index: 0 entries
Empty WrappedDataFrame2
В итоге,...
>>> unwrapped_object.someMethod(...)
# Should be mirrored by ...
>>> wrapping_object.someMethod(...)
# Including signature, docstring, etc. (i.e. all attributes); except that it
# executes inside a try/except suite (so I can catch exceptions generically).