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

cribl_control_plane/models/input.py

@@ -65,58 +65,62 @@ from .inputwindowsmetrics import InputWindowsMetrics, InputWindowsMetricsTypedDi
 from .inputwineventlogs import InputWinEventLogs, InputWinEventLogsTypedDict
 from .inputwiz import InputWiz, InputWizTypedDict
 from .inputzscalerhec import InputZscalerHec, InputZscalerHecTypedDict
+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
 
 
 InputTypedDict = TypeAliasType(
     "InputTypedDict",
     Union[
-        InputDatagenTypedDict,
         InputKubeEventsTypedDict,
+        InputDatagenTypedDict,
         InputCriblTypedDict,
         InputCriblmetricsTypedDict,
         InputKubeMetricsTypedDict,
         InputCollectionTypedDict,
         InputSystemStateTypedDict,
-        InputModelDrivenTelemetryTypedDict,
         InputWindowsMetricsTypedDict,
+        InputModelDrivenTelemetryTypedDict,
         InputSystemMetricsTypedDict,
         InputJournalFilesTypedDict,
+        InputKubeLogsTypedDict,
         InputRawUDPTypedDict,
         InputWinEventLogsTypedDict,
         InputExecTypedDict,
-        InputKubeLogsTypedDict,
         InputMetricsTypedDict,
         InputSnmpTypedDict,
         InputCriblTCPTypedDict,
         InputNetflowTypedDict,
         InputTcpjsonTypedDict,
         InputGooglePubsubTypedDict,
-        InputTCPTypedDict,
-        InputWizTypedDict,
         InputOffice365ServiceTypedDict,
+        InputWizTypedDict,
+        InputTCPTypedDict,
         InputFirehoseTypedDict,
         InputCriblHTTPTypedDict,
-        InputOffice365MgmtTypedDict,
         InputDatadogAgentTypedDict,
+        InputOffice365MgmtTypedDict,
         InputFileTypedDict,
         InputSplunkTypedDict,
         InputAppscopeTypedDict,
+        InputGrafanaTypedDict,
         InputWefTypedDict,
         InputHTTPRawTypedDict,
-        InputHTTPTypedDict,
         InputCriblLakeHTTPTypedDict,
-        InputAzureBlobTypedDict,
-        InputSqsTypedDict,
+        InputHTTPTypedDict,
         InputZscalerHecTypedDict,
-        InputEventhubTypedDict,
+        InputSyslogTypedDict,
+        InputSqsTypedDict,
+        InputAzureBlobTypedDict,
+        InputConfluentCloudTypedDict,
         InputKinesisTypedDict,
+        InputEventhubTypedDict,
         InputKafkaTypedDict,
-        InputConfluentCloudTypedDict,
         InputElasticTypedDict,
-        InputOffice365MsgTraceTypedDict,
         InputSplunkHecTypedDict,
+        InputOffice365MsgTraceTypedDict,
         InputPrometheusRwTypedDict,
         InputLokiTypedDict,
         InputPrometheusTypedDict,
@@ -128,72 +132,70 @@ InputTypedDict = TypeAliasType(
         InputMskTypedDict,
         InputS3InventoryTypedDict,
         InputSplunkSearchTypedDict,
-        InputSyslogTypedDict,
-        InputGrafanaTypedDict,
     ],
 )
 
 
-Input = TypeAliasType(
-    "Input",
+Input = Annotated[
     Union[
-        InputDatagen,
-        InputKubeEvents,
-        InputCribl,
-        InputCriblmetrics,
-        InputKubeMetrics,
-        InputCollection,
-        InputSystemState,
-        InputModelDrivenTelemetry,
-        InputWindowsMetrics,
-        InputSystemMetrics,
-        InputJournalFiles,
-        InputRawUDP,
-        InputWinEventLogs,
-        InputExec,
-        InputKubeLogs,
-        InputMetrics,
-        InputSnmp,
-        InputCriblTCP,
-        InputNetflow,
-        InputTcpjson,
-        InputGooglePubsub,
-        InputTCP,
-        InputWiz,
-        InputOffice365Service,
-        InputFirehose,
-        InputCriblHTTP,
-        InputOffice365Mgmt,
-        InputDatadogAgent,
-        InputFile,
-        InputSplunk,
-        InputAppscope,
-        InputWef,
-        InputHTTPRaw,
-        InputHTTP,
-        InputCriblLakeHTTP,
-        InputAzureBlob,
-        InputSqs,
-        InputZscalerHec,
-        InputEventhub,
-        InputKinesis,
-        InputKafka,
-        InputConfluentCloud,
-        InputElastic,
-        InputOffice365MsgTrace,
-        InputSplunkHec,
-        InputPrometheusRw,
-        InputLoki,
-        InputPrometheus,
-        InputCrowdstrike,
-        InputEdgePrometheus,
-        InputOpenTelemetry,
-        InputS3,
-        InputSecurityLake,
-        InputMsk,
-        InputS3Inventory,
-        InputSplunkSearch,
-        InputSyslog,
-        InputGrafana,
+        Annotated[InputCollection, Tag("collection")],
+        Annotated[InputKafka, Tag("kafka")],
+        Annotated[InputMsk, Tag("msk")],
+        Annotated[InputHTTP, Tag("http")],
+        Annotated[InputSplunk, Tag("splunk")],
+        Annotated[InputSplunkSearch, Tag("splunk_search")],
+        Annotated[InputSplunkHec, Tag("splunk_hec")],
+        Annotated[InputAzureBlob, Tag("azure_blob")],
+        Annotated[InputElastic, Tag("elastic")],
+        Annotated[InputConfluentCloud, Tag("confluent_cloud")],
+        Annotated[InputGrafana, Tag("grafana")],
+        Annotated[InputLoki, Tag("loki")],
+        Annotated[InputPrometheusRw, Tag("prometheus_rw")],
+        Annotated[InputPrometheus, Tag("prometheus")],
+        Annotated[InputEdgePrometheus, Tag("edge_prometheus")],
+        Annotated[InputOffice365Mgmt, Tag("office365_mgmt")],
+        Annotated[InputOffice365Service, Tag("office365_service")],
+        Annotated[InputOffice365MsgTrace, Tag("office365_msg_trace")],
+        Annotated[InputEventhub, Tag("eventhub")],
+        Annotated[InputExec, Tag("exec")],
+        Annotated[InputFirehose, Tag("firehose")],
+        Annotated[InputGooglePubsub, Tag("google_pubsub")],
+        Annotated[InputCribl, Tag("cribl")],
+        Annotated[InputCriblTCP, Tag("cribl_tcp")],
+        Annotated[InputCriblHTTP, Tag("cribl_http")],
+        Annotated[InputCriblLakeHTTP, Tag("cribl_lake_http")],
+        Annotated[InputTcpjson, Tag("tcpjson")],
+        Annotated[InputSystemMetrics, Tag("system_metrics")],
+        Annotated[InputSystemState, Tag("system_state")],
+        Annotated[InputKubeMetrics, Tag("kube_metrics")],
+        Annotated[InputKubeLogs, Tag("kube_logs")],
+        Annotated[InputKubeEvents, Tag("kube_events")],
+        Annotated[InputWindowsMetrics, Tag("windows_metrics")],
+        Annotated[InputCrowdstrike, Tag("crowdstrike")],
+        Annotated[InputDatadogAgent, Tag("datadog_agent")],
+        Annotated[InputDatagen, Tag("datagen")],
+        Annotated[InputHTTPRaw, Tag("http_raw")],
+        Annotated[InputKinesis, Tag("kinesis")],
+        Annotated[InputCriblmetrics, Tag("criblmetrics")],
+        Annotated[InputMetrics, Tag("metrics")],
+        Annotated[InputS3, Tag("s3")],
+        Annotated[InputS3Inventory, Tag("s3_inventory")],
+        Annotated[InputSnmp, Tag("snmp")],
+        Annotated[InputOpenTelemetry, Tag("open_telemetry")],
+        Annotated[InputModelDrivenTelemetry, Tag("model_driven_telemetry")],
+        Annotated[InputSqs, Tag("sqs")],
+        Annotated[InputSyslog, Tag("syslog")],
+        Annotated[InputFile, Tag("file")],
+        Annotated[InputTCP, Tag("tcp")],
+        Annotated[InputAppscope, Tag("appscope")],
+        Annotated[InputWef, Tag("wef")],
+        Annotated[InputWinEventLogs, Tag("win_event_logs")],
+        Annotated[InputRawUDP, Tag("raw_udp")],
+        Annotated[InputJournalFiles, Tag("journal_files")],
+        Annotated[InputWiz, Tag("wiz")],
+        Annotated[InputNetflow, Tag("netflow")],
+        Annotated[InputSecurityLake, Tag("security_lake")],
+        Annotated[InputZscalerHec, Tag("zscaler_hec")],
     ],
-)
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]