cribl-control-plane 0.0.21__py3-none-any.whl → 0.4.0b23__py3-none-any.whl

cribl_control_plane/models/nodeprovidedinfo.py

@@ -3,38 +3,34 @@
 from __future__ import annotations
 from .hbcriblinfo import HBCriblInfo, HBCriblInfoTypedDict
 from .heartbeatmetadata import HeartbeatMetadata, HeartbeatMetadataTypedDict
+from .outpostnodeinfo import OutpostNodeInfo, OutpostNodeInfoTypedDict
 from cribl_control_plane.types import BaseModel
 import pydantic
-from typing import List, Optional, Union
+from typing import Dict, List, Optional, Union
 from typing_extensions import Annotated, NotRequired, TypeAliasType, TypedDict
 
 
-class NodeProvidedInfoTagsTypedDict(TypedDict):
-    pass
-
-
-class NodeProvidedInfoTags(BaseModel):
-    pass
-
-
 class NodeProvidedInfoAwsTypedDict(TypedDict):
     enabled: bool
+    instance_id: str
     region: str
     type: str
     zone: str
-    tags: NotRequired[NodeProvidedInfoTagsTypedDict]
+    tags: NotRequired[Dict[str, str]]
 
 
 class NodeProvidedInfoAws(BaseModel):
     enabled: bool
 
+    instance_id: Annotated[str, pydantic.Field(alias="instanceId")]
+
     region: str
 
     type: str
 
     zone: str
 
-    tags: Optional[NodeProvidedInfoTags] = None
+    tags: Optional[Dict[str, str]] = None
 
 
 class NodeProvidedInfoHostOsTypedDict(TypedDict):
@@ -125,6 +121,7 @@ class NodeProvidedInfoTypedDict(TypedDict):
     architecture: str
     cpus: float
     cribl: HBCriblInfoTypedDict
+    env: Dict[str, str]
     free_disk_space: float
     hostname: str
     node: str
@@ -140,6 +137,7 @@ class NodeProvidedInfoTypedDict(TypedDict):
     local_time: NotRequired[float]
     metadata: NotRequired[HeartbeatMetadataTypedDict]
     os: NotRequired[OsTypedDict]
+    outpost: NotRequired[OutpostNodeInfoTypedDict]
 
 
 class NodeProvidedInfo(BaseModel):
@@ -149,6 +147,8 @@ class NodeProvidedInfo(BaseModel):
 
     cribl: HBCriblInfo
 
+    env: Dict[str, str]
+
     free_disk_space: Annotated[float, pydantic.Field(alias="freeDiskSpace")]
 
     hostname: str
@@ -182,3 +182,5 @@ class NodeProvidedInfo(BaseModel):
     metadata: Optional[HeartbeatMetadata] = None
 
     os: Optional[Os] = None
+
+    outpost: Optional[OutpostNodeInfo] = None

cribl_control_plane/models/nodeskippedupgradestatus.py

@@ -1,10 +1,11 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from cribl_control_plane import utils
 from enum import Enum
 
 
-class NodeSkippedUpgradeStatus(int, Enum):
+class NodeSkippedUpgradeStatus(int, Enum, metaclass=utils.OpenEnumMeta):
     ZERO = 0
     ONE = 1
     TWO = 2

cribl_control_plane/models/nodeupgradestate.py

@@ -1,10 +1,11 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
+from cribl_control_plane import utils
 from enum import Enum
 
 
-class NodeUpgradeState(int, Enum):
+class NodeUpgradeState(int, Enum, metaclass=utils.OpenEnumMeta):
     ZERO = 0
     ONE = 1
     TWO = 2

cribl_control_plane/models/nodeupgradestatus.py

@@ -5,9 +5,13 @@ from .nodeactiveupgradestatus import NodeActiveUpgradeStatus
 from .nodefailedupgradestatus import NodeFailedUpgradeStatus
 from .nodeskippedupgradestatus import NodeSkippedUpgradeStatus
 from .nodeupgradestate import NodeUpgradeState
+from cribl_control_plane import models
 from cribl_control_plane.types import BaseModel
+from cribl_control_plane.utils import validate_open_enum
+from pydantic import field_serializer
+from pydantic.functional_validators import PlainValidator
 from typing import Optional
-from typing_extensions import NotRequired, TypedDict
+from typing_extensions import Annotated, NotRequired, TypedDict
 
 
 class NodeUpgradeStatusTypedDict(TypedDict):
@@ -19,12 +23,54 @@ class NodeUpgradeStatusTypedDict(TypedDict):
 
 
 class NodeUpgradeStatus(BaseModel):
-    state: NodeUpgradeState
+    state: Annotated[NodeUpgradeState, PlainValidator(validate_open_enum(True))]
 
     timestamp: float
 
-    active: Optional[NodeActiveUpgradeStatus] = None
+    active: Annotated[
+        Optional[NodeActiveUpgradeStatus], PlainValidator(validate_open_enum(True))
+    ] = None
+
+    failed: Annotated[
+        Optional[NodeFailedUpgradeStatus], PlainValidator(validate_open_enum(True))
+    ] = None
+
+    skipped: Annotated[
+        Optional[NodeSkippedUpgradeStatus], PlainValidator(validate_open_enum(True))
+    ] = None
+
+    @field_serializer("active")
+    def serialize_active(self, value):
+        if isinstance(value, str):
+            try:
+                return models.NodeActiveUpgradeStatus(value)
+            except ValueError:
+                return value
+        return value
+
+    @field_serializer("failed")
+    def serialize_failed(self, value):
+        if isinstance(value, str):
+            try:
+                return models.NodeFailedUpgradeStatus(value)
+            except ValueError:
+                return value
+        return value
 
-    failed: Optional[NodeFailedUpgradeStatus] = None
+    @field_serializer("skipped")
+    def serialize_skipped(self, value):
+        if isinstance(value, str):
+            try:
+                return models.NodeSkippedUpgradeStatus(value)
+            except ValueError:
+                return value
+        return value
 
-    skipped: Optional[NodeSkippedUpgradeStatus] = None
+    @field_serializer("state")
+    def serialize_state(self, value):
+        if isinstance(value, str):
+            try:
+                return models.NodeUpgradeState(value)
+            except ValueError:
+                return value
+        return value

cribl_control_plane/models/outpostnodeinfo.py

@@ -0,0 +1,16 @@
+"""Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
+
+from __future__ import annotations
+from cribl_control_plane.types import BaseModel
+from typing_extensions import TypedDict
+
+
+class OutpostNodeInfoTypedDict(TypedDict):
+    guid: str
+    host: str
+
+
+class OutpostNodeInfo(BaseModel):
+    guid: str
+
+    host: str

cribl_control_plane/models/output.py

@@ -8,7 +8,9 @@ from .outputazuredataexplorer import (
 )
 from .outputazureeventhub import OutputAzureEventhub, OutputAzureEventhubTypedDict
 from .outputazurelogs import OutputAzureLogs, OutputAzureLogsTypedDict
+from .outputchronicle import OutputChronicle, OutputChronicleTypedDict
 from .outputclickhouse import OutputClickHouse, OutputClickHouseTypedDict
+from .outputcloudflarer2 import OutputCloudflareR2, OutputCloudflareR2TypedDict
 from .outputcloudwatch import OutputCloudwatch, OutputCloudwatchTypedDict
 from .outputconfluentcloud import OutputConfluentCloud, OutputConfluentCloudTypedDict
 from .outputcriblhttp import OutputCriblHTTP, OutputCriblHTTPTypedDict
@@ -18,6 +20,7 @@ from .outputcrowdstrikenextgensiem import (
     OutputCrowdstrikeNextGenSiem,
     OutputCrowdstrikeNextGenSiemTypedDict,
 )
+from .outputdatabricks import OutputDatabricks, OutputDatabricksTypedDict
 from .outputdatadog import OutputDatadog, OutputDatadogTypedDict
 from .outputdataset import OutputDataset, OutputDatasetTypedDict
 from .outputdefault import OutputDefault, OutputDefaultTypedDict
@@ -48,6 +51,7 @@ from .outputinfluxdb import OutputInfluxdb, OutputInfluxdbTypedDict
 from .outputkafka import OutputKafka, OutputKafkaTypedDict
 from .outputkinesis import OutputKinesis, OutputKinesisTypedDict
 from .outputloki import OutputLoki, OutputLokiTypedDict
+from .outputmicrosoftfabric import OutputMicrosoftFabric, OutputMicrosoftFabricTypedDict
 from .outputminio import OutputMinio, OutputMinioTypedDict
 from .outputmsk import OutputMsk, OutputMskTypedDict
 from .outputnetflow import OutputNetflow, OutputNetflowTypedDict
@@ -80,8 +84,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(
@@ -95,132 +101,140 @@ OutputTypedDict = TypeAliasType(
         OutputDiskSpoolTypedDict,
         OutputRingTypedDict,
         OutputGraphiteTypedDict,
-        OutputStatsdTypedDict,
         OutputStatsdExtTypedDict,
+        OutputStatsdTypedDict,
         OutputGooglePubsubTypedDict,
-        OutputCriblTCPTypedDict,
         OutputSplunkTypedDict,
         OutputSnsTypedDict,
-        OutputCloudwatchTypedDict,
-        OutputSyslogTypedDict,
+        OutputCriblTCPTypedDict,
         OutputAzureEventhubTypedDict,
-        OutputWavefrontTypedDict,
-        OutputSignalfxTypedDict,
+        OutputCloudwatchTypedDict,
+        OutputExabeamTypedDict,
+        OutputMicrosoftFabricTypedDict,
         OutputHoneycombTypedDict,
-        OutputTcpjsonTypedDict,
+        OutputSignalfxTypedDict,
+        OutputWavefrontTypedDict,
         OutputSumoLogicTypedDict,
+        OutputCrowdstrikeNextGenSiemTypedDict,
         OutputHumioHecTypedDict,
+        OutputTcpjsonTypedDict,
         OutputElasticCloudTypedDict,
-        OutputCrowdstrikeNextGenSiemTypedDict,
         OutputKinesisTypedDict,
         OutputConfluentCloudTypedDict,
         OutputKafkaTypedDict,
-        OutputExabeamTypedDict,
         OutputNewrelicEventsTypedDict,
         OutputAzureLogsTypedDict,
         OutputSplunkLbTypedDict,
         OutputSqsTypedDict,
         OutputNewrelicTypedDict,
-        OutputCriblHTTPTypedDict,
-        OutputXsiamTypedDict,
         OutputFilesystemTypedDict,
+        OutputSyslogTypedDict,
+        OutputXsiamTypedDict,
+        OutputCriblHTTPTypedDict,
         OutputDatasetTypedDict,
-        OutputLokiTypedDict,
-        OutputSplunkHecTypedDict,
         OutputDynatraceHTTPTypedDict,
-        OutputServiceNowTypedDict,
+        OutputLokiTypedDict,
+        OutputChronicleTypedDict,
         OutputDynatraceOtlpTypedDict,
-        OutputElasticTypedDict,
-        OutputGoogleChronicleTypedDict,
+        OutputServiceNowTypedDict,
+        OutputSplunkHecTypedDict,
         OutputCriblLakeTypedDict,
+        OutputElasticTypedDict,
         OutputDatadogTypedDict,
+        OutputGoogleChronicleTypedDict,
         OutputPrometheusTypedDict,
-        OutputMskTypedDict,
+        OutputDatabricksTypedDict,
         OutputSentinelOneAiSiemTypedDict,
+        OutputMskTypedDict,
         OutputSentinelTypedDict,
-        OutputInfluxdbTypedDict,
         OutputGoogleCloudStorageTypedDict,
+        OutputInfluxdbTypedDict,
         OutputAzureBlobTypedDict,
-        OutputOpenTelemetryTypedDict,
+        OutputCloudflareR2TypedDict,
         OutputMinioTypedDict,
-        OutputClickHouseTypedDict,
+        OutputOpenTelemetryTypedDict,
         OutputSecurityLakeTypedDict,
+        OutputClickHouseTypedDict,
         OutputDlS3TypedDict,
         OutputS3TypedDict,
-        OutputAzureDataExplorerTypedDict,
         OutputWebhookTypedDict,
         OutputGoogleCloudLoggingTypedDict,
+        OutputAzureDataExplorerTypedDict,
         OutputGrafanaCloudTypedDict,
     ],
 )
 
 
-Output = TypeAliasType(
-    "Output",
+Output = Annotated[
     Union[
-        OutputDevnull,
-        OutputDefault,
-        OutputRouter,
-        OutputSnmp,
-        OutputNetflow,
-        OutputDiskSpool,
-        OutputRing,
-        OutputGraphite,
-        OutputStatsd,
-        OutputStatsdExt,
-        OutputGooglePubsub,
-        OutputCriblTCP,
-        OutputSplunk,
-        OutputSns,
-        OutputCloudwatch,
-        OutputSyslog,
-        OutputAzureEventhub,
-        OutputWavefront,
-        OutputSignalfx,
-        OutputHoneycomb,
-        OutputTcpjson,
-        OutputSumoLogic,
-        OutputHumioHec,
-        OutputElasticCloud,
-        OutputCrowdstrikeNextGenSiem,
-        OutputKinesis,
-        OutputConfluentCloud,
-        OutputKafka,
-        OutputExabeam,
-        OutputNewrelicEvents,
-        OutputAzureLogs,
-        OutputSplunkLb,
-        OutputSqs,
-        OutputNewrelic,
-        OutputCriblHTTP,
-        OutputXsiam,
-        OutputFilesystem,
-        OutputDataset,
-        OutputLoki,
-        OutputSplunkHec,
-        OutputDynatraceHTTP,
-        OutputServiceNow,
-        OutputDynatraceOtlp,
-        OutputElastic,
-        OutputGoogleChronicle,
-        OutputCriblLake,
-        OutputDatadog,
-        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")],
+        Annotated[OutputChronicle, Tag("chronicle")],
+        Annotated[OutputDatabricks, Tag("databricks")],
+        Annotated[OutputMicrosoftFabric, Tag("microsoft_fabric")],
+        Annotated[OutputCloudflareR2, Tag("cloudflare_r2")],
     ],
-)
+    Discriminator(lambda m: get_discriminator(m, "type", "type")),
+]