cribl-control-plane 0.0.36__py3-none-any.whl → 0.0.37a1__py3-none-any.whl

cribl_control_plane/models/output.py

@@ -80,8 +80,10 @@ from .outputtcpjson import OutputTcpjson, OutputTcpjsonTypedDict
 from .outputwavefront import OutputWavefront, OutputWavefrontTypedDict
 from .outputwebhook import OutputWebhook, OutputWebhookTypedDict
 from .outputxsiam import OutputXsiam, OutputXsiamTypedDict
+from cribl_control_plane.utils import get_discriminator
+from pydantic import Discriminator, Tag
 from typing import Union
-from typing_extensions import TypeAliasType
+from typing_extensions import Annotated, TypeAliasType
 
 
 OutputTypedDict = TypeAliasType(
@@ -90,25 +92,25 @@ OutputTypedDict = TypeAliasType(
         OutputDevnullTypedDict,
         OutputDefaultTypedDict,
         OutputRouterTypedDict,
-        OutputNetflowTypedDict,
         OutputSnmpTypedDict,
+        OutputNetflowTypedDict,
         OutputDiskSpoolTypedDict,
         OutputRingTypedDict,
-        OutputStatsdExtTypedDict,
         OutputGraphiteTypedDict,
         OutputStatsdTypedDict,
+        OutputStatsdExtTypedDict,
         OutputGooglePubsubTypedDict,
         OutputCriblTCPTypedDict,
-        OutputSnsTypedDict,
         OutputSplunkTypedDict,
+        OutputSnsTypedDict,
         OutputCloudwatchTypedDict,
         OutputSyslogTypedDict,
         OutputAzureEventhubTypedDict,
         OutputWavefrontTypedDict,
         OutputSignalfxTypedDict,
         OutputHoneycombTypedDict,
-        OutputSumoLogicTypedDict,
         OutputTcpjsonTypedDict,
+        OutputSumoLogicTypedDict,
         OutputHumioHecTypedDict,
         OutputElasticCloudTypedDict,
         OutputCrowdstrikeNextGenSiemTypedDict,
@@ -122,6 +124,7 @@ OutputTypedDict = TypeAliasType(
         OutputSqsTypedDict,
         OutputNewrelicTypedDict,
         OutputCriblHTTPTypedDict,
+        OutputGrafanaCloudTypedDict,
         OutputXsiamTypedDict,
         OutputFilesystemTypedDict,
         OutputDatasetTypedDict,
@@ -130,10 +133,10 @@ OutputTypedDict = TypeAliasType(
         OutputServiceNowTypedDict,
         OutputLokiTypedDict,
         OutputDynatraceOtlpTypedDict,
-        OutputGoogleChronicleTypedDict,
         OutputElasticTypedDict,
-        OutputDatadogTypedDict,
+        OutputGoogleChronicleTypedDict,
         OutputCriblLakeTypedDict,
+        OutputDatadogTypedDict,
         OutputPrometheusTypedDict,
         OutputMskTypedDict,
         OutputSentinelOneAiSiemTypedDict,
@@ -150,77 +153,76 @@ OutputTypedDict = TypeAliasType(
         OutputAzureDataExplorerTypedDict,
         OutputWebhookTypedDict,
         OutputGoogleCloudLoggingTypedDict,
-        OutputGrafanaCloudTypedDict,
     ],
 )
 
 
-Output = TypeAliasType(
-    "Output",
+Output = Annotated[
     Union[
-        OutputDevnull,
-        OutputDefault,
-        OutputRouter,
-        OutputNetflow,
-        OutputSnmp,
-        OutputDiskSpool,
-        OutputRing,
-        OutputStatsdExt,
-        OutputGraphite,
-        OutputStatsd,
-        OutputGooglePubsub,
-        OutputCriblTCP,
-        OutputSns,
-        OutputSplunk,
-        OutputCloudwatch,
-        OutputSyslog,
-        OutputAzureEventhub,
-        OutputWavefront,
-        OutputSignalfx,
-        OutputHoneycomb,
-        OutputSumoLogic,
-        OutputTcpjson,
-        OutputHumioHec,
-        OutputElasticCloud,
-        OutputCrowdstrikeNextGenSiem,
-        OutputKinesis,
-        OutputConfluentCloud,
-        OutputKafka,
-        OutputExabeam,
-        OutputNewrelicEvents,
-        OutputAzureLogs,
-        OutputSplunkLb,
-        OutputSqs,
-        OutputNewrelic,
-        OutputCriblHTTP,
-        OutputXsiam,
-        OutputFilesystem,
-        OutputDataset,
-        OutputSplunkHec,
-        OutputDynatraceHTTP,
-        OutputServiceNow,
-        OutputLoki,
-        OutputDynatraceOtlp,
-        OutputGoogleChronicle,
-        OutputElastic,
-        OutputDatadog,
-        OutputCriblLake,
-        OutputPrometheus,
-        OutputMsk,
-        OutputSentinelOneAiSiem,
-        OutputSentinel,
-        OutputInfluxdb,
-        OutputGoogleCloudStorage,
-        OutputAzureBlob,
-        OutputOpenTelemetry,
-        OutputMinio,
-        OutputClickHouse,
-        OutputSecurityLake,
-        OutputDlS3,
-        OutputS3,
-        OutputAzureDataExplorer,
-        OutputWebhook,
-        OutputGoogleCloudLogging,
-        OutputGrafanaCloud,
+        Annotated[OutputDefault, Tag("default")],
+        Annotated[OutputWebhook, Tag("webhook")],
+        Annotated[OutputSentinel, Tag("sentinel")],
+        Annotated[OutputDevnull, Tag("devnull")],
+        Annotated[OutputSyslog, Tag("syslog")],
+        Annotated[OutputSplunk, Tag("splunk")],
+        Annotated[OutputSplunkLb, Tag("splunk_lb")],
+        Annotated[OutputSplunkHec, Tag("splunk_hec")],
+        Annotated[OutputTcpjson, Tag("tcpjson")],
+        Annotated[OutputWavefront, Tag("wavefront")],
+        Annotated[OutputSignalfx, Tag("signalfx")],
+        Annotated[OutputFilesystem, Tag("filesystem")],
+        Annotated[OutputS3, Tag("s3")],
+        Annotated[OutputAzureBlob, Tag("azure_blob")],
+        Annotated[OutputAzureDataExplorer, Tag("azure_data_explorer")],
+        Annotated[OutputAzureLogs, Tag("azure_logs")],
+        Annotated[OutputKinesis, Tag("kinesis")],
+        Annotated[OutputHoneycomb, Tag("honeycomb")],
+        Annotated[OutputAzureEventhub, Tag("azure_eventhub")],
+        Annotated[OutputGoogleChronicle, Tag("google_chronicle")],
+        Annotated[OutputGoogleCloudStorage, Tag("google_cloud_storage")],
+        Annotated[OutputGoogleCloudLogging, Tag("google_cloud_logging")],
+        Annotated[OutputGooglePubsub, Tag("google_pubsub")],
+        Annotated[OutputExabeam, Tag("exabeam")],
+        Annotated[OutputKafka, Tag("kafka")],
+        Annotated[OutputConfluentCloud, Tag("confluent_cloud")],
+        Annotated[OutputMsk, Tag("msk")],
+        Annotated[OutputElastic, Tag("elastic")],
+        Annotated[OutputElasticCloud, Tag("elastic_cloud")],
+        Annotated[OutputNewrelic, Tag("newrelic")],
+        Annotated[OutputNewrelicEvents, Tag("newrelic_events")],
+        Annotated[OutputInfluxdb, Tag("influxdb")],
+        Annotated[OutputCloudwatch, Tag("cloudwatch")],
+        Annotated[OutputMinio, Tag("minio")],
+        Annotated[OutputStatsd, Tag("statsd")],
+        Annotated[OutputStatsdExt, Tag("statsd_ext")],
+        Annotated[OutputGraphite, Tag("graphite")],
+        Annotated[OutputRouter, Tag("router")],
+        Annotated[OutputSns, Tag("sns")],
+        Annotated[OutputSqs, Tag("sqs")],
+        Annotated[OutputSnmp, Tag("snmp")],
+        Annotated[OutputSumoLogic, Tag("sumo_logic")],
+        Annotated[OutputDatadog, Tag("datadog")],
+        Annotated[OutputGrafanaCloud, Tag("grafana_cloud")],
+        Annotated[OutputLoki, Tag("loki")],
+        Annotated[OutputPrometheus, Tag("prometheus")],
+        Annotated[OutputRing, Tag("ring")],
+        Annotated[OutputOpenTelemetry, Tag("open_telemetry")],
+        Annotated[OutputServiceNow, Tag("service_now")],
+        Annotated[OutputDataset, Tag("dataset")],
+        Annotated[OutputCriblTCP, Tag("cribl_tcp")],
+        Annotated[OutputCriblHTTP, Tag("cribl_http")],
+        Annotated[OutputHumioHec, Tag("humio_hec")],
+        Annotated[OutputCrowdstrikeNextGenSiem, Tag("crowdstrike_next_gen_siem")],
+        Annotated[OutputDlS3, Tag("dl_s3")],
+        Annotated[OutputSecurityLake, Tag("security_lake")],
+        Annotated[OutputCriblLake, Tag("cribl_lake")],
+        Annotated[OutputDiskSpool, Tag("disk_spool")],
+        Annotated[OutputClickHouse, Tag("click_house")],
+        Annotated[OutputXsiam, Tag("xsiam")],
+        Annotated[OutputNetflow, Tag("netflow")],
+        Annotated[OutputDynatraceHTTP, Tag("dynatrace_http")],
+        Annotated[OutputDynatraceOtlp, Tag("dynatrace_otlp")],
+        Annotated[OutputSentinelOneAiSiem, Tag("sentinel_one_ai_siem")],
     ],
-)
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]