Learning on Python Decorator

decorator_learning.py

def decorate_msg(func):

	def wrapped(*args):
		
		return "the answer is %f" % func(*args)

	return wrapped


def filename_decorate(filename):

	def new_decorate(func):
        
		def wrapped(*args):

			with open(filename, "w", newline="") as file_obj:

				msg_written = "the answer is %f" % func(*args)
				file_obj.write(msg_written)

			return "file %s written successfully" % filename
		return wrapped
	
	return new_decorate


@filename_decorate("fireball.txt")
def add_two(num):

	return num + 2


@filename_decorate("yahoo.txt")
def multiply_two(num):

	return num * 2


print(add_two(9.0))
print(multiply_two(9.0)) 


"""Reference:
https://www.thecodeship.com/patterns/guide-to-python-function-decorators/
http://ot-note.logdown.com/posts/67571/-decorator-with-without-arguments-in-function-class-form
"""