class RegistryApi:
def __init__(self):
global object_list
object_list.append(self)
self.cn = None
self.service = None
def headers(self):
return {
"Authorization": self.cn["sas"],
"Request-Id": str(uuid.uuid4()),
"User-Agent": "azure-edge-e2e",
}
def connect(self, service_connection_string):
self.cn = connection_string.connection_string_to_sas_token(
service_connection_string
)
self.service = IotHubGatewayServiceAPIs("https://" + self.cn["host"]).service
def disconnect(self):
pass
def get_module_twin(self, device_id, module_id):
return self.service.get_module_twin(
device_id, module_id, custom_headers=self.headers()
).as_dict()
def patch_module_twin(self, device_id, module_id, patch):
twin = models.Twin.from_dict(patch)
self.service.update_module_twin(
device_id, module_id, twin, custom_headers=self.headers()
)
class RegistryApi(AbstractRegistryApi):
def __init__(self):
global object_list
object_list.append(self)
self.service = None
self.service_connection_string = None
def headers(self):
cn = connection_string.connection_string_to_sas_token(
self.service_connection_string)
return {
"Authorization": cn["sas"],
"Request-Id": str(uuid.uuid4()),
"User-Agent": "azure-edge-e2e",
}
def connect_sync(self, service_connection_string):
self.service_connection_string = service_connection_string
host = connection_string.connection_string_to_dictionary(
service_connection_string)["HostName"]
self.service = IotHubGatewayServiceAPIs("https://" + host).service
def disconnect_sync(self):
pass
@emulate_async
def get_module_twin(self, device_id, module_id):
return self.service.get_module_twin(
device_id, module_id,
custom_headers=self.headers()).as_dict()["properties"]
@emulate_async
def patch_module_twin(self, device_id, module_id, patch):
twin = models.Twin.from_dict({"properties": patch})
self.service.update_module_twin(device_id,
module_id,
twin,
custom_headers=self.headers())
@emulate_async
def get_device_twin(self, device_id):
return self.service.get_twin(
device_id, custom_headers=self.headers()).as_dict()["properties"]
@emulate_async
def patch_device_twin(self, device_id, patch):
twin = models.Twin.from_dict({"properties": patch})
self.service.update_twin(device_id,
twin,
custom_headers=self.headers())