-
Notifications
You must be signed in to change notification settings - Fork 1
/
errors.py
35 lines (31 loc) · 1009 Bytes
/
errors.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
import functools
from flask import jsonify
import traceback
def parse_to_user_and_log(logger):
def actual_decorator(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
if logger is not None:
try:
return func(*args, **kwargs)
except Exception as e:
data = {'error': traceback.format_exc()}
logger.exception(e)
return jsonify(data)
else:
return func(*args, **kwargs)
return wrapper
return actual_decorator
def log(logger):
def actual_decorator(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
if logger is not None:
try:
return func(*args, **kwargs)
except Exception as e:
logger.exception(e)
else:
func(*args, **kwargs)
return wrapper
return actual_decorator