remove files that are no longer used anymore. Add logic to serialize …

…StrEnum and IntEnum when writing YMLs from objects that contain them.
splunk · Dec 6, 2024 · 0237ccd · 0237ccd
1 parent ed958cc
commit 0237ccd
Show file tree

Hide file tree

Showing 6 changed files with 15 additions and 197 deletions.
diff --git a/contentctl/actions/new_content.py b/contentctl/actions/new_content.py
@@ -4,7 +4,6 @@
 import questionary
 from typing import Any
 from contentctl.input.new_content_questions import NewContentQuestions
-from contentctl.output.new_content_yml_output import NewContentYmlOutput
 from contentctl.objects.config import new, NewContentType
 import uuid
 from datetime import datetime

diff --git a/contentctl/output/detection_writer.py b/contentctl/output/detection_writer.py
diff --git a/contentctl/output/new_content_yml_output.py b/contentctl/output/new_content_yml_output.py
diff --git a/contentctl/output/yml_output.py b/contentctl/output/yml_output.py
diff --git a/contentctl/output/yml_writer.py b/contentctl/output/yml_writer.py
@@ -1,6 +1,21 @@
 
 import yaml
 from typing import Any
+from enum import StrEnum, IntEnum
+
+# Set the following so that we can write StrEnum and IntEnum
+# to files. Otherwise, we will get the following errors when trying
+# to write to files:
+# yaml.representer.RepresenterError: ('cannot represent an object',.....
+yaml.SafeDumper.add_multi_representer(
+    StrEnum,
+    yaml.representer.SafeRepresenter.represent_str
+)
+
+yaml.SafeDumper.add_multi_representer(
+    IntEnum,
+    yaml.representer.SafeRepresenter.represent_int
+)
 
 class YmlWriter:
 

diff --git a/enums_test.py b/enums_test.py