Make MVTModule.get_slug() a classmethod (#418)

mvt-project · Nov 22, 2023 · 1d075ab · 1d075ab
1 parent 7310481
commit 1d075ab
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/mvt/common/cmd_check_iocs.py b/mvt/common/cmd_check_iocs.py
@@ -53,7 +53,7 @@ def run(self) -> None:
                 if self.module_name and iocs_module.__name__ != self.module_name:
                     continue
 
-                if iocs_module().get_slug() != name_only:
+                if iocs_module.get_slug() != name_only:
                     continue
 
                 log.info(

diff --git a/mvt/common/module.py b/mvt/common/module.py
@@ -74,12 +74,13 @@ def from_json(cls, json_path: str, log: logging.Logger):
                 log.info('Loaded %d results from "%s"', len(results), json_path)
             return cls(results=results, log=log)
 
-    def get_slug(self) -> str:
+    @classmethod
+    def get_slug(cls) -> str:
         """Use the module's class name to retrieve a slug"""
-        if self.slug:
-            return self.slug
+        if cls.slug:
+            return cls.slug
 
-        sub = re.sub("(.)([A-Z][a-z]+)", r"\1_\2", self.__class__.__name__)
+        sub = re.sub("(.)([A-Z][a-z]+)", r"\1_\2", cls.__name__)
         return re.sub("([a-z0-9])([A-Z])", r"\1_\2", sub).lower()
 
     def check_indicators(self) -> None: