blob: 8d85b99c0e07de1da615ad82478ac229571984b4 [file] [log] [blame]
Martin Cosyns0efdc872021-09-27 16:24:30 +00001# Copyright 2020-present Open Networking Foundation
2# Original copyright 2020-present ADTRAN, Inc.
3#
4# Licensed under the Apache License, Version 2.0 (the "License");
5# you may not use this file except in compliance with the License.
6# You may obtain a copy of the License at
7#
8# http://www.apache.org/licenses/LICENSE-2.0
9#
10# Unless required by applicable law or agreed to in writing, software
11# distributed under the License is distributed on an "AS IS" BASIS,
12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13# See the License for the specific language governing permissions and
14from robot.api.deco import keyword
15from grpc_robot.services.service import is_connected
16
17from grpc_robot.services.service import Service
18from dmi import hw_management_service_pb2_grpc, hw_management_service_pb2, hw_pb2
19
20
21class NativeHWManagementService(Service):
22
23 prefix = 'hw_management_service_'
24
25 def __init__(self, ctx):
26 super().__init__(ctx=ctx, stub=hw_management_service_pb2_grpc.NativeHWManagementServiceStub)
27
28 @keyword
29 @is_connected
30 def hw_management_service_start_managing_device(self, param_dict, **kwargs):
31 return self._grpc_helper(self.stub.StartManagingDevice, hw_pb2.ModifiableComponent, param_dict, **kwargs)
32
33 @keyword
34 @is_connected
35 def hw_management_service_stop_managing_device(self, param_dict, **kwargs):
36 return self._grpc_helper(self.stub.StopManagingDevice, hw_management_service_pb2.StopManagingDeviceRequest, param_dict, **kwargs)
37
38 @keyword
39 @is_connected
40 def hw_management_service_get_physical_inventory(self, param_dict, **kwargs):
41 return self._grpc_helper(self.stub.GetPhysicalInventory, hw_management_service_pb2.PhysicalInventoryRequest, param_dict, **kwargs)
42
43 @keyword
44 @is_connected
45 def hw_management_service_get_hw_component_info(self, param_dict, **kwargs):
46 return self._grpc_helper(self.stub.GetHWComponentInfo, hw_management_service_pb2.HWComponentInfoGetRequest, param_dict, **kwargs)
47
48 @keyword
49 @is_connected
50 def hw_management_service_set_hw_component_info(self, param_dict, **kwargs):
51 return self._grpc_helper(self.stub.SetHWComponentInfo, hw_management_service_pb2.HWComponentInfoSetRequest, param_dict, **kwargs)
52
53 # rpc GetManagedDevices(google.protobuf.Empty) returns(ManagedDevicesResponse);
54 @keyword
55 @is_connected
56 def hw_management_service_get_managed_devices(self, **kwargs):
57 return self._grpc_helper(self.stub.GetManagedDevices, hw_management_service_pb2.ManagedDevicesResponse, **kwargs)
58
59 # rpc SetLoggingEndpoint(SetLoggingEndpointRequest) returns(SetRemoteEndpointResponse);
60 @keyword
61 @is_connected
62 def hw_management_service_set_logging_endpoint(self, param_dict, **kwargs):
63 return self._grpc_helper(self.stub.SetLoggingEndpoint, hw_management_service_pb2.SetLoggingEndpointRequest, param_dict, **kwargs)
64
65 # rpc GetLoggingEndpoint(Uuid) returns(GetLoggingEndpointResponse);
66 @keyword
67 @is_connected
68 def hw_management_service_get_logging_endpoint(self, param_dict, **kwargs):
69 return self._grpc_helper(self.stub.GetLoggingEndpoint, hw_pb2.Uuid, param_dict, **kwargs)
70
71 # rpc SetMsgBusEndpoint(SetMsgBusEndpointRequest) returns(SetRemoteEndpointResponse);
72 @keyword
73 @is_connected
74 def hw_management_service_set_msg_bus_endpoint(self, param_dict, **kwargs):
75 return self._grpc_helper(self.stub.SetMsgBusEndpoint, hw_management_service_pb2.SetMsgBusEndpointRequest, param_dict, **kwargs)
76
77 # rpc GetMsgBusEndpoint(google.protobuf.Empty) returns(GetMsgBusEndpointResponse);
78 @keyword
79 @is_connected
80 def hw_management_service_get_msg_bus_endpoint(self, **kwargs):
81 return self._grpc_helper(self.stub.GetMsgBusEndpoint, hw_management_service_pb2.GetMsgBusEndpointResponse, **kwargs)