method_decorators_example2

method_decorators_example2.py

def integer_check(method):
	def inner(ref, expo=None):
		if expo == None:
			if not isinstance(ref._val1, int) or not isinstance(ref._val2, int):
				raise TypeError('Please enter numerical numbers for val1 and val2')
			else:
				return method(ref)
		if expo:
			if not isinstance(ref._val1, int) or not isinstance(ref._val2, int)\
					or not isinstance(expo, int):
				raise TypeError('Please enter numerical numbers for val1, val2 and expo')
			else:
				return method(ref, expo)

	return inner


class NumericalOps(object):
	def __init__(self, val1, val2):
		self._val1 = val1
		self._val2 = val2


	@integer_check
	def multiply_together(self):
		return self._val1 * self._val2

	@integer_check
	def power(self, exponent):
		return self.multiply_together() ** exponent


y = NumericalOps(1, 2)


print(y.multiply_together())
print(y.power(3))