remove files that are no longer used anymore. Add logic to serialize StrEnum and IntEnum when writing YMLs from objects that contain them.

Author: pyth0n1c

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

contentctl/output/detection_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: