🐛 fix bug on empty configuration file

gpuutil/__main__.py

@@ -1,6 +1,5 @@
-from gpuutil import GPUStat, loaddict
+from gpuutil import GPUStat, loaddict, savedict
 import sys
-import json
 import argparse
 import os
 
@@ -22,8 +21,7 @@ def load_config():
 def save_config(config):
     home_dir = os.path.expanduser('~')
     configdir = os.path.join(home_dir, '.gpuutil.conf')
-    with open(configdir, 'w+', encoding='utf-8') as f:
+    savedict(configdir, config)
-        json.dump(config, f, ensure_ascii=False, indent=4)
 
 # style format: |c|l:15|r|c:14rl:13|
 def parse_style(style):

gpuutil/gpuutil.py

@@ -13,6 +13,9 @@ osname = platform.system()
 def loadfile(path):
     with open(path, 'r', encoding='utf-8') as f:
         return f.read()
+def savefile(path, content):
+    with open(path, 'w+', encoding='utf-8') as f:
+        return f.write(content)
 
 def loaddict(path):
     content = loadfile(path)
@@ -21,6 +24,9 @@ def loaddict(path):
         return json.loads(content)
     else:
         return {}
+def savedict(path, dictionary):
+    content = json.dumps(dictionary, indent=4, ensure_ascii=False)
+    savefile(path, content)
 
 def exe_cmd(command):
     pipe = os.popen(command)

gpuutil/set_redirect.py

@@ -1,6 +1,6 @@
 import argparse
 import os
-from gpuutil import loaddict
+from gpuutil import loaddict, savedict
 
 availabel_name_trans = ['command', 'user', 'pid']
 
@@ -39,5 +39,4 @@ configuration['redirect'] = {
     "ps_name_trans": parsed_name_trans
 }
 
-with open(config_file, 'w+', encoding='utf-8') as f:
+savedict(config_file, configuration)
-    f.write(json.dumps(configuration, ensure_ascii=False, indent=4))