Python Simple Counter (decorator; explicit function call)

simplecounter.py

# import inspect
from __future__ import print_function

class Counter(object):
	"""Simple counter, counting an invocation my means of a string (explicit call) or function name (decorator)"""

	counters = {}
	def __init__(self, *args, **kwargs):
		"""Construct class. First, check if class is called via a decorator or directly. If true, defer incrementing counter until wrapped function is actually called (see __call__). If false, increment counter now."""
		if callable(args[0]):
			self.func = args[0]
			self.name = args[0].__name__
		else:
			name = args[0]

			self.potentiallyIncrementByName(name)

	def __call__(self, *args, **kwargs):
		"""Function for when Counter is used as a decorator. In that case, this class method call with increment the counter."""
		self.potentiallyIncrementByName(self.name)
		return self.func(*args, **kwargs)

	@classmethod
	def potentiallyIncrementByName(self, name):
		"""Either the Counter is incremented (if it exists already) or created"""
		# frame = inspect.getframeinfo(inspect.stack()[2][0])
		if name in Counter.counters:
			Counter.counters[name] += 1
		else:
			Counter.counters[name] = 1
		# print(frame.function)
		# print(frame.lineno)

	@classmethod
	def print(self):
		"""Convenience method to print counted methods."""
		padding = len(max(Counter.counters.keys()))
		for (name, count) in Counter.counters.items():
			print("{name:>{pad}}: {count}".format(name=name, pad=padding, count=count))

@Counter
def testfunction():
	return 1

def example():
	Counter("Start")
	Counter("Start")
	Counter("End")

	testfunction()
	testfunction()
	testfunction()

	Counter.print()

if __name__ == '__main__':
	example()