Skip to content

Commit 4da51e0

Browse files
committed
addressed reviewzs
1 parent 31e245c commit 4da51e0

File tree

2 files changed

+10
-11
lines changed

2 files changed

+10
-11
lines changed

nodescraper/plugins/inband/kernel_module/analyzer_args.py

Lines changed: 3 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,6 @@
2424
#
2525
###############################################################################
2626
import re
27-
from typing import ClassVar
2827

2928
from nodescraper.models import AnalyzerArgs
3029
from nodescraper.plugins.inband.kernel_module.kernel_module_data import (
@@ -34,11 +33,7 @@
3433

3534
class KernelModuleAnalyzerArgs(AnalyzerArgs):
3635
kernel_modules: dict[str, dict] = {}
37-
38-
DEFAULT_REGEX_FILTER: ClassVar[list[str]] = ["amd"]
39-
DEFAULT_REGEX_MATCH: ClassVar[bool] = True
40-
regex_filter: list[str] = DEFAULT_REGEX_FILTER
41-
regex_match: bool = DEFAULT_REGEX_MATCH
36+
regex_filter: list[str] = ["amd"]
4237

4338
@classmethod
4439
def build_from_model(cls, datamodel: KernelModuleDataModel) -> "KernelModuleAnalyzerArgs":
@@ -51,7 +46,7 @@ def build_from_model(cls, datamodel: KernelModuleDataModel) -> "KernelModuleAnal
5146
KernelModuleAnalyzerArgs: instance of analyzer args class
5247
"""
5348

54-
pattern_regex = re.compile("|".join(cls.DEFAULT_REGEX_FILTER), re.IGNORECASE)
49+
pattern_regex = re.compile("amd", re.IGNORECASE)
5550
filtered_mods = {
5651
name: data
5752
for name, data in datamodel.kernel_modules.items()
@@ -60,6 +55,5 @@ def build_from_model(cls, datamodel: KernelModuleDataModel) -> "KernelModuleAnal
6055

6156
return cls(
6257
kernel_modules=filtered_mods,
63-
regex_filter=cls.DEFAULT_REGEX_FILTER,
64-
regex_match=cls.DEFAULT_REGEX_MATCH,
58+
regex_filter=[],
6559
)

nodescraper/plugins/inband/kernel_module/kernel_module_analyzer.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -130,10 +130,15 @@ def analyze_data(
130130
"""
131131
if not args:
132132
args = KernelModuleAnalyzerArgs()
133+
else:
134+
if args.regex_filter and args.kernel_modules:
135+
self.logger.warning(
136+
"Both regex_filter and kernel_modules provided in analyzer args. kernel_modules will be ignored"
137+
)
133138

134139
self.result.status = ExecutionStatus.OK
135140

136-
if args.regex_match:
141+
if args.regex_filter:
137142
try:
138143
filtered_modules, unmatched_pattern = self.filter_modules_by_pattern(
139144
data.kernel_modules, args.regex_filter
@@ -197,6 +202,6 @@ def analyze_data(
197202
self.result.status = ExecutionStatus.ERROR
198203
return self.result
199204
else:
200-
self.result.message = "Kernel modules and regex_match failed"
205+
self.result.message = "Kernel modules and regex_filter failed"
201206
self.result.status = ExecutionStatus.ERROR
202207
return self.result

0 commit comments

Comments
 (0)