Skip to content

Commit 2ff9c6e

Browse files
committed
addressed reviews
1 parent 7b5708d commit 2ff9c6e

File tree

3 files changed

+11
-11
lines changed

3 files changed

+11
-11
lines changed

nodescraper/cli/cli.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -426,8 +426,9 @@ def main(arg_input: Optional[list[str]] = None):
426426
)
427427

428428
if parsed_args.skip_sudo:
429-
for pcfg in plugin_config_inst_list:
430-
pcfg.global_args.setdefault("collection_args", {})["skip_sudo"] = True
429+
plugin_config_inst_list[-1].global_args.setdefault("collection_args", {})[
430+
"skip_sudo"
431+
] = True
431432

432433
log_system_info(log_path, system_info, logger)
433434
except Exception as e:

nodescraper/plugins/inband/dimm/dimm_collector.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -46,11 +46,6 @@ def collect_data(
4646
if args is None:
4747
args = DimmCollectorArgs()
4848

49-
if args.skip_sudo:
50-
self.result.message = "Skipping sudo plugin"
51-
self.result.status = ExecutionStatus.NOT_RAN
52-
return self.result, None
53-
5449
dimm_str = None
5550
if self.system_info.os_family == OSFamily.WINDOWS:
5651
res = self._run_sut_cmd("wmic memorychip get Capacity")
@@ -70,6 +65,10 @@ def collect_data(
7065
for capacity, count in capacities.items():
7166
dimm_str += f"{count} x {capacity / 1024 / 1024:.2f}GB "
7267
else:
68+
if args.skip_sudo:
69+
self.result.message = "Skipping sudo plugin"
70+
self.result.status = ExecutionStatus.NOT_RAN
71+
return self.result, None
7372
res = self._run_sut_cmd(
7473
"""sh -c 'dmidecode -t 17 | tr -s " " | grep -v "Volatile\\|None\\|Module" | grep Size' 2>/dev/null""",
7574
sudo=True,

nodescraper/plugins/inband/storage/storage_collector.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -46,10 +46,6 @@ def collect_data(
4646
if args is None:
4747
args = StorageCollectorArgs()
4848

49-
if args.skip_sudo:
50-
self.result.message = "Skipping sudo plugin"
51-
self.result.status = ExecutionStatus.NOT_RAN
52-
return self.result, None
5349
storage_data = {}
5450
if self.system_info.os_family == OSFamily.WINDOWS:
5551
res = self._run_sut_cmd(
@@ -66,6 +62,10 @@ def collect_data(
6662
percent=round((int(size) - int(free_space)) / int(size) * 100, 2),
6763
)
6864
else:
65+
if args.skip_sudo:
66+
self.result.message = "Skipping sudo plugin"
67+
self.result.status = ExecutionStatus.NOT_RAN
68+
return self.result, None
6969
res = self._run_sut_cmd("""sh -c 'df -lH -B1 | grep -v 'boot''""", sudo=True)
7070
if res.exit_code == 0:
7171
for line in res.stdout.splitlines()[1:]:

0 commit comments

Comments
 (0)