ST110 - StrictAssetsModelingRuleSchema

.changelog/4519.yml

@@ -0,0 +1,4 @@
+changes:
+- description: Added 'AssetsModelingRuleSchema' to the `ST110` validation.
+  type: feature
+pr_number: 4519

demisto_sdk/commands/content_graph/parsers/modeling_rule.py

@@ -2,14 +2,22 @@
 from pathlib import Path
 from typing import List, Optional, Set
 
+import pydantic
+
 from demisto_sdk.commands.common.constants import MarketplaceVersions
-from demisto_sdk.commands.common.tools import get_value
+from demisto_sdk.commands.common.tools import get_file, get_value
 from demisto_sdk.commands.content_graph.common import ContentType
 from demisto_sdk.commands.content_graph.parsers.yaml_content_item import (
     YAMLContentItemParser,
 )
-from demisto_sdk.commands.content_graph.strict_objects.modeling_rule import (
-    StrictModelingRule,
+from demisto_sdk.commands.content_graph.strict_objects.assets_modeling_rule import (
+    StrictAssetsModelingRule,
+)
+from demisto_sdk.commands.content_graph.strict_objects.assets_modeling_rule_schema import (
+    StrictAssetsModelingRuleSchema,
+)
+from demisto_sdk.commands.content_graph.strict_objects.base_strict_model import (
+    StructureError,
 )
 
 
@@ -43,4 +51,26 @@ def supported_marketplaces(self) -> Set[MarketplaceVersions]:
 
     @property
     def strict_object(self):
-        return StrictModelingRule
+        raise NotImplementedError("This object has a different behavior")
+
+    def validate_structure(self) -> List[StructureError]:
+        """
+        This method uses the parsed data and attempts to build a Pydantic (strict) object from it.
+        Whenever the data and schema mismatch, we store the error using the 'structure_errors' attribute,
+        which will be read during the ST110 validation run.
+        In ModelingRule, we need to check two files: the schema json and the yml, so we override the
+        method for combing all the pydantic errors from the both files.
+        """
+        directory_pydantic_error = []
+        directory = self.path if self.path.is_dir() else self.path.parent
+        for file in directory.iterdir():
+            try:
+                if file.suffix == ".yml":
+                    StrictAssetsModelingRule.parse_obj(get_file(file))
+                elif file.suffix == ".json":
+                    StrictAssetsModelingRuleSchema.parse_obj(get_file(file))
+            except pydantic.error_wrappers.ValidationError as e:
+                directory_pydantic_error += [
+                    StructureError(path=file, **error) for error in e.errors()
+                ]
+        return directory_pydantic_error

demisto_sdk/commands/content_graph/strict_objects/assets_modeling_rule_schema.py

@@ -0,0 +1,16 @@
+from typing import Dict, Literal, Optional
+
+from pydantic import Field, constr
+
+from demisto_sdk.commands.content_graph.strict_objects.common import BaseStrictModel
+
+
+class AssetType(BaseStrictModel):
+    type: Literal["string", "int", "float", "datetime", "boolean"] = Field(
+        description="Type of the asset"
+    )
+    is_array: bool = Field(description="Whether the asset is an array")
+
+
+class StrictAssetsModelingRuleSchema(BaseStrictModel):
+    __root__: Optional[Dict[constr(regex=r".+"), Dict[str, AssetType]]] = None  # type:ignore[valid-type]