def wait_private_endpoint_conn(cmd, name, conn_name, resource_group_name=None):
rp = ResourceProvider(cmd)
return rp.get_private_endpoint_conn(
name=name,
resource_group_name=resource_group_name,
conn_name=conn_name,
wait=True)
def _add_endpoint_eventhub(
cmd,
name,
endpoint_name,
eventhub_name,
eventhub_resource_group,
eventhub_policy,
eventhub_namespace,
resource_group_name=None,
endpoint_subscription=None,
tags=None,
):
rp = ResourceProvider(cmd)
return rp.add_endpoint(
name=name,
resource_group_name=resource_group_name,
endpoint_name=endpoint_name,
endpoint_resource_type=ADTEndpointType.eventhub,
endpoint_resource_name=eventhub_name,
endpoint_resource_group=eventhub_resource_group,
endpoint_resource_namespace=eventhub_namespace,
endpoint_resource_policy=eventhub_policy,
endpoint_subscription=endpoint_subscription,
tags=tags,
)
def _add_endpoint_servicebus(
cmd,
name,
endpoint_name,
servicebus_topic_name,
servicebus_resource_group,
servicebus_policy,
servicebus_namespace,
resource_group_name=None,
endpoint_subscription=None,
tags=None,
):
rp = ResourceProvider(cmd)
return rp.add_endpoint(
name=name,
resource_group_name=resource_group_name,
endpoint_name=endpoint_name,
endpoint_resource_type=ADTEndpointType.servicebus,
endpoint_resource_name=servicebus_topic_name,
endpoint_resource_group=servicebus_resource_group,
endpoint_resource_namespace=servicebus_namespace,
endpoint_resource_policy=servicebus_policy,
endpoint_subscription=endpoint_subscription,
tags=tags,
)
def add_endpoint_eventhub(
cmd,
name,
endpoint_name,
eventhub_name,
eventhub_resource_group,
eventhub_namespace,
eventhub_policy=None,
resource_group_name=None,
endpoint_subscription=None,
dead_letter_uri=None,
dead_letter_secret=None,
auth_type=ADTEndpointAuthType.keybased.value,
):
rp = ResourceProvider(cmd)
return rp.add_endpoint(
name=name,
resource_group_name=resource_group_name,
endpoint_name=endpoint_name,
endpoint_resource_type=ADTEndpointType.eventhub.value,
endpoint_resource_name=eventhub_name,
endpoint_resource_group=eventhub_resource_group,
endpoint_resource_namespace=eventhub_namespace,
endpoint_resource_policy=eventhub_policy,
endpoint_subscription=endpoint_subscription,
dead_letter_uri=dead_letter_uri,
dead_letter_secret=dead_letter_secret,
auth_type=auth_type,
)
def add_endpoint_servicebus(
cmd,
name,
endpoint_name,
servicebus_topic_name,
servicebus_resource_group,
servicebus_namespace,
servicebus_policy=None,
resource_group_name=None,
endpoint_subscription=None,
dead_letter_uri=None,
dead_letter_secret=None,
auth_type=ADTEndpointAuthType.keybased.value,
):
rp = ResourceProvider(cmd)
return rp.add_endpoint(
name=name,
resource_group_name=resource_group_name,
endpoint_name=endpoint_name,
endpoint_resource_type=ADTEndpointType.servicebus.value,
endpoint_resource_name=servicebus_topic_name,
endpoint_resource_group=servicebus_resource_group,
endpoint_resource_namespace=servicebus_namespace,
endpoint_resource_policy=servicebus_policy,
endpoint_subscription=endpoint_subscription,
dead_letter_uri=dead_letter_uri,
dead_letter_secret=dead_letter_secret,
auth_type=auth_type,
)
def add_endpoint_eventgrid(
cmd,
name,
endpoint_name,
eventgrid_topic_name,
eventgrid_resource_group,
resource_group_name=None,
endpoint_subscription=None,
dead_letter_uri=None,
dead_letter_secret=None,
auth_type=ADTEndpointAuthType.keybased.value,
):
rp = ResourceProvider(cmd)
return rp.add_endpoint(
name=name,
resource_group_name=resource_group_name,
endpoint_name=endpoint_name,
endpoint_resource_type=ADTEndpointType.eventgridtopic.value,
endpoint_resource_name=eventgrid_topic_name,
endpoint_resource_group=eventgrid_resource_group,
endpoint_subscription=endpoint_subscription,
dead_letter_uri=dead_letter_uri,
dead_letter_secret=dead_letter_secret,
auth_type=auth_type,
)
def assign_role(cmd, name, role_type, assignee, resource_group_name=None):
rp = ResourceProvider(cmd)
return rp.assign_role(
name=name,
role_type=role_type,
assignee=assignee,
resource_group_name=resource_group_name,
)
def delete_private_endpoint_conn(cmd,
name,
conn_name,
resource_group_name=None):
rp = ResourceProvider(cmd)
return rp.delete_private_endpoint_conn(
name=name,
resource_group_name=resource_group_name,
conn_name=conn_name)
def __init__(self, cmd, name, rg=None):
assert cmd
assert name
self.cmd = cmd
self.name = name
self.rg = rg
self.resource_id = DIGITALTWINS_RESOURCE_ID
self.rp = ResourceProvider(self.cmd)
def list_assignments(cmd,
name,
include_inherited=False,
role_type=None,
resource_group_name=None):
rp = ResourceProvider(cmd)
return rp.get_role_assignments(
name=name,
include_inherited=include_inherited,
resource_group_name=resource_group_name,
role_type=role_type,
)
class DigitalTwinsProvider(object):
def __init__(self, cmd, name, rg=None):
assert cmd
assert name
self.cmd = cmd
self.name = name
self.rg = rg
self.resource_id = DIGITALTWINS_RESOURCE_ID
self.rp = ResourceProvider(self.cmd)
def _get_endpoint(self):
host_name = None
https_prefix = "https://"
http_prefix = "http://"
if self.name.lower().startswith(https_prefix):
self.name = self.name[len(https_prefix):]
elif self.name.lower().startswith(http_prefix):
self.name = self.name[len(http_prefix):]
if not all([valid_hostname(self.name), "." in self.name]):
instance = self.rp.find_instance(name=self.name, resource_group_name=self.rg)
host_name = instance.host_name
if not host_name:
raise CLIError("Instance has invalid hostName. Aborting operation...")
else:
host_name = self.name
return "https://{}".format(host_name)
def get_sdk(self):
creds = DigitalTwinAuthentication(cmd=self.cmd, resource_id=self.resource_id)
return AzureDigitalTwinsAPI(base_url=self._get_endpoint(), credentials=creds)
def set_private_endpoint_conn(
cmd,
name,
conn_name,
status,
description=None,
group_ids=None,
actions_required=None,
resource_group_name=None,
):
rp = ResourceProvider(cmd)
return rp.set_private_endpoint_conn(
name=name,
resource_group_name=resource_group_name,
conn_name=conn_name,
status=status,
description=description,
group_ids=group_ids,
actions_required=actions_required,
)
def _add_endpoint_eventgrid(
cmd,
name,
endpoint_name,
eventgrid_topic_name,
eventgrid_resource_group,
resource_group_name=None,
endpoint_subscription=None,
tags=None,
):
rp = ResourceProvider(cmd)
return rp.add_endpoint(
name=name,
resource_group_name=resource_group_name,
endpoint_name=endpoint_name,
endpoint_resource_type=ADTEndpointType.eventgridtopic,
endpoint_resource_name=eventgrid_topic_name,
endpoint_resource_group=eventgrid_resource_group,
endpoint_subscription=endpoint_subscription,
tags=tags,
)
def _add_endpoint_eventgrid(
cmd,
name,
endpoint_name,
eventgrid_topic_name,
eventgrid_resource_group,
timeout=15,
resource_group_name=None,
tags=None,
):
rp = ResourceProvider(cmd)
return rp.add_endpoint(
name=name,
resource_group_name=resource_group_name,
endpoint_name=endpoint_name,
endpoint_resource_type=ADTEndpointType.eventgridtopic,
endpoint_resource_name=eventgrid_topic_name,
endpoint_resource_group=eventgrid_resource_group,
tags=tags,
timeout=timeout,
)
def create_instance(
cmd,
name,
resource_group_name,
location=None,
tags=None,
assign_identity=None,
scopes=None,
role_type="Contributor",
public_network_access=ADTPublicNetworkAccessType.enabled.value,
):
rp = ResourceProvider(cmd)
return rp.create(
name=name,
resource_group_name=resource_group_name,
location=location,
tags=tags,
assign_identity=assign_identity,
scopes=scopes,
role_type=role_type,
public_network_access=public_network_access,
)
class DigitalTwinsProvider(object):
def __init__(self, cmd, name, rg=None):
assert cmd
assert name
self.cmd = cmd
self.name = name
self.rg = rg
self.resource_id = DIGITALTWINS_RESOURCE_ID
self.rp = ResourceProvider(self.cmd)
def _get_endpoint(self):
host_name = None
https_prefix = "https://"
http_prefix = "http://"
if self.name.lower().startswith(https_prefix):
self.name = self.name[len(https_prefix):]
elif self.name.lower().startswith(http_prefix):
self.name = self.name[len(http_prefix):]
if not all([valid_hostname(self.name), "." in self.name]):
instance = self.rp.find_instance(name=self.name,
resource_group_name=self.rg)
host_name = instance.host_name
if not host_name:
raise CLIError(
"Instance has invalid hostName. Aborting operation...")
else:
host_name = self.name
return "https://{}".format(host_name)
def get_sdk(self):
from azure.cli.core.commands.client_factory import get_mgmt_service_client
client = get_mgmt_service_client(
cli_ctx=self.cmd.cli_ctx,
client_or_resource_type=AzureDigitalTwinsAPI,
base_url=self._get_endpoint(),
resource=self.resource_id,
subscription_bound=False,
base_url_bound=False,
)
client.config.add_user_agent(USER_AGENT)
return client
class DigitalTwinsProvider(object):
def __init__(self, cmd, name, rg=None):
assert cmd
assert name
self.cmd = cmd
self.name = name
self.rg = rg
self.resource_id = DIGITALTWINS_RESOURCE_ID
self.rp = ResourceProvider(self.cmd)
def _get_endpoint(self):
instance = self.rp.find_instance(name=self.name, resource_group_name=self.rg)
host_name = instance.host_name
if not host_name:
raise CLIError("Retrieved hostName was null which is invalid.")
return "https://{}".format(instance.host_name)
def get_sdk(self):
creds = DigitalTwinAuthentication(cmd=self.cmd, resource_id=self.resource_id)
return AzureDigitalTwinsAPI(base_url=self._get_endpoint(), credentials=creds)
def delete_endpoint(cmd, name, endpoint_name, resource_group_name=None):
rp = ResourceProvider(cmd)
return rp.delete_endpoint(name=name,
endpoint_name=endpoint_name,
resource_group_name=resource_group_name)
def show_endpoint(cmd, name, endpoint_name, resource_group_name=None):
rp = ResourceProvider(cmd)
return rp.get_endpoint(name=name,
endpoint_name=endpoint_name,
resource_group_name=resource_group_name)
def list_endpoints(cmd, name, resource_group_name=None):
rp = ResourceProvider(cmd)
return rp.list_endpoints(name=name,
resource_group_name=resource_group_name)
def delete_instance(cmd, name, resource_group_name=None):
rp = ResourceProvider(cmd)
return rp.delete(name=name, resource_group_name=resource_group_name)
def show_instance(cmd, name, resource_group_name=None):
rp = ResourceProvider(cmd)
return rp.find_instance(name=name, resource_group_name=resource_group_name)
def show_private_link(cmd, name, link_name, resource_group_name=None):
rp = ResourceProvider(cmd)
return rp.get_private_link(name=name,
resource_group_name=resource_group_name,
link_name=link_name)
def list_private_links(cmd, name, resource_group_name=None):
rp = ResourceProvider(cmd)
return rp.list_private_links(name=name,
resource_group_name=resource_group_name)
def create_instance(cmd, name, resource_group_name, location=None, tags=None):
rp = ResourceProvider(cmd)
return rp.create(name=name,
resource_group_name=resource_group_name,
location=location,
tags=tags)
def wait_instance(cmd, name, resource_group_name=None):
rp = ResourceProvider(cmd)
return rp.find_instance(name=name,
resource_group_name=resource_group_name,
wait=True)
def list_instances(cmd, resource_group_name=None):
rp = ResourceProvider(cmd)
if not resource_group_name:
return rp.list()
return rp.list_by_resouce_group(resource_group_name)
def dt_rp(fixture_cmd2):
rp = ResourceProvider(fixture_cmd2)
yield rp