cribl-control-plane 0.2.1rc7__py3-none-any.whl → 0.3.0a1__py3-none-any.whl

cribl_control_plane/models/output.py

@@ -19,7 +19,6 @@ 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
@@ -82,10 +81,8 @@ 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 Annotated, TypeAliasType
+from typing_extensions import TypeAliasType
 
 
 OutputTypedDict = TypeAliasType(
@@ -98,137 +95,135 @@ OutputTypedDict = TypeAliasType(
         OutputNetflowTypedDict,
         OutputDiskSpoolTypedDict,
         OutputRingTypedDict,
-        OutputGraphiteTypedDict,
         OutputStatsdExtTypedDict,
+        OutputGraphiteTypedDict,
         OutputStatsdTypedDict,
         OutputGooglePubsubTypedDict,
-        OutputExabeamTypedDict,
+        OutputCriblTCPTypedDict,
         OutputSplunkTypedDict,
         OutputSnsTypedDict,
-        OutputCriblTCPTypedDict,
-        OutputAzureEventhubTypedDict,
         OutputCloudwatchTypedDict,
+        OutputAzureEventhubTypedDict,
+        OutputWavefrontTypedDict,
         OutputSignalfxTypedDict,
         OutputHoneycombTypedDict,
-        OutputWavefrontTypedDict,
-        OutputHumioHecTypedDict,
         OutputSumoLogicTypedDict,
         OutputCrowdstrikeNextGenSiemTypedDict,
+        OutputHumioHecTypedDict,
         OutputTcpjsonTypedDict,
         OutputElasticCloudTypedDict,
-        OutputNewrelicEventsTypedDict,
-        OutputFilesystemTypedDict,
+        OutputKinesisTypedDict,
         OutputConfluentCloudTypedDict,
         OutputKafkaTypedDict,
+        OutputExabeamTypedDict,
+        OutputNewrelicEventsTypedDict,
         OutputAzureLogsTypedDict,
-        OutputKinesisTypedDict,
         OutputSplunkLbTypedDict,
         OutputSyslogTypedDict,
         OutputSqsTypedDict,
-        OutputCriblHTTPTypedDict,
         OutputNewrelicTypedDict,
+        OutputCriblHTTPTypedDict,
         OutputXsiamTypedDict,
+        OutputFilesystemTypedDict,
         OutputDatasetTypedDict,
         OutputLokiTypedDict,
-        OutputCriblLakeTypedDict,
+        OutputSplunkHecTypedDict,
         OutputDynatraceHTTPTypedDict,
         OutputServiceNowTypedDict,
-        OutputSplunkHecTypedDict,
         OutputChronicleTypedDict,
         OutputDynatraceOtlpTypedDict,
-        OutputDatabricksTypedDict,
         OutputElasticTypedDict,
-        OutputDatadogTypedDict,
         OutputGoogleChronicleTypedDict,
+        OutputCriblLakeTypedDict,
+        OutputDatadogTypedDict,
         OutputPrometheusTypedDict,
-        OutputSentinelOneAiSiemTypedDict,
         OutputMskTypedDict,
-        OutputGoogleCloudStorageTypedDict,
+        OutputSentinelOneAiSiemTypedDict,
         OutputSentinelTypedDict,
-        OutputAzureBlobTypedDict,
         OutputInfluxdbTypedDict,
-        OutputMinioTypedDict,
-        OutputSecurityLakeTypedDict,
+        OutputGoogleCloudStorageTypedDict,
+        OutputAzureBlobTypedDict,
         OutputOpenTelemetryTypedDict,
+        OutputMinioTypedDict,
         OutputClickHouseTypedDict,
+        OutputSecurityLakeTypedDict,
         OutputDlS3TypedDict,
         OutputS3TypedDict,
         OutputWebhookTypedDict,
-        OutputGoogleCloudLoggingTypedDict,
         OutputAzureDataExplorerTypedDict,
+        OutputGoogleCloudLoggingTypedDict,
         OutputGrafanaCloudTypedDict,
     ],
 )
 
 
-Output = Annotated[
+Output = TypeAliasType(
+    "Output",
     Union[
-        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")],
+        OutputDevnull,
+        OutputDefault,
+        OutputRouter,
+        OutputSnmp,
+        OutputNetflow,
+        OutputDiskSpool,
+        OutputRing,
+        OutputStatsdExt,
+        OutputGraphite,
+        OutputStatsd,
+        OutputGooglePubsub,
+        OutputCriblTCP,
+        OutputSplunk,
+        OutputSns,
+        OutputCloudwatch,
+        OutputAzureEventhub,
+        OutputWavefront,
+        OutputSignalfx,
+        OutputHoneycomb,
+        OutputSumoLogic,
+        OutputCrowdstrikeNextGenSiem,
+        OutputHumioHec,
+        OutputTcpjson,
+        OutputElasticCloud,
+        OutputKinesis,
+        OutputConfluentCloud,
+        OutputKafka,
+        OutputExabeam,
+        OutputNewrelicEvents,
+        OutputAzureLogs,
+        OutputSplunkLb,
+        OutputSyslog,
+        OutputSqs,
+        OutputNewrelic,
+        OutputCriblHTTP,
+        OutputXsiam,
+        OutputFilesystem,
+        OutputDataset,
+        OutputLoki,
+        OutputSplunkHec,
+        OutputDynatraceHTTP,
+        OutputServiceNow,
+        OutputChronicle,
+        OutputDynatraceOtlp,
+        OutputElastic,
+        OutputGoogleChronicle,
+        OutputCriblLake,
+        OutputDatadog,
+        OutputPrometheus,
+        OutputMsk,
+        OutputSentinelOneAiSiem,
+        OutputSentinel,
+        OutputInfluxdb,
+        OutputGoogleCloudStorage,
+        OutputAzureBlob,
+        OutputOpenTelemetry,
+        OutputMinio,
+        OutputClickHouse,
+        OutputSecurityLake,
+        OutputDlS3,
+        OutputS3,
+        OutputWebhook,
+        OutputAzureDataExplorer,
+        OutputGoogleCloudLogging,
+        OutputGrafanaCloud,
     ],
-    Discriminator(lambda m: get_discriminator(m, "type", "type")),
-]
+)

cribl_control_plane/models/outputazureblob.py

@@ -1,12 +1,11 @@
 """Code generated by Speakeasy (https://speakeasy.com). DO NOT EDIT."""
 
 from __future__ import annotations
-from cribl_control_plane import models, utils
+from cribl_control_plane import utils
 from cribl_control_plane.types import BaseModel
 from cribl_control_plane.utils import validate_open_enum
 from enum import Enum
 import pydantic
-from pydantic import field_serializer
 from pydantic.functional_validators import PlainValidator
 from typing import List, Optional
 from typing_extensions import Annotated, NotRequired, TypedDict
@@ -19,29 +18,22 @@ class OutputAzureBlobType(str, Enum):
 class OutputAzureBlobDataFormat(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Format of the output data"""
 
-    # JSON
     JSON = "json"
-    # Raw
     RAW = "raw"
-    # Parquet
     PARQUET = "parquet"
 
 
 class OutputAzureBlobBackpressureBehavior(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""How to handle events when all receivers are exerting backpressure"""
 
-    # Block
     BLOCK = "block"
-    # Drop
     DROP = "drop"
 
 
 class OutputAzureBlobDiskSpaceProtection(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""How to handle events when disk space is below the global 'Min free disk space' limit"""
 
-    # Block
     BLOCK = "block"
-    # Drop
     DROP = "drop"
 
 
@@ -53,15 +45,10 @@ class OutputAzureBlobAuthenticationMethod(str, Enum, metaclass=utils.OpenEnumMet
 
 
 class BlobAccessTier(str, Enum, metaclass=utils.OpenEnumMeta):
-    # Default account access tier
     INFERRED = "Inferred"
-    # Hot tier
     HOT = "Hot"
-    # Cool tier
     COOL = "Cool"
-    # Cold tier
     COLD = "Cold"
-    # Archive tier
     ARCHIVE = "Archive"
 
 
@@ -75,31 +62,23 @@ class OutputAzureBlobCompression(str, Enum, metaclass=utils.OpenEnumMeta):
 class OutputAzureBlobCompressionLevel(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Compression level to apply before moving files to final destination"""
 
-    # Best Speed
     BEST_SPEED = "best_speed"
-    # Normal
     NORMAL = "normal"
-    # Best Compression
     BEST_COMPRESSION = "best_compression"
 
 
 class OutputAzureBlobParquetVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Determines which data types are supported and how they are represented"""
 
-    # 1.0
     PARQUET_1_0 = "PARQUET_1_0"
-    # 2.4
     PARQUET_2_4 = "PARQUET_2_4"
-    # 2.6
     PARQUET_2_6 = "PARQUET_2_6"
 
 
 class OutputAzureBlobDataPageVersion(str, Enum, metaclass=utils.OpenEnumMeta):
     r"""Serialization format of data pages. Note that some reader implementations use Data page V2's attributes to work more efficiently, while others ignore it."""
 
-    # V1
     DATA_PAGE_V1 = "DATA_PAGE_V1"
-    # V2
     DATA_PAGE_V2 = "DATA_PAGE_V2"
 
 
@@ -185,8 +164,6 @@ class OutputAzureBlobTypedDict(TypedDict):
     r"""Compression level to apply before moving files to final destination"""
     automatic_schema: NotRequired[bool]
     r"""Automatically calculate the schema based on the events of each Parquet file generated"""
-    parquet_schema: NotRequired[str]
-    r"""To add a new schema, navigate to Processing > Knowledge > Parquet Schemas"""
     parquet_version: NotRequired[OutputAzureBlobParquetVersion]
     r"""Determines which data types are supported and how they are represented"""
     parquet_data_page_version: NotRequired[OutputAzureBlobDataPageVersion]
@@ -390,11 +367,6 @@ class OutputAzureBlob(BaseModel):
     ] = False
     r"""Automatically calculate the schema based on the events of each Parquet file generated"""
 
-    parquet_schema: Annotated[Optional[str], pydantic.Field(alias="parquetSchema")] = (
-        None
-    )
-    r"""To add a new schema, navigate to Processing > Knowledge > Parquet Schemas"""
-
     parquet_version: Annotated[
         Annotated[
             Optional[OutputAzureBlobParquetVersion],
@@ -495,84 +467,3 @@ class OutputAzureBlob(BaseModel):
     r"""Select or create a stored text secret"""
 
     certificate: Optional[OutputAzureBlobCertificate] = None
-
-    @field_serializer("format_")
-    def serialize_format_(self, value):
-        if isinstance(value, str):
-            try:
-                return models.OutputAzureBlobDataFormat(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("on_backpressure")
-    def serialize_on_backpressure(self, value):
-        if isinstance(value, str):
-            try:
-                return models.OutputAzureBlobBackpressureBehavior(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("on_disk_full_backpressure")
-    def serialize_on_disk_full_backpressure(self, value):
-        if isinstance(value, str):
-            try:
-                return models.OutputAzureBlobDiskSpaceProtection(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("auth_type")
-    def serialize_auth_type(self, value):
-        if isinstance(value, str):
-            try:
-                return models.OutputAzureBlobAuthenticationMethod(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("storage_class")
-    def serialize_storage_class(self, value):
-        if isinstance(value, str):
-            try:
-                return models.BlobAccessTier(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("compress")
-    def serialize_compress(self, value):
-        if isinstance(value, str):
-            try:
-                return models.OutputAzureBlobCompression(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("compression_level")
-    def serialize_compression_level(self, value):
-        if isinstance(value, str):
-            try:
-                return models.OutputAzureBlobCompressionLevel(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("parquet_version")
-    def serialize_parquet_version(self, value):
-        if isinstance(value, str):
-            try:
-                return models.OutputAzureBlobParquetVersion(value)
-            except ValueError:
-                return value
-        return value
-
-    @field_serializer("parquet_data_page_version")
-    def serialize_parquet_data_page_version(self, value):
-        if isinstance(value, str):
-            try:
-                return models.OutputAzureBlobDataPageVersion(value)
-            except ValueError:
-                return value
-        return value