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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
from __future__ import unicode_literals
from builtins import str
from datetime import datetime
from functools import wraps
from snips_inference_agl.common.utils import json_debug_string
class DifferedLoggingMessage(object):
def __init__(self, fn, *args, **kwargs):
self.fn = fn
self.args = args
self.kwargs = kwargs
def __str__(self):
return str(self.fn(*self.args, **self.kwargs))
def log_elapsed_time(logger, level, output_msg=None):
if output_msg is None:
output_msg = "Elapsed time ->:\n{elapsed_time}"
def get_wrapper(fn):
@wraps(fn)
def wrapped(*args, **kwargs):
start = datetime.now()
msg_fmt = dict()
res = fn(*args, **kwargs)
if "elapsed_time" in output_msg:
msg_fmt["elapsed_time"] = datetime.now() - start
logger.log(level, output_msg.format(**msg_fmt))
return res
return wrapped
return get_wrapper
def log_result(logger, level, output_msg=None):
if output_msg is None:
output_msg = "Result ->:\n{result}"
def get_wrapper(fn):
@wraps(fn)
def wrapped(*args, **kwargs):
msg_fmt = dict()
res = fn(*args, **kwargs)
if "result" in output_msg:
try:
res_debug_string = json_debug_string(res)
except TypeError:
res_debug_string = str(res)
msg_fmt["result"] = res_debug_string
logger.log(level, output_msg.format(**msg_fmt))
return res
return wrapped
return get_wrapper
|
