From 8a01a2b87b7f1d2739af0535dd770aac73e43404 Mon Sep 17 00:00:00 2001 From: Sam Washko Date: Tue, 23 Jan 2024 13:51:47 -0800 Subject: [PATCH] is_compatible --- modelscan/modelscan.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/modelscan/modelscan.py b/modelscan/modelscan.py index 9100784..d39f98f 100644 --- a/modelscan/modelscan.py +++ b/modelscan/modelscan.py @@ -174,6 +174,20 @@ def _generate_results(self) -> Dict[str, Any]: return report + def is_compatible(self, path: str) -> bool: + # Determines whether a single file path is compatible with any of the available scanners + compatible: bool = False + + for scanner_path, scanner_settings in self._settings["scanners"].items(): + if ( + "supported_extensions" in scanner_settings.keys() + and Path(path).suffix + in self._settings["scanners"][scanner_path]["supported_extensions"] + ): + compatible = True + + return compatible + @property def issues(self) -> Issues: return self._issues