semantic-link-labs 0.8.11__py3-none-any.whl → 0.9.1__py3-none-any.whl

sempy_labs/_capacities.py

@@ -1,11 +1,14 @@
 import sempy.fabric as fabric
-from typing import Optional, List
+from typing import Optional, List, Tuple
 from sempy._utils._log import log
 import sempy_labs._icons as icons
 from sempy.fabric.exceptions import FabricHTTPException
 import requests
-from sempy_labs._helper_functions import _get_azure_token_credentials
 import pandas as pd
+from sempy_labs._authentication import _get_headers, ServicePrincipalTokenProvider
+from uuid import UUID
+from sempy_labs._helper_functions import _is_valid_uuid
+import sempy_labs._authentication as auth
 
 
 def _add_sll_tag(payload, tags):
@@ -24,20 +27,19 @@ def _add_sll_tag(payload, tags):
 def create_fabric_capacity(
     capacity_name: str,
     azure_subscription_id: str,
-    key_vault_uri: str,
-    key_vault_tenant_id: str,
-    key_vault_client_id: str,
-    key_vault_client_secret: str,
     resource_group: str,
     region: str,
     sku: str,
     admin_members: str | List[str],
     tags: Optional[dict] = None,
+    **kwargs,
 ):
     """
     This function creates a new Fabric capacity within an Azure subscription.
 
-    This is a wrapper function for the following API: `Fabric Capacities - Create Or Update <https://learn.microsoft.com/rest/api/microsoftfabric/fabric-capacities/create-or-update?view=rest-microsoftfabric-2023-11-01>`_.
+    This is a wrapper function for the following API: `Fabric Capacities - Create Or Update <https://learn.microsoft.com/rest/api/microsoftfabric/fabric-capacities/create-or-update>`_.
+
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
 
     Parameters
     ----------
@@ -45,14 +47,6 @@ def create_fabric_capacity(
         Name of the Fabric capacity.
     azure_subscription_id : str
         The Azure subscription ID.
-    key_vault_uri : str
-        The name of the `Azure key vault <https://azure.microsoft.com/products/key-vault>`_ URI. Example: "https://<Key Vault Name>.vault.azure.net/"
-    key_vault_tenant_id : str
-        The name of the Azure key vault secret storing the Tenant ID.
-    key_vault_client_id : str
-        The name of the Azure key vault secret storing the Client ID.
-    key_vault_client_secret : str
-        The name of the Azure key vault secret storing the Client Secret.
     resource_group : str
         The name of the Azure resource group.
     region : str
@@ -65,8 +59,6 @@ def create_fabric_capacity(
         Tag(s) to add to the capacity. Example: {'tagName': 'tagValue'}.
     """
 
-    from azure.mgmt.resource import ResourceManagementClient
-
     if isinstance(admin_members, str):
         admin_members = [admin_members]
 
@@ -143,29 +135,41 @@ def create_fabric_capacity(
             f"{icons.red_dot} Invalid region. Valid options: {valid_regions}."
         )
 
-    azure_token, credential, headers = _get_azure_token_credentials(
-        key_vault_uri=key_vault_uri,
-        key_vault_tenant_id=key_vault_tenant_id,
-        key_vault_client_id=key_vault_client_id,
-        key_vault_client_secret=key_vault_client_secret,
-    )
-
-    resource_client = ResourceManagementClient(credential, azure_subscription_id)
+    token_provider = auth.token_provider.get()
+    if token_provider is None:
+        token_provider = ServicePrincipalTokenProvider.from_azure_key_vault(
+            key_vault_uri=kwargs["key_vault_uri"],
+            key_vault_tenant_id=kwargs["key_vault_tenant_id"],
+            key_vault_client_id=kwargs["key_vault_client_id"],
+            key_vault_client_secret=kwargs["key_vault_client_secret"],
+        )
+        print(
+            f"{icons.info} Please use the 'token_provider' parameter instead of the key vault parameters within this function as the key vault parameters have been deprecated."
+        )
+    headers = _get_headers(token_provider, audience="azure")
 
     if resource_group is None:
-        for i in resource_client.resources.list(
-            "resourceType eq 'Microsoft.PowerBIDedicated/capacities'"
-        ):
-            if i.name == capacity_name.removesuffix(icons.migrate_capacity_suffix):
-                resource_group = i.id.split("/")[4]
-                print(
-                    f"{icons.yellow_dot} Override resource group flag detected for A SKUs - using the existing resource group '{resource_group}' for the '{capacity_name}' capacity."
-                )
+        dfRG = list_resource_groups(
+            azure_subscription_id=azure_subscription_id,
+            filter="resourceType eq 'Microsoft.PowerBIDedicated/capacities'",
+        )
+        dfRG_filt = dfRG[
+            dfRG["Resource Group Name"]
+            == capacity_name.removesuffix(icons.migrate_capacity_suffix)
+        ]
+        if not dfRG_filt.empty:
+            resource_group = dfRG_filt["Resource Group Name"].iloc[0]
+            print(
+                f"{icons.yellow_dot} Override resource group flag detected for A SKUs - using the existing resource group '{resource_group}' for the '{capacity_name}' capacity."
+            )
     else:
         # Attempt to get the resource group
         try:
-            rg = resource_client.resource_groups.get(resource_group)
-            if rg.location != region:
+            dfRG = get_resource_group(
+                azure_subscription_id=azure_subscription_id,
+                resource_group=resource_group,
+            )
+            if dfRG["Location"].iloc[0] != region:
                 print(
                     f"{icons.yellow_dot} The '{resource_group}' resource group exists, but in a different region."
                 )
@@ -177,11 +181,10 @@ def create_fabric_capacity(
             print(
                 f"{icons.in_progress} Creating the '{resource_group}' resource group in the '{region}' region"
             )
-            rg_result = resource_client.resource_groups.create_or_update(
-                resource_group, {"location": region}
-            )
-            print(
-                f"{icons.green_dot} Provisioned resource group with ID: {rg_result.id}"
+            create_or_update_resource_group(
+                azure_subscription_id=azure_subscription_id,
+                resource_group=resource_group,
+                region=region,
             )
 
     payload = {
@@ -243,20 +246,20 @@ def get_capacity_resource_governance(capacity_name: str):
     return response.json()["workloadSettings"]
 
 
+@log
 def suspend_fabric_capacity(
     capacity_name: str,
     azure_subscription_id: str,
     resource_group: str,
-    key_vault_uri: str,
-    key_vault_tenant_id: str,
-    key_vault_client_id: str,
-    key_vault_client_secret: str,
+    **kwargs,
 ):
     """
     This function suspends a Fabric capacity.
 
     This is a wrapper function for the following API: `Fabric Capacities - Suspend <https://learn.microsoft.com/rest/api/microsoftfabric/fabric-capacities/suspend?view=rest-microsoftfabric-2023-11-01>`_.
 
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     capacity_name : str
@@ -265,23 +268,21 @@ def suspend_fabric_capacity(
         The Azure subscription ID.
     resource_group : str
         The name of the Azure resource group.
-    key_vault_uri : str
-        The name of the `Azure key vault <https://azure.microsoft.com/products/key-vault>`_ URI. Example: "https://<Key Vault Name>.vault.azure.net/"
-    key_vault_tenant_id : str
-        The name of the Azure key vault secret storing the Tenant ID.
-    key_vault_client_id : str
-        The name of the Azure key vault secret storing the Client ID.
-    key_vault_client_secret : str
-        The name of the Azure key vault secret storing the Client Secret.
-    """
-    # https://learn.microsoft.com/en-us/rest/api/microsoftfabric/fabric-capacities/suspend?view=rest-microsoftfabric-2023-11-01&tabs=HTTP
-
-    azure_token, credential, headers = _get_azure_token_credentials(
-        key_vault_uri=key_vault_uri,
-        key_vault_tenant_id=key_vault_tenant_id,
-        key_vault_client_id=key_vault_client_id,
-        key_vault_client_secret=key_vault_client_secret,
-    )
+    """
+
+    token_provider = auth.token_provider.get()
+    if token_provider is None:
+        token_provider = ServicePrincipalTokenProvider.from_azure_key_vault(
+            key_vault_uri=kwargs["key_vault_uri"],
+            key_vault_tenant_id=kwargs["key_vault_tenant_id"],
+            key_vault_client_id=kwargs["key_vault_client_id"],
+            key_vault_client_secret=kwargs["key_vault_client_secret"],
+        )
+        print(
+            f"{icons.info} Please use the 'token_provider' parameter instead of the key vault parameters within this function as the key vault parameters have been deprecated."
+        )
+
+    headers = _get_headers(token_provider, audience="azure")
 
     url = f"https://management.azure.com/subscriptions/{azure_subscription_id}/resourceGroups/{resource_group}/providers/Microsoft.Fabric/capacities/{capacity_name}/suspend?api-version={icons.azure_api_version}"
 
@@ -293,20 +294,20 @@ def suspend_fabric_capacity(
     print(f"{icons.green_dot} The '{capacity_name}' capacity has been suspended.")
 
 
+@log
 def resume_fabric_capacity(
     capacity_name: str,
     azure_subscription_id: str,
     resource_group: str,
-    key_vault_uri: str,
-    key_vault_tenant_id: str,
-    key_vault_client_id: str,
-    key_vault_client_secret: str,
+    **kwargs,
 ):
     """
     This function resumes a Fabric capacity.
 
     This is a wrapper function for the following API: `Fabric Capacities - Resume <https://learn.microsoft.com/rest/api/microsoftfabric/fabric-capacities/resume?view=rest-microsoftfabric-2023-11-01>`_.
 
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     capacity_name : str
@@ -315,24 +316,21 @@ def resume_fabric_capacity(
         The Azure subscription ID.
     resource_group : str
         The name of the Azure resource group.
-    key_vault_uri : str
-        The name of the `Azure key vault <https://azure.microsoft.com/products/key-vault>`_ URI. Example: "https://<Key Vault Name>.vault.azure.net/"
-    key_vault_tenant_id : str
-        The name of the Azure key vault secret storing the Tenant ID.
-    key_vault_client_id : str
-        The name of the Azure key vault secret storing the Client ID.
-    key_vault_client_secret : str
-        The name of the Azure key vault secret storing the Client Secret.
-    """
-
-    # https://learn.microsoft.com/en-us/rest/api/microsoftfabric/fabric-capacities/resume?view=rest-microsoftfabric-2023-11-01&tabs=HTTP
-
-    azure_token, credential, headers = _get_azure_token_credentials(
-        key_vault_uri=key_vault_uri,
-        key_vault_tenant_id=key_vault_tenant_id,
-        key_vault_client_id=key_vault_client_id,
-        key_vault_client_secret=key_vault_client_secret,
-    )
+    """
+
+    token_provider = auth.token_provider.get()
+    if token_provider is None:
+        token_provider = ServicePrincipalTokenProvider.from_azure_key_vault(
+            key_vault_uri=kwargs["key_vault_uri"],
+            key_vault_tenant_id=kwargs["key_vault_tenant_id"],
+            key_vault_client_id=kwargs["key_vault_client_id"],
+            key_vault_client_secret=kwargs["key_vault_client_secret"],
+        )
+        print(
+            f"{icons.info} Please use the 'token_provider' parameter instead of the key vault parameters within this function as the key vault parameters have been deprecated."
+        )
+
+    headers = _get_headers(token_provider, audience="azure")
 
     url = f"https://management.azure.com/subscriptions/{azure_subscription_id}/resourceGroups/{resource_group}/providers/Microsoft.Fabric/capacities/{capacity_name}/resume?api-version={icons.azure_api_version}"
 
@@ -344,18 +342,18 @@ def resume_fabric_capacity(
     print(f"{icons.green_dot} The '{capacity_name}' capacity has been resumed.")
 
 
+@log
 def delete_embedded_capacity(
     capacity_name: str,
     azure_subscription_id: str,
     resource_group: str,
-    key_vault_uri: str,
-    key_vault_tenant_id: str,
-    key_vault_client_id: str,
-    key_vault_client_secret: str,
+    **kwargs,
 ):
     """
     This function deletes a Power BI Embedded capacity.
 
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     capacity_name : str
@@ -364,24 +362,21 @@ def delete_embedded_capacity(
         The Azure subscription ID.
     resource_group : str
         The name of the Azure resource group.
-    key_vault_uri : str
-        The name of the `Azure key vault <https://azure.microsoft.com/products/key-vault>`_ URI. Example: "https://<Key Vault Name>.vault.azure.net/"
-    key_vault_tenant_id : str
-        The name of the Azure key vault secret storing the Tenant ID.
-    key_vault_client_id : str
-        The name of the Azure key vault secret storing the Client ID.
-    key_vault_client_secret : str
-        The name of the Azure key vault secret storing the Client Secret.
-    """
-
-    # https://learn.microsoft.com/en-us/rest/api/power-bi-embedded/capacities/delete?view=rest-power-bi-embedded-2021-01-01&tabs=HTTP
-
-    azure_token, credential, headers = _get_azure_token_credentials(
-        key_vault_uri=key_vault_uri,
-        key_vault_tenant_id=key_vault_tenant_id,
-        key_vault_client_id=key_vault_client_id,
-        key_vault_client_secret=key_vault_client_secret,
-    )
+    """
+
+    token_provider = auth.token_provider.get()
+    if token_provider is None:
+        token_provider = ServicePrincipalTokenProvider.from_azure_key_vault(
+            key_vault_uri=kwargs["key_vault_uri"],
+            key_vault_tenant_id=kwargs["key_vault_tenant_id"],
+            key_vault_client_id=kwargs["key_vault_client_id"],
+            key_vault_client_secret=kwargs["key_vault_client_secret"],
+        )
+        print(
+            f"{icons.info} Please use the 'token_provider' parameter instead of the key vault parameters within this function as the key vault parameters have been deprecated."
+        )
+
+    headers = _get_headers(token_provider, audience="azure")
 
     url = f"https://management.azure.com/subscriptions/{azure_subscription_id}/resourceGroups/{resource_group}/providers/Microsoft.PowerBIDedicated/capacities/{capacity_name}?api-version={icons.azure_api_version}"
 
@@ -393,6 +388,7 @@ def delete_embedded_capacity(
     print(f"{icons.green_dot} The '{capacity_name}' capacity has been deleted.")
 
 
+@log
 def delete_premium_capacity(capacity_name: str):
     """
     This function deletes a Power BI Premium capacity.
@@ -421,20 +417,20 @@ def delete_premium_capacity(capacity_name: str):
     print(f"{icons.green_dot} The '{capacity_name}' capacity has been deleted.")
 
 
+@log
 def delete_fabric_capacity(
     capacity_name: str,
     azure_subscription_id: str,
     resource_group: str,
-    key_vault_uri: str,
-    key_vault_tenant_id: str,
-    key_vault_client_id: str,
-    key_vault_client_secret: str,
+    **kwargs,
 ):
     """
     This function deletes a Fabric capacity.
 
     This is a wrapper function for the following API: `Fabric Capacities - Delete <https://learn.microsoft.com/rest/api/microsoftfabric/fabric-capacities/delete?view=rest-microsoftfabric-2023-11-01>`_.
 
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     capacity_name : str
@@ -443,24 +439,21 @@ def delete_fabric_capacity(
         The Azure subscription ID.
     resource_group : str
         The name of the Azure resource group.
-    key_vault_uri : str
-        The name of the `Azure key vault <https://azure.microsoft.com/products/key-vault>`_ URI. Example: "https://<Key Vault Name>.vault.azure.net/"
-    key_vault_tenant_id : str
-        The name of the Azure key vault secret storing the Tenant ID.
-    key_vault_client_id : str
-        The name of the Azure key vault secret storing the Client ID.
-    key_vault_client_secret : str
-        The name of the Azure key vault secret storing the Client Secret.
-    """
-
-    # https://learn.microsoft.com/en-us/rest/api/microsoftfabric/fabric-capacities/delete?view=rest-microsoftfabric-2023-11-01&tabs=HTTP
-
-    azure_token, credential, headers = _get_azure_token_credentials(
-        key_vault_uri=key_vault_uri,
-        key_vault_tenant_id=key_vault_tenant_id,
-        key_vault_client_id=key_vault_client_id,
-        key_vault_client_secret=key_vault_client_secret,
-    )
+    """
+
+    token_provider = auth.token_provider.get()
+    if token_provider is None:
+        token_provider = ServicePrincipalTokenProvider.from_azure_key_vault(
+            key_vault_uri=kwargs["key_vault_uri"],
+            key_vault_tenant_id=kwargs["key_vault_tenant_id"],
+            key_vault_client_id=kwargs["key_vault_client_id"],
+            key_vault_client_secret=kwargs["key_vault_client_secret"],
+        )
+        print(
+            f"{icons.info} Please use the 'token_provider' parameter instead of the key vault parameters within this function as the key vault parameters have been deprecated."
+        )
+
+    headers = _get_headers(token_provider, audience="azure")
 
     url = f"https://management.azure.com/subscriptions/{azure_subscription_id}/resourceGroups/{resource_group}/providers/Microsoft.Fabric/capacities/{capacity_name}?api-version={icons.azure_api_version}"
 
@@ -472,23 +465,23 @@ def delete_fabric_capacity(
     print(f"{icons.green_dot} The '{capacity_name}' capacity has been deleted.")
 
 
+@log
 def update_fabric_capacity(
     capacity_name: str,
     azure_subscription_id: str,
     resource_group: str,
-    key_vault_uri: str,
-    key_vault_tenant_id: str,
-    key_vault_client_id: str,
-    key_vault_client_secret: str,
     sku: Optional[str] = None,
     admin_members: Optional[str | List[str]] = None,
     tags: Optional[dict] = None,
+    **kwargs,
 ):
     """
     This function updates a Fabric capacity's properties.
 
     This is a wrapper function for the following API: `Fabric Capacities - Update <https://learn.microsoft.com/rest/api/microsoftfabric/fabric-capacities/update?view=rest-microsoftfabric-2023-11-01>`_.
 
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     capacity_name : str
@@ -497,14 +490,6 @@ def update_fabric_capacity(
         The Azure subscription ID.
     resource_group : str
         The name of the Azure resource group.
-    key_vault_uri : str
-        The name of the `Azure key vault <https://azure.microsoft.com/products/key-vault>`_ URI. Example: "https://<Key Vault Name>.vault.azure.net/"
-    key_vault_tenant_id : str
-        The name of the Azure key vault secret storing the Tenant ID.
-    key_vault_client_id : str
-        The name of the Azure key vault secret storing the Client ID.
-    key_vault_client_secret : str
-        The name of the Azure key vault secret storing the Client Secret.
     sku : str, default=None
          The `sku size <https://azure.microsoft.com/pricing/details/microsoft-fabric/>`_ of the Fabric capacity.
     admin_members : str | List[str], default=None
@@ -513,21 +498,19 @@ def update_fabric_capacity(
         Tag(s) to add to the capacity. Example: {'tagName': 'tagValue'}.
     """
 
-    # https://learn.microsoft.com/en-us/rest/api/microsoftfabric/fabric-capacities/update?view=rest-microsoftfabric-2023-11-01&tabs=HTTP
-
-    if isinstance(admin_members, str):
-        admin_members = [admin_members]
-    if tags is not None and not isinstance(tags, dict):
-        raise ValueError(
-            f"{icons.red_dot} If specified, the 'tags' parameter must be a dictionary."
+    token_provider = auth.token_provider.get()
+    if token_provider is None:
+        token_provider = ServicePrincipalTokenProvider.from_azure_key_vault(
+            key_vault_uri=kwargs["key_vault_uri"],
+            key_vault_tenant_id=kwargs["key_vault_tenant_id"],
+            key_vault_client_id=kwargs["key_vault_client_id"],
+            key_vault_client_secret=kwargs["key_vault_client_secret"],
+        )
+        print(
+            f"{icons.info} Please use the 'token_provider' parameter instead of the key vault parameters within this function as the key vault parameters have been deprecated."
         )
 
-    azure_token, credential, headers = _get_azure_token_credentials(
-        key_vault_uri=key_vault_uri,
-        key_vault_tenant_id=key_vault_tenant_id,
-        key_vault_client_id=key_vault_client_id,
-        key_vault_client_secret=key_vault_client_secret,
-    )
+    headers = _get_headers(token_provider, audience="azure")
 
     url = f"https://management.azure.com/subscriptions/{azure_subscription_id}/resourceGroups/{resource_group}/providers/Microsoft.Fabric/capacities/{capacity_name}?api-version={icons.azure_api_version}"
 
@@ -575,20 +558,20 @@ def update_fabric_capacity(
     )
 
 
+@log
 def check_fabric_capacity_name_availablility(
     capacity_name: str,
     azure_subscription_id: str,
     region: str,
-    key_vault_uri: str,
-    key_vault_tenant_id: str,
-    key_vault_client_id: str,
-    key_vault_client_secret: str,
+    **kwargs,
 ) -> bool:
     """
     This function updates a Fabric capacity's properties.
 
     This is a wrapper function for the following API: `Fabric Capacities - Check Name Availability <https://learn.microsoft.com/rest/api/microsoftfabric/fabric-capacities/check-name-availability?view=rest-microsoftfabric-2023-11-01>`_.
 
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
     Parameters
     ----------
     capacity_name : str
@@ -597,34 +580,32 @@ def check_fabric_capacity_name_availablility(
         The Azure subscription ID.
     region : str
         The region name.
-    key_vault_uri : str
-        The name of the `Azure key vault <https://azure.microsoft.com/products/key-vault>`_ URI. Example: "https://<Key Vault Name>.vault.azure.net/"
-    key_vault_tenant_id : str
-        The name of the Azure key vault secret storing the Tenant ID.
-    key_vault_client_id : str
-        The name of the Azure key vault secret storing the Client ID.
-    key_vault_client_secret : str
-        The name of the Azure key vault secret storing the Client Secret.
 
     Returns
     -------
     bool
         An indication as to whether the Fabric capacity name is available or not.
     """
-    # https://learn.microsoft.com/en-us/rest/api/microsoftfabric/fabric-capacities/check-name-availability?view=rest-microsoftfabric-2023-11-01&tabs=HTTP
 
-    azure_token, credential, headers = _get_azure_token_credentials(
-        key_vault_uri=key_vault_uri,
-        key_vault_tenant_id=key_vault_tenant_id,
-        key_vault_client_id=key_vault_client_id,
-        key_vault_client_secret=key_vault_client_secret,
-    )
+    token_provider = auth.token_provider.get()
+    if token_provider is None:
+        token_provider = ServicePrincipalTokenProvider.from_azure_key_vault(
+            key_vault_uri=kwargs["key_vault_uri"],
+            key_vault_tenant_id=kwargs["key_vault_tenant_id"],
+            key_vault_client_id=kwargs["key_vault_client_id"],
+            key_vault_client_secret=kwargs["key_vault_client_secret"],
+        )
+        print(
+            f"{icons.info} Please use the 'token_provider' parameter instead of the key vault parameters within this function as the key vault parameters have been deprecated."
+        )
+
+    headers = _get_headers(token_provider, audience="azure")
 
     payload = {"name": capacity_name, "type": "Microsoft.Fabric/capacities"}
 
     url = f"https://management.azure.com/subscriptions/{azure_subscription_id}/providers/Microsoft.Fabric/locations/{region}/checkNameAvailability?api-version={icons.azure_api_version}"
 
-    response = requests.post(url, headers=headers, data=payload)
+    response = requests.post(url, headers=headers, json=payload)
 
     if response.status_code != 202:
         raise FabricHTTPException(response)
@@ -632,59 +613,648 @@ def check_fabric_capacity_name_availablility(
     return bool(response.json().get("nameAvailable"))
 
 
+@log
 def create_resource_group(
     azure_subscription_id: str,
-    key_vault_uri: str,
-    key_vault_tenant_id: str,
-    key_vault_client_id: str,
-    key_vault_client_secret: str,
     resource_group: str,
     region: str,
+    **kwargs,
 ):
     """
     This function creates a resource group in a region within an Azure subscription.
 
-    This is a wrapper function for the following API: `ResourceGroupsOperations Class - CreateOrUpdate <https://learn.microsoft.com/python/api/azure-mgmt-resource/azure.mgmt.resource.resources.v2022_09_01.operations.resourcegroupsoperations?view=azure-python#azure-mgmt-resource-resources-v2022-09-01-operations-resourcegroupsoperations-create-or-update>`_.
+    This is a wrapper function for the following API: `Resource Groups - Create Or Update <https://learn.microsoft.com/rest/api/resources/resource-groups/create-or-update>`_.
+
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
 
     Parameters
     ----------
     azure_subscription_id : str
         The Azure subscription ID.
-    key_vault_uri : str
-        The name of the `Azure key vault <https://azure.microsoft.com/products/key-vault>`_ URI. Example: "https://<Key Vault Name>.vault.azure.net/"
-    key_vault_tenant_id : str
-        The name of the Azure key vault secret storing the Tenant ID.
-    key_vault_client_id : str
-        The name of the Azure key vault secret storing the Client ID.
-    key_vault_client_secret : str
-        The name of the Azure key vault secret storing the Client Secret.
     resource_group : str
         The name of the Azure resource group to be created.
     region : str
         The name of the region in which the resource group will be created.
     """
 
-    from azure.mgmt.resource import ResourceManagementClient
-
-    azure_token, credential, headers = _get_azure_token_credentials(
-        key_vault_uri=key_vault_uri,
-        key_vault_tenant_id=key_vault_tenant_id,
-        key_vault_client_id=key_vault_client_id,
-        key_vault_client_secret=key_vault_client_secret,
-    )
-
-    resource_client = ResourceManagementClient(credential, azure_subscription_id)
-
-    if resource_client.resource_groups.check_existence(resource_group):
+    if check_resource_group_existence(
+        azure_subscription_id=azure_subscription_id,
+        resource_group=resource_group,
+    ):
         print(
             f"{icons.info} The '{resource_group}' resource group already exists in the '{region}' region within the '{azure_subscription_id}' Azure subscription."
         )
         return
 
-    resource_client.resource_groups.create_or_update(
-        resource_group, {"location": region}
+    create_or_update_resource_group(
+        azure_subscription_id=azure_subscription_id,
+        resource_group=resource_group,
+        region=region,
+    )
+
+
+@log
+def list_skus_for_capacity(
+    capacity: str,
+    azure_subscription_id: str,
+    resource_group: str,
+) -> pd.DataFrame:
+    """
+    Lists eligible SKUs for a Microsoft Fabric resource.
+
+    This is a wrapper function for the following API: `Fabric Capacities - List Skus For Capacity <https://learn.microsoft.com/rest/api/microsoftfabric/fabric-capacities/list-skus-for-capacity?view=rest-microsoftfabric-2023-11-01>`_.
+
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    capacity : str
+        The capacity name.
+    azure_subscription_id : str
+        The Azure subscription ID.
+    resource_group : str
+        The name of the resource group.
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing a list of eligible SKUs for a Microsoft Fabric resource.
+    """
+
+    df = pd.DataFrame(columns=["Resource Type", "Sku", "Sku Tier"])
+    url = f"https://management.azure.com/subscriptions/{azure_subscription_id}/resourceGroups/{resource_group}/providers/Microsoft.Fabric/capacities/{capacity}/skus?api-version=2023-11-01"
+    headers = _get_headers(token_provider=auth.token_provider.get(), audience="azure")
+
+    response = requests.get(url, headers=headers)
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    for v in response.json().get("value", []):
+        sku = v.get("sku", {})
+        new_data = {
+            "Resource Type": v.get("resourceType"),
+            "Sku": sku.get("name"),
+            "Sku Tier": sku.get("tier"),
+        }
+
+        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    return df
+
+
+@log
+def list_skus(
+    azure_subscription_id: str,
+) -> pd.DataFrame:
+    """
+    Lists eligible SKUs for Microsoft Fabric resource provider.
+
+    This is a wrapper function for the following API: `Fabric Capacities - List Skus For Capacity <https://learn.microsoft.com/rest/api/microsoftfabric/fabric-capacities/list-skus?view=rest-microsoftfabric-2023-11-01>`_.
+
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    azure_subscription_id : str
+        The Azure subscription ID.
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing a list of eligible SKUs for Microsoft Fabric resource provider.
+    """
+
+    df = pd.DataFrame(columns=["Sku", "Locations"])
+    url = f"https://management.azure.com/subscriptions/{azure_subscription_id}/providers/Microsoft.Fabric/skus?api-version=2023-11-01"
+    headers = _get_headers(token_provider=auth.token_provider.get(), audience="azure")
+
+    response = requests.get(url, headers=headers)
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    for v in response.json().get("value", []):
+        new_data = {
+            "Sku": v.get("name"),
+            "Locations": v.get("locations", []),
+        }
+
+        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    return df
+
+
+@log
+def list_subscriptions() -> pd.DataFrame:
+    """
+    Gets all subscriptions for a tenant.
+
+    This is a wrapper function for the following API: `Subscriptions - List <https://learn.microsoft.com/rest/api/resources/subscriptions/list?view=rest-resources-2022-12-01>`_.
+
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing a list of all subscriptions for a tenant.
+    """
+
+    df = pd.DataFrame(
+        columns=[
+            "Subscription Id",
+            "Subscription Name",
+            "Tenant Id",
+            "State",
+            "Location Placement Id",
+            "Quota Id",
+            "Spending Limit",
+            "Authorization Source",
+            "Managed By Tenants",
+            "Tags",
+        ]
     )
+    url = "https://management.azure.com/subscriptions?api-version=2022-12-01"
+    headers = _get_headers(token_provider=auth.token_provider.get(), audience="azure")
+
+    response = requests.get(url, headers=headers)
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    for v in response.json().get("value", []):
+        policy = v.get("subscriptionPolicies", {})
+        tenants = v.get("managedByTenants")
+        new_data = {
+            "Subscription Id": v.get("subscriptionId"),
+            "Subscription Name": v.get("displayName"),
+            "Tenant Id": v.get("tenantId"),
+            "State": v.get("state"),
+            "Location Placement Id": policy.get("locationPlacementId"),
+            "Quota Id": policy.get("quotaId"),
+            "Spending Limit": policy.get("spendingLimit"),
+            "Authorization Source": v.get("authorizationSource"),
+            "Managed by Tenants": tenants if tenants is not None else [],
+            "Tags": v.get("tags", {}),
+        }
+
+        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    return df
+
+
+@log
+def get_subscription(azure_subscription_id: str) -> pd.DataFrame:
+    """
+    Gets details about a specified subscription.
+
+    This is a wrapper function for the following API: `Subscriptions - Get <https://learn.microsoft.com/rest/api/resources/subscriptions/get?view=rest-resources-2022-12-01>`_.
+
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    azure_subscription_id : str
+        The Azure subscription ID.
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing details of a specific subscription.
+    """
+
+    df = pd.DataFrame(
+        columns=[
+            "Subscription Id",
+            "Subscription Name",
+            "Tenant Id",
+            "State",
+            "Location Placement Id",
+            "Quota Id",
+            "Spending Limit",
+            "Authorization Source",
+            "Managed By Tenants",
+            "Tags",
+        ]
+    )
+    url = f"https://management.azure.com/subscriptions/{azure_subscription_id}?api-version=2022-12-01"
+    headers = _get_headers(token_provider=auth.token_provider.get(), audience="azure")
+
+    response = requests.get(url, headers=headers)
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    v = response.json()
+    policy = v.get("subscriptionPolicies", {})
+    tenants = v.get("managedByTenants")
+    new_data = {
+        "Subscription Id": v.get("subscriptionId"),
+        "Subscription Name": v.get("displayName"),
+        "Tenant Id": v.get("tenantId"),
+        "State": v.get("state"),
+        "Location Placement Id": policy.get("locationPlacementId"),
+        "Quota Id": policy.get("quotaId"),
+        "Spending Limit": policy.get("spendingLimit"),
+        "Authorization Source": v.get("authorizationSource"),
+        "Managed by Tenants": tenants if tenants is not None else [],
+        "Tags": v.get("tags", {}),
+    }
+
+    df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    return df
+
+
+def _resolve_subscription_name_and_id(
+    azure_subscription: str | UUID,
+) -> Tuple[str, UUID]:
+
+    if _is_valid_uuid(azure_subscription):
+        subscription_id = azure_subscription
+        df = get_subscription(azure_subscription_id=subscription_id)
+        if df.empty:
+            raise ValueError(f"{icons.red_dot} The subscription ID does not exist.")
+        subscription_name = df["Subscription Name"].iloc[0]
+    else:
+        subscription_name = azure_subscription
+        df = list_subscriptions()
+        df_filt = df[df["Subscription Name"] == subscription_name]
+        if df_filt.empty:
+            raise ValueError(f"{icons.red_dot} The subscription name does not exist.")
+        subscription_id = df_filt["Subscription Id"].iloc[0]
+
+    return subscription_name, subscription_id
+
+
+@log
+def list_tenants() -> pd.DataFrame:
+    """
+    Gets the tenants for your account.
+
+    This is a wrapper function for the following API: `Tenants - List <https://learn.microsoft.com/rest/api/resources/tenants/list?view=rest-resources-2022-12-01>`_.
+
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing a list of all tenants for your account.
+    """
+
+    df = pd.DataFrame(
+        columns=[
+            "Tenant Id",
+            "Tenant Name",
+            "Country Code",
+            "Domains",
+            "Tenant Category",
+            "Default Domain",
+            "Tenant Type",
+            "Tenant Branding Logo Url",
+        ]
+    )
+    url = "https://management.azure.com/tenants?api-version=2022-12-01"
+    headers = _get_headers(token_provider=auth.token_provider.get(), audience="azure")
+
+    response = requests.get(url, headers=headers)
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    for v in response.json().get("value", []):
+        d = v.get("domains")
+        new_data = {
+            "Tenant Id": v.get("tenantId"),
+            "Tenant Name": v.get("displayName"),
+            "Country Code": v.get("countryCode"),
+            "Domains": d if d is not None else "",
+            "Tenant Category": v.get("tenantCategory"),
+            "Default Domain": v.get("defaultDomain"),
+            "Tenant Type": v.get("tenantType"),
+            "Tenant Branding Logo Url": v.get("tenantBrandingLogoUrl"),
+        }
+
+        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    return df
+
+
+@log
+def create_or_update_resource_group(
+    azure_subscription_id: str,
+    resource_group: str,
+    region: str,
+):
+    """
+    Creates or updates a resource group.
+
+    This is a wrapper function for the following API: `Resource Groups - Create Or Update <https://learn.microsoft.com/rest/api/resources/resource-groups/create-or-update>`_.
+
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    azure_subscription_id : str
+        The Azure subscription Id.
+    resource_group : str
+        The name of the resource group.
+    region : str
+        The name of the region.
+    """
+
+    headers = _get_headers(auth.token_provider.get(), audience="azure")
+    url = f"https://management.azure.com/subscriptions/{azure_subscription_id}/resourcegroups/{resource_group}?api-version=2021-04-01"
+
+    payload = {
+        "location": region,
+    }
+
+    response = requests.put(url, headers=headers, json=payload)
+    if response.status_code not in [200, 201]:
+        raise FabricHTTPException(response)
 
     print(
-        f"{icons.green_dot} The '{resource_group}' resource group has been created within the '{region}' region within the '{azure_subscription_id}' Azure subscription."
+        f"{icons.green_dot} The '{resource_group}' resource group has been created/updated."
     )
+
+
+@log
+def create_storage_account(
+    azure_subscription_id: str,
+    resource_group: str,
+    storage_account: str,
+    region: str,
+):
+    """
+    Asynchronously creates a new storage account with the specified parameters. If an account is already created and a subsequent create request is issued with different properties, the account properties will be updated. If an account is already created and a subsequent create or update request is issued with the exact same set of properties, the request will succeed.
+
+    This is a wrapper function for the following API: `Storage Accounts - Create <https://learn.microsoft.com/rest/api/storagerp/storage-accounts/create`_.
+
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    azure_subscription_id : str
+        The Azure subscription Id.
+    resource_group : str
+        The name of the resource group.
+    storage_account : str
+        The name of the storage account to be created.
+    region : str
+        The name of the region.
+    """
+
+    headers = _get_headers(auth.token_provider.get(), audience="azure")
+    url = f"https://management.azure.com/subscriptions/{azure_subscription_id}/resourceGroups/{resource_group}/providers/Microsoft.Storage/storageAccounts/{storage_account}?api-version=2018-02-01"
+
+    payload = {
+        "sku": {"name": "Standard_GRS"},
+        "kind": "StorageV2",
+        "location": region,
+    }
+
+    response = requests.put(url, headers=headers, json=payload)
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    print(
+        f"{icons.green_dot} The '{storage_account}' storage account has been created."
+    )
+
+
+@log
+def list_storage_accounts(
+    azure_subscription_id: str,
+    resource_group: Optional[str] = None,
+) -> pd.DataFrame:
+    """
+    Lists all the storage accounts available under the subscription (or resource group). Note that storage keys are not returned; use the ListKeys operation for this.
+
+    This is a wrapper function for the following APIs: `Storage Accounts - List <https://learn.microsoft.com/rest/api/storagerp/storage-accounts/list>`_, `Storage Accounts - List By Resource Group <https://learn.microsoft.com/rest/api/storagerp/storage-accounts/list-by-resource-group>`_.
+
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    azure_subscription_id : str
+        The Azure subscription Id.
+    resource_group : str, default=None
+        If set to None, retrieves all storage accounts for the subscription. If not None, shows the storage accounts within that resource group.
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing a list of all storage accounts within the subscription (or resource group).
+    """
+
+    headers = _get_headers(auth.token_provider.get(), audience="azure")
+
+    url = f"https://management.azure.com/subscriptions/{azure_subscription_id}"
+
+    if resource_group is not None:
+        url += f"/resourceGroups/{resource_group}"
+
+    url += "/providers/Microsoft.Storage/storageAccounts?api-version=2023-05-01"
+
+    df = pd.DataFrame(
+        columns=[
+            "Storage Account Id",
+            "Storage Account Name",
+            "Kind",
+            "Location",
+            "Sku Name",
+            "Sku Tier",
+            "Is HNS Enabled",
+            "Creation Time",
+            "Web Endpoint",
+            "DFS Endpoint",
+            "Blob Endpoint",
+            "File Endpoint",
+            "Queue Endpoint",
+            "Table Endpoint",
+            "Primary Location",
+            "Provisioning State",
+            "Secondary Location",
+            "Status of Primary",
+            "Status of Secondary",
+            "Supports HTTPS Traffic Only",
+            "Tags",
+        ]
+    )
+    response = requests.get(url, headers=headers)
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    for v in response.json().get("value", []):
+        p = v.get("properties", {})
+        new_data = {
+            "Storage Account Id": v.get("id"),
+            "Storage Account Name": v.get("name"),
+            "Kind": v.get("kind"),
+            "Location": v.get("location"),
+            "Sku Name": v.get("sku", {}).get("name"),
+            "Sku Tier": v.get("sku", {}).get("tier"),
+            "Is HNS Enabled": p.get("isHnsEnabled"),
+            "Creation Time": p.get("creationTime"),
+            "Web Endpoint": p.get("primaryEndpoints", {}).get("web"),
+            "DFS Endpoint": p.get("primaryEndpoints", {}).get("dfs"),
+            "Blob Endpoint": p.get("primaryEndpoints", {}).get("blob"),
+            "File Endpoint": p.get("primaryEndpoints", {}).get("file"),
+            "Queue Endpoint": p.get("primaryEndpoints", {}).get("queue"),
+            "Table Endpoint": p.get("primaryEndpoints", {}).get("table"),
+            "Primary Location": p.get("primaryLocation"),
+            "Provisioning State": p.get("provisioningState"),
+            "Secondary Location": p.get("secondaryLocation"),
+            "Status of Primary": p.get("statusOfPrimary"),
+            "Status of Secondary": p.get("statusOfSecondary"),
+            "Supports HTTPS Traffic Only": p.get("supportsHttpsTrafficOnly"),
+            "Tags": v.get("tags"),
+        }
+
+        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    bool_cols = ["Is HNS Enabled", "Supports HTTPS Traffic Only"]
+    df[bool_cols] = df[bool_cols].astype(bool)
+    df["Creation Time"] = pd.to_datetime(df["Creation Time"])
+
+    return df
+
+
+@log
+def check_resource_group_existence(
+    azure_subscription_id: str, resource_group: str
+) -> bool:
+    """
+    Checks whether a resource group exists.
+
+    This is a wrapper function for the following API: `Resource Groups - Check Existence <https://learn.microsoft.com/rest/api/resources/resource-groups/check-existence>`_.
+
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    azure_subscription_id : str
+        The Azure subscription Id.
+    resource_group : str
+        The name of the resource group.
+
+    Returns
+    -------
+    bool
+        True/False indicating if the resource group exists or not.
+    """
+
+    headers = _get_headers(auth.token_provider.get(), audience="azure")
+    url = f"https://management.azure.com/subscriptions/{azure_subscription_id}/resourceGroups/{resource_group}?api-version=2021-04-01"
+
+    response = requests.get(url, headers=headers)
+
+    if response.status_code not in [200, 204, 404]:
+        raise FabricHTTPException(response)
+
+    if response.status_code == 200:
+        return True
+    elif response.status_code in [204, 404]:
+        return False
+
+
+@log
+def list_resource_groups(
+    azure_subscription_id: str,
+    filter: Optional[str] = None,
+    top: Optional[int] = None,
+) -> pd.DataFrame:
+    """
+    Lists all resource groups within a subscription.
+
+    This is a wrapper function for the following API: `Resource Groups - List <https://learn.microsoft.com/rest/api/resources/resource-groups/list>`_.
+
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    azure_subscription_id : str
+        The Azure subscription Id.
+    filter : str, default=None
+        The filter to apply to the operation. Example: filter="tagname eq 'tagvalue'".
+    top : int, default=None
+        The number of results to return. If not specified, returns all results.
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing a list of all resource groups within the subscription.
+    """
+
+    headers = _get_headers(auth.token_provider.get(), audience="azure")
+    url = f"https://management.azure.com/subscriptions/{azure_subscription_id}/resourceGroups?"
+
+    if filter is not None:
+        url += f"$filter={filter}&"
+    if top is not None:
+        url += f"$top={top}&"
+
+    url += "api-version=2021-04-01"
+
+    df = pd.DataFrame(columns=["Resource Group Name", "Location", "Tags"])
+    response = requests.get(url, headers=headers)
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    for v in response.json().get("value", []):
+        new_data = {
+            "Resource Group Id": v.get("id"),
+            "Resource Group Name": v.get("name"),
+            "Location": v.get("location"),
+            "Managed By": v.get("managedBy"),
+            "Tags": v.get("tags"),
+            "Type": v.get("type"),
+            "Provisioning State": v.get("properties", {}).get("provisioningState"),
+        }
+
+        df = pd.concat([df, pd.DataFrame(new_data, index=[0])], ignore_index=True)
+
+    return df
+
+
+@log
+def get_resource_group(azure_subscription_id: str, resource_group: str) -> pd.DataFrame:
+    """
+    Gets details about a specified resource group.
+
+    This is a wrapper function for the following API: `Resource Groups - Get <https://learn.microsoft.com/rest/api/resources/resource-groups/get>`_.
+
+    Service Principal Authentication is required (see `here <https://github.com/microsoft/semantic-link-labs/blob/main/notebooks/Service%20Principal.ipynb>`_ for examples).
+
+    Parameters
+    ----------
+    azure_subscription_id : str
+        The Azure subscription Id.
+    resource_group : str
+        The name of the resource group.
+
+    Returns
+    -------
+    pandas.DataFrame
+        A pandas dataframe showing details of a specific resource group.
+    """
+
+    headers = _get_headers(auth.token_provider.get(), audience="azure")
+    url = f"https://management.azure.com/subscriptions/{azure_subscription_id}/resourceGroups/{resource_group}?api-version=2021-04-01"
+
+    response = requests.get(url, headers=headers)
+
+    if response.status_code != 200:
+        raise FabricHTTPException(response)
+
+    v = response.json()
+    new_data = {
+        "Resource Group Id": v.get("id"),
+        "Resource Group Name": v.get("name"),
+        "Location": v.get("location"),
+        "Managed By": v.get("managedBy"),
+        "Tags": v.get("tags"),
+        "Type": v.get("type"),
+        "Provisioning State": v.get("properties", {}).get("provisioningState"),
+    }
+
+    return pd.DataFrame(new_data, index=[0])