tb-rest-client 4.1.0__py3-none-any.whl → 4.3__py3-none-any.whl

tb_rest_client/rest_client_base.py

@@ -1,4 +1,4 @@
-#  Copyright 2025. ThingsBoard
+#  Copyright 2026. ThingsBoard
 #
 #  Licensed under the Apache License, Version 2.0 (the "License");
 #  you may not use this file except in compliance with the License.
@@ -70,7 +70,6 @@ from tb_rest_client.api.api_ce.two_factor_auth_config_controller_api import TwoF
 from tb_rest_client.api.api_ce.device_connectivity_controller_api import DeviceConnectivityControllerApi
 from tb_rest_client.api.api_ce.mail_config_template_controller_api import MailConfigTemplateControllerApi
 from tb_rest_client.api.api_ce.image_controller_api import ImageControllerApi
-from tb_rest_client.api.api_ce.mobile_application_controller_api import MobileApplicationControllerApi
 from tb_rest_client.api.api_ce.queue_stats_controller_api import QueueStatsControllerApi
 from tb_rest_client.api.api_ce.domain_controller_api import DomainControllerApi
 from tb_rest_client.api.api_ce import MobileAppControllerApi, TrendzControllerApi, JobControllerApi
@@ -78,6 +77,7 @@ from tb_rest_client.api.api_ce import MobileAppBundleControllerApi
 from tb_rest_client.api.api_ce import RuleEngineControllerApi
 from tb_rest_client.api.api_ce import QrCodeSettingsControllerApi
 from tb_rest_client.api.api_ce import CalculatedFieldControllerApi
+from tb_rest_client.api.api_ce import AiModelControllerApi
 # from tb_rest_client.models.models_pe import *
 from tb_rest_client.configuration import Configuration
 from tb_rest_client.api_client import ApiClient
@@ -143,6 +143,11 @@ class RestClientBase(Thread):
         self.__save_token(token_json)
         self.__load_configuration()
 
+    def api_key_login(self, api_key: str):
+        self.configuration.api_key_prefix["X-Authorization"] = "ApiKey"
+        self.configuration.api_key["X-Authorization"] = api_key
+        self.__load_configuration()
+
     def token_login(self, token, refresh_token=None):
         token_json = {
             "token": token,
@@ -771,6 +776,10 @@ class RestClientBase(Thread):
                                   update_related: Optional[bool] = None) -> RuleChainMetaData:
         return self.rule_chain_controller.save_rule_chain_meta_data_using_post(body=body, update_related=update_related)
 
+    def get_rule_chains_by_ids(self, rule_chain_ids: List[str]) -> List[str]:
+        rule_chain_ids = ','.join(rule_chain_ids)
+        return self.rule_chain_controller.get_rule_chains_by_ids(rule_chain_ids=rule_chain_ids)
+
     def delete_rpc(self, rpc_id: RpcId) -> None:
         rpc_id = self.get_id(rpc_id)
         return self.rpc_v2_controller.delete_rpc_using_delete(rpc_id=rpc_id)
@@ -801,6 +810,10 @@ class RestClientBase(Thread):
         customer_id = self.get_id(customer_id)
         return self.customer_controller.delete_customer_using_delete(customer_id=customer_id)
 
+    def get_customers_by_ids(self, customer_ids: List[str]) -> list:
+        customer_ids = ','.join(customer_ids)
+        return self.customer_controller.get_customers_by_ids(customer_ids=customer_ids)
+
     # User Controller #
     def get_user_token(self, user_id: UserId) -> JwtPair:
         user_id = self.get_id(user_id)
@@ -848,6 +861,9 @@ class RestClientBase(Thread):
         user_id = self.get_id(user_id)
         return self.user_controller.get_activation_link_using_get(user_id=user_id)
 
+    def get_users_by_ids(self, user_ids: list) -> List[User]:
+        return self.user_controller.get_users_by_ids(user_ids=str(user_ids))
+
     # Queue Controller
     def get_tenant_queues_by_service_type(self, service_type: str, page_size: int, page: int,
                                           type: Optional[str] = None,
@@ -1050,6 +1066,10 @@ class RestClientBase(Thread):
                                                                                sort_property=sort_property,
                                                                                sort_order=sort_order)
 
+    def get_entity_views_by_ids(self, entity_view_ids: list[str]):
+        entity_view_ids = ','.join(entity_view_ids)
+        return self.entity_view_controller.get_entity_views_by_ids(entity_view_ids=entity_view_ids)
+
     # Admin Controller
     def send_test_sms(self, body: Optional[TestSmsRequest] = None) -> None:
         return self.admin_controller.send_test_sms_using_post(body=body)
@@ -1141,6 +1161,9 @@ class RestClientBase(Thread):
                                                                           sort_property=sort_property,
                                                                           sort_order=sort_order)
 
+    def get_system_or_tenant_resources_by_ids(self, resource_ids: list[EntityId]) -> PageDataTbResourceInfo:
+        return self.tb_resource_controller.get_system_or_tenant_resources_by_ids(resource_ids=str(resource_ids))
+
     def code_processing_url(self, code: str, state: str):
         return self.admin_controller.code_processing_url_using_get(code=code, state=state)
 
@@ -1183,6 +1206,19 @@ class RestClientBase(Thread):
                                                                         key=key,
                                                                         if_none_match=if_none_match)
 
+    def upload_resource(self, title, resource_type, descriptor, resource_sub_type, file) -> TbResourceInfo:
+        return self.tb_resource_controller.upload_resource(title=title, resource_type=resource_type,
+                                                           descriptor=descriptor, resource_sub_type=resource_sub_type,
+                                                           file=file)
+
+    def update_resource_info(self, body: TbResourceInfo, resource_id: EntityId) -> TbResourceInfo:
+        resource_id = self.get_id(resource_id)
+        return self.tb_resource_controller.update_resource_info(id=resource_id, body=body)
+
+    def update_resource_data(self, id: EntityId, file: str) -> TbResourceInfo:
+        id = self.get_id(id)
+        return self.tb_resource_controller.update_resource_data(id=id, file=file)
+
     def get_repository_settings_info(self) -> RepositorySettingsInfo:
         return self.admin_controller.get_repository_settings_info_using_get()
 
@@ -1296,6 +1332,10 @@ class RestClientBase(Thread):
     def get_widgets_bundles(self, ):
         return self.widgets_bundle_controller.get_widgets_bundles_using_get()
 
+    def get_widgets_bundles_by_ids(self, widget_bundle_ids: List[str]) -> List[WidgetsBundle]:
+        widget_bundle_ids = ','.join(widget_bundle_ids)
+        return self.widgets_bundle_controller.get_widgets_bundles_by_ids(widgets_bundle_ids=widget_bundle_ids)
+
     # Device Profile Controller
     def get_device_profile_infos(self, page_size: int, page: int, text_search: Optional[str] = None,
                                  sort_property: Optional[str] = None, sort_order: Optional[str] = None,
@@ -1428,6 +1468,11 @@ class RestClientBase(Thread):
     def get_edqs_state(self) -> EdqsState:
         return self.entity_query_controller.get_edqs_state()
 
+    def find_available_entity_keys_by_query(self, body: EntityDataQuery, timeseries: bool, attributes: bool,
+                                            scope: Optional[str] = None) -> AvailableEntityKeys:
+        return self.entity_query_controller.find_available_entity_keys_by_query(body=body, timeseries=timeseries,
+                                                                                attributes=attributes, scope=scope)
+
     # Widget Type Controller
     def get_bundle_widget_type_fqns(self, widgets_bundle_id: WidgetsBundleId) -> List[str]:
         widgets_bundle_id = self.get_id(widgets_bundle_id)
@@ -1665,6 +1710,10 @@ class RestClientBase(Thread):
         tenant_id = self.get_id(tenant_id)
         return self.tenant_controller.delete_tenant_using_delete(tenant_id=tenant_id)
 
+    def get_tenants_by_ids(self, tenant_ids: list) -> List[Tenant]:
+        tenant_ids = ','.join(tenant_ids)
+        return self.tenant_controller.get_tenants_by_ids(tenant_ids=str(tenant_ids))
+
     # OTA Package Controller
     def delete_ota_package(self, ota_package_id: OtaPackageId) -> None:
         ota_package_id = self.get_id(ota_package_id)
@@ -1800,6 +1849,9 @@ class RestClientBase(Thread):
     def request_two_fa_verification_code(self, provider_type: str) -> None:
         return self.two_factor_auth_controller.request_two_fa_verification_code_using_post(provider_type=provider_type)
 
+    def authenticate_by_two_fa_configuration_token(self) -> JwtPair:
+        return self.two_factor_auth_controller.authenticate_by_two_fa_configuration_token()
+
     # Tow Factor Auth Config Controller
     def delete_two_fa_account_config(self, provider_type: str) -> AccountTwoFaSettings:
         return self.two_factor_auth_config_controller.delete_two_fa_account_config_using_delete(provider_type=provider_type)
@@ -1890,6 +1942,9 @@ class RestClientBase(Thread):
     def save_user_notification_settings(self, body: UserNotificationSettings) -> UserNotificationSettings:
         return self.notification_controller.save_user_notification_settings_using_post(body=body)
 
+    def send_entities_limit_increase_request(self, entity_type: str):
+        return self.notification_controller.send_entities_limit_increase_request(entity_type=entity_type)
+
     def delete_notification_rule(self, id: str):
         return self.notification_rule_controller.delete_notification_rule_using_delete(id=id)
 
@@ -2027,6 +2082,10 @@ class RestClientBase(Thread):
     def get_asset_profile_by_names(self, active_only: Optional[bool]) -> List[EntityInfo]:
         return self.asset_profile_controller.get_asset_profile_names_using_get(active_only=active_only)
 
+    def get_asset_profiles_by_ids(self, asset_profile_ids: List[str]):
+        asset_profile_ids = ','.join(asset_profile_ids)
+        return self.asset_profile_controller.get_asset_profiles_by_ids(asset_profile_ids=asset_profile_ids)
+
     def download_gateway_docker_compose(self, device_id: DeviceId) -> Resource:
         device_id = self.get_id(device_id)
         return self.device_connectivity_controller.download_gateway_docker_compose_using_get(device_id=device_id)
@@ -2045,21 +2104,6 @@ class RestClientBase(Thread):
         edge_id = self.get_id(edge_id)
         return self.edge_controller.is_edge_upgrade_available_using_get(edge_id=edge_id)
 
-    def get_application_redirect(self, user_agent: str):
-        return self.mobile_application_controller.get_application_redirect(user_agent=user_agent)
-
-    def get_mobile_app_deep_link(self) -> str:
-        return self.mobile_application_controller.get_mobile_app_deep_link()
-
-    def get_mobile_app_settings(self) -> MobileAppSettings:
-        return self.mobile_application_controller.get_mobile_app_settings()
-
-    def get_user_token_by_mobile_secret(self, secret: str) -> JwtPair:
-        return self.mobile_application_controller.get_user_token_by_mobile_secret(secret=secret)
-
-    def save_mobile_app_settings(self, body: MobileAppSettings) -> MobileAppSettings:
-        return self.mobile_application_controller.save_mobile_app_settings(body=body)
-
     def get_queue_stats_by_id(self, queue_stats_id: QueueId) -> QueueStats:
         queue_stats_id = self.get_id(queue_stats_id)
         return self.queue_stats_controller.get_queue_stats_by_id(queue_stats_id=queue_stats_id)
@@ -2128,6 +2172,24 @@ class RestClientBase(Thread):
     def mobile_app_update_oauth2_clients(self, body: List[str], id: str):
         return self.mobile_app_controller.update_oauth2_clients(body=body, id=id)
 
+    def get_login_mobile_info(self, pkg_name: str, platform: str) -> LoginMobileInfo:
+        return self.mobile_app_controller.get_login_mobile_info(pkg_name=pkg_name, platform=platform)
+
+    def get_mobile_app_by_id(self, mobile_app_id: MobileAppId) -> MobileApp:
+        mobile_app_id = self.get_id(mobile_app_id)
+        return self.mobile_app_controller.get_mobile_app_by_id(id=mobile_app_id)
+
+    def get_tenant_mobile_apps(self, page_size: int, page: int, text_search: Optional[str] = None,
+                               sort_property: Optional[str] = None,
+                               sort_order: Optional[str] = None) -> PageDataMobileApp:
+        return self.mobile_app_controller.get_tenant_mobile_apps(page_size=page_size, page=page,
+                                                                 text_search=text_search,
+                                                                 sort_property=sort_property,
+                                                                 sort_order=sort_order)
+
+    def get_user_mobile_info(self, pkg_name: str, platform: str) -> UserMobileInfo:
+        return self.mobile_app_controller.get_user_mobile_info(pkg_name=pkg_name, platform=platform)
+
     def get_qr_code_settings(self) -> QrCodeSettings:
         return self.qr_code_settings_controller.get_qr_code_settings()
 
@@ -2184,6 +2246,14 @@ class RestClientBase(Thread):
     def save_calculated_field(self, body: CalculatedField) -> CalculatedField:
         return self.calculated_field_controller.save_calculated_field(body=body)
 
+    def get_calculated_field_names(self, calculated_filed_type: str, page_size: int, page: int,
+                                   text_search: Optional[str] = None,
+                                   sort_order: Optional[str] = None) -> PageDataString:
+        return self.calculated_field_controller.get_calculated_field_names(type=calculated_filed_type,
+                                                                           page_size=page_size, page=page,
+                                                                           text_search=text_search,
+                                                                           sort_order=sort_order)
+
     # Trendz Controller
     def get_trendz_settings(self) -> TrendzSettings:
         return self.trendz_controller.get_trendz_settings()
@@ -2214,6 +2284,26 @@ class RestClientBase(Thread):
         id = self.get_id(id)
         return self.job_controller.reprocess_job(id=id)
 
+    # AI Model Controller
+    def delete_ai_model_by_id(self, ai_model_id: AiModelId) -> None:
+        ai_model_id = self.get_id(ai_model_id)
+        return self.ai_model_controller.delete_ai_model_by_id(model_uuid=ai_model_id)
+
+    def get_ai_model_by_id(self, ai_model_id: AiModelId) -> AiModel:
+        ai_model_id = self.get_id(ai_model_id)
+        return self.ai_model_controller.get_ai_model_by_id(model_uuid=ai_model_id)
+
+    def get_ai_models(self, page_size: int, page: int, text_search: Optional[str] = None,
+                        sort_property: Optional[str] = None, sort_order: Optional[str] = None) -> PageDataAiModel:
+        return self.ai_model_controller.get_ai_models(page_size=page_size, page=page, text_search=text_search,
+                                                     sort_property=sort_property, sort_order=sort_order)
+
+    def save_ai_model(self, body: AiModel) -> AiModel:
+        return self.ai_model_controller.save_ai_model(body=body)
+
+    def send_chat_request(self, body: TbChatRequest) -> None:
+        return self.ai_model_controller.send_chat_request(body=body)
+
     def __load_controllers(self):
         self.audit_log_controller = AuditLogControllerApi(self.api_client)
         self.o_auth2_config_template_controller = OAuth2ConfigTemplateControllerApi(self.api_client)
@@ -2261,7 +2351,6 @@ class RestClientBase(Thread):
         self.device_connectivity_controller = DeviceConnectivityControllerApi(self.api_client)
         self.mail_config_template_controller = MailConfigTemplateControllerApi(self.api_client)
         self.image_controller = ImageControllerApi(self.api_client)
-        self.mobile_application_controller = MobileApplicationControllerApi(self.api_client)
         self.queue_stats_controller = QueueStatsControllerApi(self.api_client)
         self.domain_controller = DomainControllerApi(self.api_client)
         self.mobile_app_controller = MobileAppControllerApi(self.api_client)
@@ -2271,6 +2360,7 @@ class RestClientBase(Thread):
         self.calculated_field_controller = CalculatedFieldControllerApi(self.api_client)
         self.trendz_controller = TrendzControllerApi(self.api_client)
         self.job_controller = JobControllerApi(self.api_client)
+        self.ai_model_controller = AiModelControllerApi(self.api_client)
 
     @staticmethod
     def get_type(type):

tb_rest_client/rest_client_ce.py

@@ -1,4 +1,4 @@
-#  Copyright 2025. ThingsBoard
+#  Copyright 2026. ThingsBoard
 #
 #  Licensed under the Apache License, Version 2.0 (the "License");
 #  you may not use this file except in compliance with the License.
@@ -974,6 +974,10 @@ class RestClientCE(RestClientBase):
                                                                             sort_property=sort_property,
                                                                             sort_order=sort_order)
 
+    def get_device_profile_infos_by_ids(self, device_profile_ids: list[str]):
+        device_profile_ids = ','.join(device_profile_ids)
+        return self.device_profile_controller.get_device_profile_infos_by_ids(device_profile_ids=device_profile_ids)
+
     # Dashboard Controller
     def add_dashboard_customers(self, dashboard_id: DashboardId, body: Optional[List[str]] = None) -> Dashboard:
         dashboard_id = self.get_id(dashboard_id)