gis-7683 automate parser/render collection fixes

nazargesyk · nazargesyk · commit 9985f6b33bfc · 2024-04-30T11:44:07.000+03:00
diff --git a/uncoder-core/app/translator/managers.py b/uncoder-core/app/translator/managers.py
@@ -42,21 +42,25 @@ def get_platforms_details(self) -> list[TranslatorPlatform]:
 
 class ParserManager(Manager):
     platforms = {}
-    roota_parsers = {}
-    parsers = {}
+    supported_by_roota_platforms = {}
+    main_platforms = {}
 
-    def get_roota_parser(self, platform_id: str):  # noqa: ANN201
-        if platform := self.roota_parsers.get(platform_id):
+    def get_supported_by_roota(self, platform_id: str):  # noqa: ANN201
+        if platform := self.supported_by_roota_platforms.get(platform_id):
             return platform
         raise UnsupportedRootAParser(parser=platform_id)
 
-    def register_roota_parser(self, cls):
-        self.roota_parsers[cls.details.platform_id] = cls()
-        return super().register(cls)
+    def register_supported_by_roota(self, cls):
+        parser = cls()
+        self.supported_by_roota_platforms[cls.details.platform_id] = parser
+        self.platforms[cls.details.platform_id] = parser
+        return cls
 
-    def register_parser(self, cls):
-        self.parsers[cls.details.platform_id] = cls()
-        return super().register(cls)
+    def register_main(self, cls):
+        parser = cls()
+        self.main_platforms[cls.details.platform_id] = parser
+        self.platforms[cls.details.platform_id] = parser
+        return cls
 
     @cached_property
     def get_platforms_details(self) -> list[TranslatorPlatform]:
@@ -73,7 +77,7 @@ def get_platforms_details(self) -> list[TranslatorPlatform]:
                 alt_platform=platform.details.alt_platform,
                 first_choice=platform.details.first_choice,
             )
-            for platform in self.parsers.values()
+            for platform in self.main_platforms.values()
         ]
         return sorted(platforms, key=lambda platform: platform.group_name)
 
diff --git a/uncoder-core/app/translator/platforms/athena/parsers/athena.py b/uncoder-core/app/translator/platforms/athena/parsers/athena.py
@@ -28,7 +28,7 @@
 from app.translator.platforms.athena.tokenizer import AthenaTokenizer
 
 
-@parser_manager.register_roota_parser
+@parser_manager.register_supported_by_roota
 class AthenaQueryParser(PlatformQueryParser):
     details: PlatformDetails = athena_details
     mappings: AthenaMappings = athena_mappings
diff --git a/uncoder-core/app/translator/platforms/chronicle/parsers/chronicle.py b/uncoder-core/app/translator/platforms/chronicle/parsers/chronicle.py
@@ -26,7 +26,7 @@
 from app.translator.platforms.chronicle.tokenizer import ChronicleQueryTokenizer
 
 
-@parser_manager.register_roota_parser
+@parser_manager.register_supported_by_roota
 class ChronicleQueryParser(PlatformQueryParser):
     mappings: ChronicleMappings = chronicle_mappings
     tokenizer: ChronicleQueryTokenizer = ChronicleQueryTokenizer()
diff --git a/uncoder-core/app/translator/platforms/crowdstrike/parsers/crowdstrike.py b/uncoder-core/app/translator/platforms/crowdstrike/parsers/crowdstrike.py
@@ -23,7 +23,7 @@
 from app.translator.platforms.crowdstrike.mapping import CrowdstrikeMappings, crowdstrike_mappings
 
 
-@parser_manager.register_roota_parser
+@parser_manager.register_supported_by_roota
 class CrowdStrikeQueryParser(SplQueryParser):
     details: PlatformDetails = crowdstrike_query_details
 
diff --git a/uncoder-core/app/translator/platforms/elasticsearch/parsers/elasticsearch.py b/uncoder-core/app/translator/platforms/elasticsearch/parsers/elasticsearch.py
@@ -23,7 +23,7 @@
 from app.translator.platforms.elasticsearch.mapping import ElasticSearchMappings, elasticsearch_mappings
 
 
-@parser_manager.register_roota_parser
+@parser_manager.register_supported_by_roota
 class ElasticSearchQueryParser(LuceneQueryParser):
     details: PlatformDetails = elasticsearch_lucene_query_details
     mappings: ElasticSearchMappings = elasticsearch_mappings
diff --git a/uncoder-core/app/translator/platforms/graylog/parsers/graylog.py b/uncoder-core/app/translator/platforms/graylog/parsers/graylog.py
@@ -23,7 +23,7 @@
 from app.translator.platforms.graylog.mapping import GraylogMappings, graylog_mappings
 
 
-@parser_manager.register_roota_parser
+@parser_manager.register_supported_by_roota
 class GraylogQueryParser(LuceneQueryParser):
     details: PlatformDetails = graylog_details
     mappings: GraylogMappings = graylog_mappings
diff --git a/uncoder-core/app/translator/platforms/logscale/parsers/logscale.py b/uncoder-core/app/translator/platforms/logscale/parsers/logscale.py
@@ -28,7 +28,7 @@
 from app.translator.platforms.logscale.tokenizer import LogScaleTokenizer
 
 
-@parser_manager.register_roota_parser
+@parser_manager.register_supported_by_roota
 class LogScaleQueryParser(PlatformQueryParser):
     details: PlatformDetails = logscale_query_details
     platform_functions: LogScaleFunctions = log_scale_functions
diff --git a/uncoder-core/app/translator/platforms/microsoft/parsers/microsoft_defender.py b/uncoder-core/app/translator/platforms/microsoft/parsers/microsoft_defender.py
@@ -24,7 +24,7 @@
 from app.translator.platforms.microsoft.parsers.microsoft_sentinel import MicrosoftSentinelQueryParser
 
 
-@parser_manager.register_roota_parser
+@parser_manager.register_supported_by_roota
 class MicrosoftDefenderQueryParser(MicrosoftSentinelQueryParser):
     mappings: MicrosoftDefenderMappings = microsoft_defender_mappings
     details: PlatformDetails = microsoft_defender_details
diff --git a/uncoder-core/app/translator/platforms/microsoft/parsers/microsoft_sentinel.py b/uncoder-core/app/translator/platforms/microsoft/parsers/microsoft_sentinel.py
@@ -28,7 +28,7 @@
 from app.translator.platforms.microsoft.tokenizer import MicrosoftSentinelTokenizer
 
 
-@parser_manager.register_roota_parser
+@parser_manager.register_supported_by_roota
 class MicrosoftSentinelQueryParser(PlatformQueryParser):
     platform_functions: MicrosoftFunctions = microsoft_sentinel_functions
     mappings: MicrosoftSentinelMappings = microsoft_sentinel_mappings
diff --git a/uncoder-core/app/translator/platforms/opensearch/parsers/opensearch.py b/uncoder-core/app/translator/platforms/opensearch/parsers/opensearch.py
@@ -23,7 +23,7 @@
 from app.translator.platforms.opensearch.mapping import OpenSearchMappings, opensearch_mappings
 
 
-@parser_manager.register_roota_parser
+@parser_manager.register_supported_by_roota
 class OpenSearchQueryParser(LuceneQueryParser):
     details: PlatformDetails = opensearch_query_details
     mappings: OpenSearchMappings = opensearch_mappings
diff --git a/uncoder-core/app/translator/platforms/qradar/parsers/qradar.py b/uncoder-core/app/translator/platforms/qradar/parsers/qradar.py
@@ -29,7 +29,7 @@
 from app.translator.tools.utils import get_match_group
 
 
-@parser_manager.register_roota_parser
+@parser_manager.register_supported_by_roota
 class QradarQueryParser(PlatformQueryParser):
     details: PlatformDetails = qradar_query_details
     tokenizer = QradarTokenizer()
diff --git a/uncoder-core/app/translator/platforms/roota/parsers/roota.py b/uncoder-core/app/translator/platforms/roota/parsers/roota.py
@@ -27,7 +27,7 @@
 from app.translator.platforms.roota.const import ROOTA_RULE_DETAILS
 
 
-@parser_manager.register_parser
+@parser_manager.register_main
 class RootAParser(QueryParser, YamlRuleMixin):
     parser_manager = parser_manager
     details: PlatformDetails = PlatformDetails(**ROOTA_RULE_DETAILS)
@@ -65,7 +65,7 @@ def __parse_meta_info(self, rule: dict) -> MetaInfoContainer:
         )
 
     def __get_parser_class(self, parser: str) -> PlatformQueryParser:
-        parser_class = self.parser_manager.get_roota_parser(parser)
+        parser_class = self.parser_manager.get_supported_by_roota(parser)
         if parser_class:
             return parser_class
         raise UnsupportedRootAParser(parser=parser)
diff --git a/uncoder-core/app/translator/platforms/sigma/parsers/sigma.py b/uncoder-core/app/translator/platforms/sigma/parsers/sigma.py
@@ -32,7 +32,7 @@
 from app.translator.platforms.sigma.tokenizer import SigmaConditionTokenizer, SigmaTokenizer
 
 
-@parser_manager.register_parser
+@parser_manager.register_main
 class SigmaParser(YamlRuleMixin):
     details: PlatformDetails = PlatformDetails(**SIGMA_RULE_DETAILS)
     condition_tokenizer = SigmaConditionTokenizer()
diff --git a/uncoder-core/app/translator/platforms/splunk/parsers/splunk.py b/uncoder-core/app/translator/platforms/splunk/parsers/splunk.py
@@ -24,7 +24,7 @@
 from app.translator.platforms.splunk.mapping import SplunkMappings, splunk_mappings
 
 
-@parser_manager.register_roota_parser
+@parser_manager.register_supported_by_roota
 class SplunkQueryParser(SplQueryParser):
     details: PlatformDetails = splunk_query_details
 
