Check trace for a function execution wrapper decorator

Print Trace.py

# Decorators 
import sys, traceback
import time

# Create a function with function argument
def timerit(func):
  #Create a wrapper to that function with args
  def wrapper(*arg):
    # function preporcessing 
    t=time.time()
    #Call function with args 
    res = func(*arg)
    #Function post processing 
    print(time.time()-t)
    # return the function response 
    return res
  # Return wrapper 
  return wrapper

# Wrap the function on the wrapper 
@timerit
def myfunc(cat):
  for lin in traceback.format_stack():
    if "<ipython-input-" in lin: 
      print(lin)
  return cat

# Calling this will first call 
#   timerit --> Wrapper -- > myfunc --> return myfunc --> return wratter -- return timer it
myfunc('jod')
#traceback.format_stack()