text: def decorator(F):
      def new_F(a, b):
          print("input", a, b)
          return F(a, b)
      return new_F
  
  # get square sum
  @decorator
  def square_sum(a, b):
      return a**2 + b**2
  
  # get square diff
  @decorator
  def square_diff(a, b):
      return a**2 - b**2
  
  print(square_sum(3, 4))
  print(square_diff(3, 4))
  
  #square_sum = decorator(square_sum)
  #square_sum(3, 4)
  