From 69e558c8b55f11a1b2d85dc53654350fce48485d Mon Sep 17 00:00:00 2001 From: samson0v Date: Wed, 19 Jul 2023 11:37:18 +0300 Subject: [PATCH] Moved common method from CE to base --- tb_rest_client/rest_client_base.py | 3 +++ tb_rest_client/rest_client_ce.py | 3 --- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tb_rest_client/rest_client_base.py b/tb_rest_client/rest_client_base.py index 12ab2d89..4e449622 100644 --- a/tb_rest_client/rest_client_base.py +++ b/tb_rest_client/rest_client_base.py @@ -1765,6 +1765,9 @@ def list_slack_conversations(self, type: str, token: Optional[str] = None) -> Li def save_notification_template(self, body: NotificationTemplate) -> NotificationTemplate: return self.notification_template_controller.save_notification_template_using_post(body=body) + def count_entities_by_query(self, body: Optional[EntityCountQuery] = None) -> int: + return self.entity_query_controller.count_entities_by_query_using_post(body=body) + def __load_controllers(self): self.audit_log_controller = AuditLogControllerApi(self.api_client) self.o_auth2_config_template_controller = OAuth2ConfigTemplateControllerApi(self.api_client) diff --git a/tb_rest_client/rest_client_ce.py b/tb_rest_client/rest_client_ce.py index 1a8f799e..cda0a85c 100644 --- a/tb_rest_client/rest_client_ce.py +++ b/tb_rest_client/rest_client_ce.py @@ -1126,9 +1126,6 @@ def unassign_dashboard_from_customer(self, customer_id: CustomerId, dashboard_id dashboard_id=dashboard_id) # Entity Query Controller - def count_entities_by_query(self, body: Optional[EntityCountQuery] = None) -> int: - return self.entity_query_controller.count_entities_by_query_using_post(body=body) - def find_entity_timeseries_and_attributes_keys_by_query(self, timeseries: bool, attributes: bool, body: Optional[ EntityDataQuery]) -> DeferredResultResponseEntity: return self.entity_query_controller.find_entity_timeseries_and_attributes_keys_by_query_using_post(