ZThK"ddlmZddlZddlmZddlmZddlmZddl m Z m Z ddl m Z mZmZmZmZer ddlmZdd lmZGd d eZy) ) annotationsN) TYPE_CHECKING)RunsLowLevelClient) ResourceBase)Run RunUpdate)containsequals equals_nullmatchnot_)datetime) SiftClientceZdZdZdfd Z ddZddddddddddddddd ddZddZ d dd Zdd Z dd Z dd Z dd Z ddZ xZS) RunsAPIAsyncaHigh-level API for interacting with runs. This class provides a Pythonic, notebook-friendly interface for interacting with the RunsAPI. It handles automatic handling of gRPC services, seamless type conversion, and clear error handling. All methods in this class use the Run class from the low-level wrapper, which is a user-friendly representation of a run using standard Python data structures and types. cnt||t|jj|_y)z`Initialize the RunsAPI. Args: sift_client: The Sift client to use. ) grpc_clientN)super__init__rclientr_low_level_client)self sift_client __class__s _/home/www/backend.miabetepe.com/venv/lib/python3.12/site-packages/sift_client/resources/runs.pyrzRunsAPIAsync.__init__s) %!3 @W@W!XcxK|jj|d{}|j|S7w)zpGet a Run. Args: run_id: The ID of the run. Returns: The Run. run_idN)rget_run_apply_client_to_instance)rrruns rgetzRunsAPIAsync.get$s;**22&2AA--c22Bs :8:NF)name name_contains name_regex descriptiondescription_containsduration_seconds client_keyasset_id asset_namecreated_by_user_id is_stoppedinclude_archivedorder_bylimitcKg}|r|jtd|na|r|jtd|nC|rAt|tj r |j }|jtd||r|jtd|n|r|jtd||r|jtd||r|jtd||r|jtd|| r|jtd| | r|jtd| | #|jttd | s|jtd d|rd j|nd}|jj|| | d{}|j|S7w) aList runs with optional filtering. Args: name: Exact name of the run. name_contains: Partial name of the run. name_regex: Regular expression string to filter runs by name. description: Exact description of the run. description_contains: Partial description of the run. duration_seconds: Duration of the run in seconds. client_key: Client key to filter by. asset_id: Asset ID to filter by. asset_name: Asset name to filter by. created_by_user_id: User ID who created the run. is_stopped: Whether the run is stopped. include_archived: Whether to include archived runs. order_by: How to order the retrieved runs. limit: How many runs to retrieve. If None, retrieves all matches. Returns: A list of Runs that matches the filter. r$r'durationr*r+r,r-N stop_time archived_datez && ) query_filterr0 max_results)appendr r isinstancerePatternpatternr r r joinr list_all_runs_apply_client_to_instances)rr$r%r&r'r(r)r*r+r,r-r.r/r0r1 filter_partsr6runss rlist_zRunsAPIAsync.list_4sP    vt 4 5     ? @ *bjj1'//   fj 9 :    }k B C !   8L M N    z3C D E    |Z @ A    z8 < =    |Z @ A    ';=O P Q  !   [%= > ?    = >4@v{{<0d ++99%:   ..t44  sGG G G cK|jdi|d{}t|dkDr tdt|dk(r|dSy71w)aFind a single run matching the given query. Takes the same arguments as `list`. If more than one run is found, raises an error. Args: **kwargs: Keyword arguments to pass to `list`. Returns: The Run found or None. NzMultiple runs found for queryr)rBlen ValueError)rkwargsrAs rfindzRunsAPIAsync.findsQ TZZ)&)) t9q=<= = Y!^7N *sA A 2A c K|jj||||||||d{} |j| S7w)aCreate a new run. Args: name: The name of the run. description: The description of the run. tags: Tags to associate with the run. start_time: The start time of the run. stop_time: The stop time of the run. organization_id: The organization ID. client_key: A unique client key for the run. metadata: Metadata values for the run. Returns: The created Run. )r$r'tags start_timer4organization_idr*metadataN)r create_runr!) rr$r'rKrLr4rMr*rN created_runs rcreatezRunsAPIAsync.createsX4!22==#!+!>   --k::  s'A?Ac<Kt|tr|j|d{}t|trt j |}|j |_|jj||d{}|j|S7o7w)zUpdate a Run. Args: run: The Run or run ID to update. update: Updates to apply to the Run. Returns: The updated Run. rN) r9strr#dictrmodel_validateid_ resource_idr update_runr!)rr"update updated_runs rrYzRunsAPIAsync.updates c3 ,,C fd #--f5F WW 22==c6JJ --k::- Ks"&BBABBBBcKt|tr |jn|}t|tst dt ||j j|d{y7w)zUArchive a run. Args: run: The Run or run ID to archive. zrun_id must be a string not rN)r9rrVrS TypeErrortyper archive_runrr"rs rarchivezRunsAPIAsync.archivesZ'sC0c&#&:4<.IJ J$$000???sA%A/'A-(A/cKt|tr |jn|}|jj |xsdd{y7wz|Stop a run by setting its stop time to the current time. Args: run: The Run or run ID to stop. rNr9rrVrstop_runr_s rstopzRunsAPIAsync.stops='sC0c$$--V\r-BBBsAA A A cKt|tr|jxsdn|}|jj ||d{y7w)zAssociate assets with a run for automatic data ingestion. Args: run: The Run or run ID. asset_names: List of asset names to associate. rc)r asset_namesN)r9rrVr+create_automatic_run_association_for_assets)rr"rhrs r'create_automatic_association_for_assetsz4RunsAPIAsync.create_automatic_association_for_assetssI#-S#"6BC$$PP{Q   sAA A A cKt|tr|jxsdn|}|jj |xsdd{y7wrbrdr_s rrezRunsAPIAsync.stop_runsA #-S#"6BC$$--V\r-BBBsAAA A)rr)rrSreturnr)r$ str | Noner%rmr&zstr | re.Pattern | Noner'rmr(rmr) int | Noner*rmr+rmr,rmr-rmr.z bool | Noner/boolr0rmr1rnrlz list[Run])rlz Run | None)NNNNNN)r$rSr'rSrKzlist[str] | NonerLdatetime | Noner4rprMrmr*rmrNz$dict[str, str | float | bool] | Nonerlr)r" str | RunrYzRunUpdate | dictrlr)r"rqrlNone)r"rqrhz list[str]rlrr)__name__ __module__ __qualname____doc__rr#rBrIrQrYr`rfrjre __classcell__)rs@rrrsY33 3& $(.2"&+/'+!%#!%)-"&!&# !T5T5" T5 , T5  T5)T5%T5T5T5T5'T5 T5T5T5 !T5" #T5l*"&&*%)&*!%9=$;$;$; $; $ $; # $;$$;$;7$; $;L;( @ @  @ C C  C      Crr) __future__rr:typingr-sift_client._internal.low_level_wrappers.runsrsift_client.resources._basersift_client.sift_types.runrrsift_client.util.cel_utilsr r r r r rsift_client.clientrrrErrrs6" L45QQ!-wC<wCr