zmyme/gpuutil
1
0
Fork 0
Code Issues Pull Requests Releases 5 Wiki Activity

🐛 fix bug on empty configuration file

Browse Source
This commit is contained in:
mingyang 2020-12-22 20:17:24 +08:00
parent 9ec4d877eb
commit 742a717b49
3 changed files with 21 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
gpuutil/__main__.py
View File

@ -1,4 +1,4 @@
from gpuutil import GPUStat
from gpuutil import GPUStat, loaddict
import sys
import json
import argparse
@ -18,8 +18,7 @@ def load_config():
configpath = os.path.join(home_dir, '.gpuutil.conf')
if not os.path.isfile(configpath):
return {}
with open(configpath, 'r', encoding='utf-8') as f:
return json.load(f)
return loaddict(configpath)
def save_config(config):
home_dir = os.path.expanduser('~')
configdir = os.path.join(home_dir, '.gpuutil.conf')

27
gpuutil/gpuutil.py
View File

@ -14,6 +14,14 @@ def loadfile(path):
with open(path, 'r', encoding='utf-8') as f:
return f.read()
def loaddict(path):
content = loadfile(path)
content = content.strip()
if len(content) != 0:
return json.loads(content)
else:
return {}
def exe_cmd(command):
pipe = os.popen(command)
return pipe.read()
@ -311,16 +319,15 @@ class GPUStat():
self.load_configure()
def load_configure(self):
configuration_path = os.path.expanduser('~/.gpuutil.conf')
if os.path.exists(configuration_path):
with open(configuration_path, 'r', encoding='utf-8') as f:
configuration = json.load(f)
if 'redirect' in configuration:
if 'nvsmi_src' in configuration['redirect']:
self.nvsmi_source = configuration['redirect']['nvsmi_src']
if 'ps_src' in configuration['redirect']:
self.ps_source = configuration['redirect']['ps_src']
if 'ps_name_trans' in configuration['redirect']:
self.ps_name_trans = configuration['redirect']['ps_name_trans']
if os.path.isfile(configuration_path):
configuration = loaddict(configuration_path)
if 'redirect' in configuration:
if 'nvsmi_src' in configuration['redirect']:
self.nvsmi_source = configuration['redirect']['nvsmi_src']
if 'ps_src' in configuration['redirect']:
self.ps_source = configuration['redirect']['ps_src']
if 'ps_name_trans' in configuration['redirect']:
self.ps_name_trans = configuration['redirect']['ps_name_trans']
def get_process_info(self):

5
gpuutil/set_redirect.py
View File

@ -1,6 +1,6 @@
import argparse
import os
import json
from gpuutil import loaddict
availabel_name_trans = ['command', 'user', 'pid']
@ -32,8 +32,7 @@ if name_trans is not None:
config_file = os.path.expanduser('~/.gpuutil.conf')
configuration = {}
if os.path.isfile(config_file):
with open(config_file, 'r', encoding='utf-8') as f:
configuration = json.load(f)
configuration = loaddict(config_file)
configuration['redirect'] = {
"nvsmi_src": args.nvsmi,
"ps_src": args.ps,