catcher.py

import inspect
import logging
import sys

def GetWrapper(obj, f):
	def wrapper(*args, **kwargs):
		try:
			ret=f(obj, *args, **kwargs)
			return ret
		except:
			(E, e, trace) = sys.exc_info()
			logging.warning('exception caught: '+str(e))
			logging.warning(str(trace))
	return wrapper

def AttachCatcher(obj, functions=None):
	cdict=obj.__class__.__dict__
	if functions==None:
		functions=cdict
	if isinstance(functions, basestring):
		functions=(functions,)
	for func in functions:
		if func.startswith('__'):
			continue
		f=cdict[func]
		if not inspect.isfunction(f):
			continue
		logging.debug("Wrapping "+func)
		wrapper=GetWrapper(obj, f)
		obj.__dict__[func]=wrapper
	return obj

if __name__=="__main__":
	class kkkk:
		def abcd(a,b):
			print "a,b:", a,b
			raise IOError
		def asdf(a,s):
			print a+s
	k=kkkk()
	kc=AttachCatcher(k, ["abcd", "asdf"])
	print(kc.__dict__)
	kc.abcd(2)
	# kc.abcd(2)
	kc.asdf("asdf")
	print ("OK")