Forráskód Böngészése

HUE-8936 [tracing] Convert api_error_handler with @wraps function

Romain 6 éve
szülő
commit
f3bb1c4b7d

+ 1 - 1
desktop/libs/notebook/src/notebook/api.py

@@ -175,7 +175,7 @@ def _execute_notebook(request, notebook, snippet):
 
 @require_POST
 @check_document_access_permission
-# @api_error_handler
+@api_error_handler
 def execute(request, engine=None):
   notebook = json.loads(request.POST.get('notebook', '{}'))
   snippet = json.loads(request.POST.get('snippet', '{}'))

+ 5 - 4
desktop/libs/notebook/src/notebook/decorators.py

@@ -97,12 +97,13 @@ def check_document_modify_permission():
   return inner
 
 
-def api_error_handler(func):
-  def decorator(*args, **kwargs):
+def api_error_handler(f):
+  @wraps(f)
+  def wrapper(*args, **kwargs):
     response = {}
 
     try:
-      return func(*args, **kwargs)
+      return f(*args, **kwargs)
     except SessionExpired as e:
       response['status'] = -2
     except QueryExpired as e:
@@ -154,7 +155,7 @@ def api_error_handler(func):
       if response:
         return JsonResponse(response)
 
-  return decorator
+  return wrapper
 
 def _closest_power_of_2(number):
   return math.pow(2, math.ceil(math.log(number, 2)))