Rewrite `export_to_file` as decorator

Change-Id: Ibf852038fac986801a3fc2c608a0ddb1572abbde
Signed-off-by: Yujun Zhang <zhang.yujunz@zte.com.cn>

1 files changed, 8 insertions, 3 deletions

diff --git a/qtip/util/export_to.py b/qtip/util/export_to.py
index 98c39b5f..17adae7c 100644
--- a/qtip/util/export_to.py
+++ b/qtip/util/export_to.py
@@ -10,6 +10,11 @@
 import json
 
 
-def export_to_file(content, filename):
-    with open(filename, 'w+') as f:
-        f.write(json.dumps(content, indent=2))
+def export_to_file(func):
+    def func_wrapper(spec, metrics, dest=None):
+        content = func(spec, metrics)
+        if dest is not None:
+            with open(dest, 'w+') as f:
+                f.write(json.dumps(content, indent=2))
+        return content
+    return func_wrapper