ZTh ddlmZddlZddlmZmZddlmZddlm Z erddl m Z ddl m Z ddl mZejeZGd d e Zy) ) annotationsN) TYPE_CHECKINGAny)IngestionLowLevelClient) ResourceBase)datetime) SiftClient)FlowcjeZdZdZdfd Zdddd d dZ d dZd d dZxZS) IngestionAPIAsyncaHigh-level API for interacting with ingestion services. This class provides a Pythonic, notebook-friendly interface for interacting with the IngestionAPI. It handles automatic handling of gRPC services, seamless type conversion, and clear error handling. All methods in this class use the Flow class from the types module, which is a user-friendly representation of ingestion flows using standard Python data structures and types. cnt||t|jj|_y)zeInitialize the IngestionAPI. Args: sift_client: The Sift client to use. ) grpc_clientN)super__init__rclientr_low_level_client)self sift_client __class__s d/home/www/backend.miabetepe.com/venv/lib/python3.12/site-packages/sift_client/resources/ingestion.pyrzIngestionAPIAsync.__init__s) %!8T[[E\E\!]N)run_id client_keyorganization_idcK|s td|s td|jj||||d{}|D]'}|j|j|s!||_)|S72w)a Create an ingestion config. Args: asset_name: The name of the asset for this ingestion config. run_id: Optionally provide a run ID to create a run for the given asset. flows: List of flow configurations. client_key: Optional client key for identifying this config. organization_id: The organization ID. Returns: The ingestion config ID. Raises: ValueError: If asset_name is not provided or flows is empty. zasset_name must be providedzflows must not be empty) asset_nameflowsrrN) ValueErrorrcreate_ingestion_config_apply_client_to_instancerr)rrrrrringestion_config_idflows rrz)IngestionAPIAsync.create_ingestion_config%s0:; ;67 7$($:$:$R$R!!+ %S%    %D  * *4;; 7$  % #" s=A4A2&A4' A4c@|jj|||y)zIngest data for a flow. Args: flow: The flow to ingest data for. timestamp: The timestamp of the data. channel_values: Dictionary mapping channel names to their values. )r" timestampchannel_valuesN)r ingest_flow)rr"r$r%s ringestzIngestionAPIAsync.ingestOs& **) + rcdtjd|jj|y)zWait for all ingestion to complete. Args: run_id: The id of the run to wait for. timeout: The timeout in seconds to wait for ingestion to complete. If None, will wait forever. z!Waiting for ingestion to completeN)loggerinforwait_for_ingestion_to_complete)rtimeouts rr+z0IngestionAPIAsync.wait_for_ingestion_to_completecs%  78 ==gFr)rr ) rstrr str | Nonerz list[Flow]rr.rr.returnr-)r"r r$rr%zdict[str, Any])N)r,z float | None) __name__ __module__ __qualname____doc__rrr'r+ __classcell__)rs@rr r s^"!%&*(#(# (#  (#  (#$(# (#T    '  (GGrr ) __future__rloggingtypingrr2sift_client._internal.low_level_wrappers.ingestionrsift_client.resources._baserrsift_client.clientr sift_client.sift_types.ingestionr getLoggerr0r)r rrr>sB"%V4!-5   8 $YG YGr