ZThg;ddlmZddlmZmZddlmZmZddlmZddl m Z ddl m Z m Z ddlmZerddlZdd lmZdd lmZGd d e Zy) ) annotations)datetimetimezone) TYPE_CHECKINGAny) CalculatedChannelsLowLevelClient) ResourceBase)CalculatedChannelCalculatedChannelUpdate) cel_utilsN) SiftClient)ChannelReferenceceZdZdZdfd Zdddd ddZddddddddddddddddddddd ddZdd Zd dddddd d  dd Zdd  ddZ ddZ ddddddddddddddd ddZ xZ S)CalculatedChannelsAPIAsyncaHigh-level API for interacting with calculated channels. This class provides a Pythonic, notebook-friendly interface for interacting with the CalculatedChannelsAPI. It handles automatic handling of gRPC services, seamless type conversion, and clear error handling. All methods in this class use the CalculatedChannel class from the low-level wrapper, which is a user-friendly representation of a calculated channel using standard Python data structures and types. cnt||t|jj|_y)znInitialize the CalculatedChannelsAPI. Args: sift_client: The Sift client to use. ) grpc_clientN)super__init__rclientr_low_level_client)self sift_client __class__s n/home/www/backend.miabetepe.com/venv/lib/python3.12/site-packages/sift_client/resources/calculated_channels.pyrz#CalculatedChannelsAPIAsync.__init__!s, %!A //" Ncalculated_channel_id client_keyorganization_idcK|s |s td|jj|||d{}|j|S7w)aGet a Calculated Channel. Args: calculated_channel_id: The ID of the calculated channel. client_key: The client key of the calculated channel. organization_id: The organization ID (required if using client_key and user belongs to multiple organizations). Returns: The CalculatedChannel. Raises: ValueError: If neither calculated_channel_id nor client_key is provided. z;Either calculated_channel_id or client_key must be providedrN) ValueErrorrget_calculated_channel_apply_client_to_instance)rrrrcalculated_channels rgetzCalculatedChannelsAPIAsync.get,s](%ZZ[ [#'#9#9#P#P"7!+$Q$   --.@AA  s1A A A F)name name_contains name_regex created_aftercreated_beforemodified_aftermodified_before created_by modified_byrasset_id asset_nametag_idtag_nameversioninclude_archived filter_queryorder_bylimitrcK|sEg}|r%|jtjd||r%|jtjd||r%|jtjd||r%|jtj d||r%|jtj d||r%|jtj d||r%|jtj d||rt| rt| r%|jtjd| | r%|jtjd| | r%|jtjd| | r%|jtjd| |r%|jtjd||r%|jtjd ||s$|jtjd tj|}|jj|||| d {}|j|S7w) aList calculated channels with optional filtering. Args: name: Exact name of the calculated channel. name_contains: Partial name of the calculated channel. name_regex: Regular expression string to filter calculated channels by name. created_after: Created after this date. created_before: Created before this date. modified_after: Modified after this date. modified_before: Modified before this date. created_by: Calculated channels created by this user. modified_by: Calculated channels last modified by this user. client_key: The client key of the calculated channel. asset_id: The asset ID associated with the calculated channel. asset_name: The asset name associated with the calculated channel. tag_id: The tag ID associated with the calculated channel. tag_name: The tag name associated with the calculated channel. version: The version of the calculated channel. include_archived: Include archived calculated channels. filter_query: Explicit CEL query to filter calculated channels. order_by: How to order the retrieved calculated channels. limit: How many calculated channels to retrieve. If None, retrieves all matches. organization_id: The organization ID (required if user belongs to multiple organizations). Returns: A list of CalculatedChannels that matches the filter. r& created_date modified_daterr/r0r1r2r3 archived_date) query_filterr6 max_resultsrN) appendcelequalscontainsmatch greater_than less_thanNotImplementedError equals_nulland_rlist_all_calculated_channels_apply_client_to_instances)rr&r'r(r)r*r+r,r-r.rr/r0r1r2r3r4r5r6r7rfilterscalculated_channelss rlist_z CalculatedChannelsAPIAsync.list_KsfGszz&$78s||FMBCsyy<=s// NOs}}^^LMs//PQs}}_oNO))))szz, CDszz*h?@szz, CDszz(F;<szz*h?@szz)W=>#s?@88W-L$($:$:$W$W%+ %X%   ../BCC  sI+J-J.JcK|jdi|d{}t|dkDrtd|dt|dk(r|dSy75w)aeFind a single calculated channel matching the given query. Takes the same arguments as `list` but handles checking for multiple matches. Will raise an error if multiple calculated channels are found. Args: **kwargs: Keyword arguments to pass to `list`. Returns: The CalculatedChannel found or None. Nz.Multiple calculated channels found for query: z8. Use `list` to handle all matching calculated channels.r)rLlenr!)rkwargsrKs rfindzCalculatedChannelsAPIAsync.findsq%/DJJ$8$88 " #a '@III $ % *&q) )9sAA6A) descriptionunitsr asset_idstag_ids all_assets user_notesc K| r|s|r td| s|s |s td|jj|| |||||| || d{\} } |j| S7w)aCreate a calculated channel. Args: name: The name of the calculated channel. expression: The expression to calculate the value of the calculated channel. channel_references: A list of channel references that are used in the expression. description: The description of the calculated channel. units: The units of the calculated channel. client_key: A user-defined unique identifier for the calculated channel. asset_ids: A list of asset IDs to make the calculation available for. tag_ids: A list of tag IDs to make the calculation available for. all_assets: A flag that, when set to True, associates the calculated channel with all assets. user_notes: User notes for the calculated channel. Returns: The created CalculatedChannel. Raises: ValueError: If asset configuration is invalid. z4Cannot specify both all_assets and asset_ids/tag_idsz@Must specify either all_assets=True or provide asset_ids/tag_ids) r&rXrVrW expressionchannel_referencesrTrYrUrN)r!rcreate_calculated_channelr#) rr&r[r\rTrUrrVrWrXrYr$inapplicable_assetss rcreatez!CalculatedChannelsAPIAsync.createsF 9ST T)G_` ` ((BB!!1#!!C    --.@AA sA A( A&A()rYcKt|tr |jn|}t|trt j |}||_|jj||d{\}}|j|S7w)aCUpdate a Calculated Channel. Args: calculated_channel: The CalculatedChannel or id of the CalculatedChannel to update. update: Updates to apply to the CalculatedChannel. user_notes: User notes for the update. Returns: The updated CalculatedChannel. )updaterYN) isinstancer id_dictr model_validate resource_idrupdate_calculated_channelr#)rr$rarYrupdated_calculated_channelr^s rraz!CalculatedChannelsAPIAsync.updates&,.?@  " "#  fd #,;;FCF2 ((BBjC   &  --.HII  sA+B-B.BcKttjtj}|j ||d{y7w)zArchive a Calculated Channel.)tz)r;)r$raN)r rnowrutcra)rr$ras rarchivez"CalculatedChannelsAPIAsync.archives9(",,(,,7 kk-?kOOOsAAA A)rrrr&r'r(r/r0r1r2r3r4r6r7cKtd||||fDdk7r td|r |r td| r | r tdg}|r%|jtjd||r%|jtj d||r%|jtj d||r%|jtjd||r%|jtjd|| r%|jtjd | | r%|jtjd | | r%|jtjd | | s$|jtjd tj|}|jj||||| | d{}|j|S7w)aList versions of a calculated channel. Args: calculated_channel_id: The ID of the calculated channel. client_key: The client key of the calculated channel. name: The name of the calculated channel. name_contains: The name of the calculated channel. name_regex: The name of the calculated channel. asset_id: The asset ID of the calculated channel. asset_name: The asset name of the calculated channel. tag_id: The tag ID of the calculated channel. tag_name: The tag name of the calculated channel. version: The version of the calculated channel. include_archived: Whether to include archived calculated channels. organization_id: The organization ID. Required if your user belongs to multiple organizations. order_by: The field to order by. limit: How many versions to retrieve. If None, retrieves all matches. Returns: A list of CalculatedChannel versions. Raises: ValueError: If neither calculated_channel_id nor client_key is provided. c32K|]}t|yw)N)bool).0vs r z;CalculatedChannelsAPIAsync.list_versions..NsY1tAwYsrNzYExactly one of calculated_channel_id, name, name_contains, or name_regex must be providedz+Cannot specify both asset_id and asset_namez'Cannot specify both tag_id and tag_namer&r/r0r1r2r3r;)rrrr<r6r7N) sumr!r>r?r@rArBrFrGr$list_all_calculated_channel_versionsrI)rrrrr&r'r(r/r0r1r2r3r4r6r7filter_query_partsr5versionss r list_versionsz(CalculatedChannelsAPIAsync.list_versions$sT Y!6mZ XY Y]^ ^k   JK K hFG G   % %cjj&> ?   % %cll6=&I J   % %cii &C D   % %cjjX&F G   % %cjjz&J K   % %cjj6&B C   % %cjjX&F G   % %cjjG&D E  % %cooo&F Gxx!34 //TT"7!+% U  ..x88 sGG3G1G3)rr )r str | Nonerryrryreturnr )*r&ryr'ryr(str | re.Pattern | Noner)datetime | Noner*r|r+r|r,r|r- Any | Noner.r}rryr/ryr0ryr1ryr2ryr3 int | Noner4rpr5ryr6ryr7r~rryrzlist[CalculatedChannel])rzzCalculatedChannel | None)r&strr[rr\zlist[ChannelReference]rTrrUryrryrVlist[str] | NonerWrrXrprYrrzr )r$str | CalculatedChannelrazCalculatedChannelUpdate | dictrYryrzr )r$rrzNone)rryrryrryr&ryr'ryr(r{r/ryr0ryr1ryr2ryr3r~r4rpr6ryr7r~rzr) __name__ __module__ __qualname____doc__rr%rLrRr_rarmrx __classcell__)rs@rrrs>  -1!%&* B *B B $ B  BD $(.2)-*.*.+/!%"&!%#!%!#"!&#'# &*-]D]D" ]D , ]D ' ]D(]D(]D)]D]D ]D]D]D]D]D !]D"#]D$%]D&!']D()]D*+]D,$-]D. !/]D~4 !%&*$( 8B8B 8B 3 8B  8B8B8B$8B"8B8B8B 8B~"& #J4#J/ #J  #J  #JJP-1!%&*$(.2#!%!#"!&# !Q9 *Q9 Q9 $ Q9  Q9"Q9,Q9Q9Q9Q9Q9Q9Q9Q9 !Q9" !#Q9rr) __future__rrrtypingrrrs<"'%5. -?^9^9r