semantic-link-labs 0.8.2__py3-none-any.whl → 0.8.4__py3-none-any.whl

sempy_labs/_capacities.py

@@ -4,7 +4,7 @@ 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
+from sempy_labs._helper_functions import _get_azure_token_credentials
 import pandas as pd
 
 
@@ -37,6 +37,8 @@ def create_fabric_capacity(
     """
     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>`_.
+
     Parameters
     ----------
     capacity_name : str
@@ -65,8 +67,6 @@ def create_fabric_capacity(
 
     from azure.mgmt.resource import ResourceManagementClient
 
-    capacity_suffix = "fsku"
-
     if isinstance(admin_members, str):
         admin_members = [admin_members]
 
@@ -143,7 +143,7 @@ def create_fabric_capacity(
             f"{icons.red_dot} Invalid region. Valid options: {valid_regions}."
         )
 
-    azure_token, credential, headers = get_azure_token_credentials(
+    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,
@@ -255,6 +255,8 @@ def suspend_fabric_capacity(
     """
     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>`_.
+
     Parameters
     ----------
     capacity_name : str
@@ -274,7 +276,7 @@ def suspend_fabric_capacity(
     """
     # 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(
+    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,
@@ -288,7 +290,7 @@ def suspend_fabric_capacity(
     if response.status_code != 202:
         raise FabricHTTPException(response)
 
-    print(f"{icons.green_dot} The '{capacity_name} capacity has been suspended.")
+    print(f"{icons.green_dot} The '{capacity_name}' capacity has been suspended.")
 
 
 def resume_fabric_capacity(
@@ -303,6 +305,8 @@ def resume_fabric_capacity(
     """
     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>`_.
+
     Parameters
     ----------
     capacity_name : str
@@ -323,7 +327,7 @@ def resume_fabric_capacity(
 
     # 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(
+    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,
@@ -337,7 +341,7 @@ def resume_fabric_capacity(
     if response.status_code != 202:
         raise FabricHTTPException(response)
 
-    print(f"{icons.green_dot} The '{capacity_name} capacity has been resumed.")
+    print(f"{icons.green_dot} The '{capacity_name}' capacity has been resumed.")
 
 
 def delete_embedded_capacity(
@@ -372,7 +376,7 @@ def delete_embedded_capacity(
 
     # 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(
+    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,
@@ -386,7 +390,7 @@ def delete_embedded_capacity(
     if response.status_code not in [200, 202]:
         raise FabricHTTPException(response)
 
-    print(f"{icons.green_dot} The '{capacity_name} capacity has been deleted.")
+    print(f"{icons.green_dot} The '{capacity_name}' capacity has been deleted.")
 
 
 def delete_premium_capacity(capacity_name: str):
@@ -429,6 +433,8 @@ def delete_fabric_capacity(
     """
     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>`_.
+
     Parameters
     ----------
     capacity_name : str
@@ -449,7 +455,7 @@ def delete_fabric_capacity(
 
     # 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(
+    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,
@@ -463,7 +469,7 @@ def delete_fabric_capacity(
     if response.status_code != 202:
         raise FabricHTTPException(response)
 
-    print(f"{icons.green_dot} The '{capacity_name} capacity has been deleted.")
+    print(f"{icons.green_dot} The '{capacity_name}' capacity has been deleted.")
 
 
 def update_fabric_capacity(
@@ -481,6 +487,8 @@ def update_fabric_capacity(
     """
     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>`_.
+
     Parameters
     ----------
     capacity_name : str
@@ -503,7 +511,6 @@ def update_fabric_capacity(
         The email address(es) of the admin(s) of the Fabric capacity.
     tags : dict, default=None
         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
@@ -515,7 +522,7 @@ def update_fabric_capacity(
             f"{icons.red_dot} If specified, the 'tags' parameter must be a dictionary."
         )
 
-    azure_token, credential, headers = get_azure_token_credentials(
+    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,
@@ -524,26 +531,47 @@ def update_fabric_capacity(
 
     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}"
 
+    get_response = requests.get(url, headers=headers)
+    if get_response.status_code != 200:
+        raise FabricHTTPException(get_response)
+
+    get_json = get_response.json()
+    current_sku = get_json.get("sku", {}).get("name")
+    current_admins = (
+        get_json.get("properties", {}).get("administration", {}).get("members")
+    )
+    current_tags = get_json.get("tags")
+
     payload = {}
+    payload["sku"] = {
+        "name": current_sku,
+        "tier": "Fabric",
+    }
+    payload["tags"] = current_tags
+    payload["properties"] = get_json["properties"]
+
     if sku is not None:
-        payload["sku"] = {"name": sku, "tier": "Fabric"}
+        payload["sku"]["name"] = sku
     if admin_members is not None:
-        payload["properties"] = {"administration": {"members": [admin_members]}}
-
-    payload = _add_sll_tag(payload, tags)
-
-    if payload == {}:
-        raise ValueError(
-            f"{icons.warning} No parameters have been set to update the '{capacity_name}' capacity."
+        payload["properties"] = {"administration": {"members": admin_members}}
+    if tags is not None:
+        payload["tags"] = tags
+
+    # Do not proceed if no properties are being changed
+    if current_sku == sku and current_admins == admin_members and current_tags == tags:
+        print(
+            f"{icons.yellow_dot} The properties of the '{capacity_name}' are the same as those specified in the parameters of this function. No changes have been made."
         )
+        return
 
-    response = requests.patch(url, headers=headers, data=payload)
+    payload = _add_sll_tag(payload, tags)
+    response = requests.patch(url, headers=headers, json=payload)
 
     if response.status_code != 202:
         raise FabricHTTPException(response)
 
     print(
-        f"{icons.green_dot} The '{capacity_name} capacity has been updated accordingly."
+        f"{icons.green_dot} The '{capacity_name}' capacity has been updated accordingly."
     )
 
 
@@ -556,9 +584,36 @@ def check_fabric_capacity_name_availablility(
     key_vault_client_id: str,
     key_vault_client_secret: str,
 ) -> 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>`_.
+
+    Parameters
+    ----------
+    capacity_name : str
+        Name of the Fabric capacity.
+    azure_subscription_id : str
+        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(
+    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,
@@ -575,3 +630,61 @@ def check_fabric_capacity_name_availablility(
         raise FabricHTTPException(response)
 
     return bool(response.json().get("nameAvailable"))
+
+
+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,
+):
+    """
+    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>`_.
+
+    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):
+        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}
+    )
+
+    print(
+        f"{icons.green_dot} The '{resource_group}' resource group has been created within the '{region}' region within the '{azure_subscription_id}' Azure subscription."
+    )

sempy_labs/_capacity_migration.py

@@ -7,6 +7,7 @@ from sempy_labs._workspaces import assign_workspace_to_capacity
 from sempy_labs.admin._basic_functions import (
     assign_workspaces_to_capacity,
     _list_capacities_meta,
+    list_capacities,
 )
 from sempy_labs._helper_functions import (
     resolve_capacity_id,
@@ -15,6 +16,34 @@ from sempy_labs._helper_functions import (
 from sempy_labs._capacities import create_fabric_capacity
 
 
+def migrate_settings(source_capacity: str, target_capacity: str):
+
+    migrate_capacity_settings(
+        source_capacity=source_capacity,
+        target_capacity=target_capacity,
+    )
+    migrate_access_settings(
+        source_capacity=source_capacity,
+        target_capacity=target_capacity,
+    )
+    migrate_notification_settings(
+        source_capacity=source_capacity,
+        target_capacity=target_capacity,
+    )
+    migrate_spark_settings(
+        source_capacity=source_capacity,
+        target_capacity=target_capacity,
+    )
+    migrate_delegated_tenant_settings(
+        source_capacity=source_capacity,
+        target_capacity=target_capacity,
+    )
+    migrate_disaster_recovery_settings(
+        source_capacity=source_capacity,
+        target_capacity=target_capacity,
+    )
+
+
 @log
 def migrate_workspaces(
     source_capacity: str,
@@ -146,8 +175,6 @@ def migrate_capacities(
         If set to True, only migrates P skus. If set to False, migrates both P and A skus.
     """
 
-    from sempy_labs._list_functions import list_capacities
-
     if isinstance(capacities, str):
         capacities = [capacities]
 
@@ -158,7 +185,7 @@ def migrate_capacities(
     if capacities is None:
         dfC_filt = dfC.copy()
     else:
-        dfC_filt = dfC[dfC["Display Name"].isin(capacities)]
+        dfC_filt = dfC[dfC["Capacity Name"].isin(capacities)]
 
     if p_sku_only:
         dfC_filt = dfC_filt[dfC_filt["Sku"].str.startswith("P")]
@@ -177,14 +204,14 @@ def migrate_capacities(
         return
 
     for _, r in dfC_filt.iterrows():
-        cap_name = r["Display Name"]
+        cap_name = r["Capacity Name"]
         region = r["Region"]
         sku_size = r["Sku"]
         admins = r["Admins"]
         tgt_capacity = f"{convert_to_alphanumeric_lowercase(cap_name)}{icons.migrate_capacity_suffix}"
 
         # Check if target capacity exists
-        dfC_tgt = dfC[dfC["Display Name"] == tgt_capacity]
+        dfC_tgt = dfC[dfC["Capacity Name"] == tgt_capacity]
 
         if sku_size[:1] == "A" and use_existing_rg_for_A_sku:
             rg = None
@@ -221,24 +248,7 @@ def migrate_capacities(
             )
 
             # Migrate settings to new capacity
-            migrate_capacity_settings(
-                source_capacity=cap_name, target_capacity=tgt_capacity
-            )
-            migrate_access_settings(
-                source_capacity=cap_name, target_capacity=tgt_capacity
-            )
-            migrate_notification_settings(
-                source_capacity=cap_name, target_capacity=tgt_capacity
-            )
-            migrate_delegated_tenant_settings(
-                source_capacity=cap_name, target_capacity=tgt_capacity
-            )
-            migrate_disaster_recovery_settings(
-                source_capacity=cap_name, target_capacity=tgt_capacity
-            )
-            migrate_spark_settings(
-                source_capacity=cap_name, target_capacity=tgt_capacity
-            )
+            migrate_settings(source_capacity=cap_name, target_capacity=tgt_capacity)
 
 
 @log
@@ -252,24 +262,21 @@ def migrate_capacity_settings(source_capacity: str, target_capacity: str):
         Name of the source capacity.
     target_capacity : str
         Name of the target capacity.
-
-    Returns
-    -------
     """
 
-    dfC = fabric.list_capacities()
-    dfC_filt = dfC[dfC["Display Name"] == source_capacity]
+    dfC = list_capacities()
+    dfC_filt = dfC[dfC["Capacity Name"] == source_capacity]
     if len(dfC_filt) == 0:
         raise ValueError(
             f"{icons.red_dot} The '{source_capacity}' capacity does not exist."
         )
-    source_capacity_id = dfC_filt["Id"].iloc[0].upper()
-    dfC_filt = dfC[dfC["Display Name"] == target_capacity]
+    source_capacity_id = dfC_filt["Capacity Id"].iloc[0].upper()
+    dfC_filt = dfC[dfC["Capacity Name"] == target_capacity]
     if len(dfC_filt) == 0:
         raise ValueError(
             f"{icons.red_dot} The '{target_capacity}' capacity does not exist."
         )
-    target_capacity_id = dfC_filt["Id"].iloc[0].upper()
+    target_capacity_id = dfC_filt["Capacity Id"].iloc[0].upper()
 
     workloads_params = "capacityCustomParameters?workloadIds=ADM&workloadIds=CDSA&workloadIds=DMS&workloadIds=RsRdlEngine&workloadIds=ScreenshotEngine&workloadIds=AS&workloadIds=QES&workloadIds=DMR&workloadIds=ESGLake&workloadIds=NLS&workloadIds=lake&workloadIds=TIPS&workloadIds=Kusto&workloadIds=Lakehouse&workloadIds=SparkCore&workloadIds=DI&workloadIds=Notebook&workloadIds=ML&workloadIds=ES&workloadIds=Reflex&workloadIds=Must&workloadIds=dmh&workloadIds=PowerBI&workloadIds=HLS"
 
@@ -350,19 +357,19 @@ def migrate_disaster_recovery_settings(source_capacity: str, target_capacity: st
         Name of the target capacity.
     """
 
-    dfC = fabric.list_capacities()
-    dfC_filt = dfC[dfC["Display Name"] == source_capacity]
+    dfC = list_capacities()
+    dfC_filt = dfC[dfC["Capacity Name"] == source_capacity]
     if len(dfC_filt) == 0:
         raise ValueError(
             f"{icons.red_dot} The '{source_capacity}' capacity does not exist."
         )
-    source_capacity_id = dfC_filt["Id"].iloc[0].upper()
-    dfC_filt = dfC[dfC["Display Name"] == target_capacity]
+    source_capacity_id = dfC_filt["Capacity Id"].iloc[0].upper()
+    dfC_filt = dfC[dfC["Capacity Name"] == target_capacity]
     if len(dfC_filt) == 0:
         raise ValueError(
             f"{icons.red_dot} The '{target_capacity}' capacity does not exist."
         )
-    target_capacity_id = dfC_filt["Id"].iloc[0].upper()
+    target_capacity_id = dfC_filt["Capacity Id"].iloc[0].upper()
 
     client = fabric.PowerBIRestClient()
     response_get_source = client.get(f"capacities/{source_capacity_id}/config")
@@ -395,24 +402,21 @@ def migrate_access_settings(source_capacity: str, target_capacity: str):
         Name of the source capacity.
     target_capacity : str
         Name of the target capacity.
-
-    Returns
-    -------
     """
 
-    dfC = fabric.list_capacities()
-    dfC_filt = dfC[dfC["Display Name"] == source_capacity]
+    dfC = list_capacities()
+    dfC_filt = dfC[dfC["Capacity Name"] == source_capacity]
     if len(dfC_filt) == 0:
         raise ValueError(
             f"{icons.red_dot} The '{source_capacity}' capacity does not exist."
         )
-    source_capacity_id = dfC_filt["Id"].iloc[0].upper()
-    dfC_filt = dfC[dfC["Display Name"] == target_capacity]
+    source_capacity_id = dfC_filt["Capacity Id"].iloc[0].upper()
+    dfC_filt = dfC[dfC["Capacity Name"] == target_capacity]
     if len(dfC_filt) == 0:
         raise ValueError(
             f"{icons.red_dot} The '{target_capacity}' capacity does not exist."
         )
-    target_capacity_id = dfC_filt["Id"].iloc[0].upper()
+    target_capacity_id = dfC_filt["Capacity Id"].iloc[0].upper()
 
     client = fabric.PowerBIRestClient()
     response_get_source = client.get(f"capacities/{source_capacity_id}")
@@ -444,24 +448,21 @@ def migrate_notification_settings(source_capacity: str, target_capacity: str):
         Name of the source capacity.
     target_capacity : str
         Name of the target capacity.
-
-    Returns
-    -------
     """
 
-    dfC = fabric.list_capacities()
-    dfC_filt = dfC[dfC["Display Name"] == source_capacity]
+    dfC = list_capacities()
+    dfC_filt = dfC[dfC["Capacity Name"] == source_capacity]
     if len(dfC_filt) == 0:
         raise ValueError(
             f"{icons.red_dot} The '{source_capacity}' capacity does not exist."
         )
-    source_capacity_id = dfC_filt["Id"].iloc[0].upper()
-    dfC_filt = dfC[dfC["Display Name"] == target_capacity]
+    source_capacity_id = dfC_filt["Capacity Id"].iloc[0].upper()
+    dfC_filt = dfC[dfC["Capacity Name"] == target_capacity]
     if len(dfC_filt) == 0:
         raise ValueError(
             f"{icons.red_dot} The '{target_capacity}' capacity does not exist."
         )
-    target_capacity_id = dfC_filt["Id"].iloc[0].upper()
+    target_capacity_id = dfC_filt["Capacity Id"].iloc[0].upper()
 
     client = fabric.PowerBIRestClient()
     response_get_source = client.get(f"capacities/{source_capacity_id}")
@@ -495,26 +496,23 @@ def migrate_delegated_tenant_settings(source_capacity: str, target_capacity: str
         Name of the source capacity.
     target_capacity : str
         Name of the target capacity.
-
-    Returns
-    -------
     """
 
-    dfC = fabric.list_capacities()
+    dfC = list_capacities()
 
-    dfC_filt = dfC[dfC["Display Name"] == source_capacity]
+    dfC_filt = dfC[dfC["Capacity Name"] == source_capacity]
     if len(dfC_filt) == 0:
         raise ValueError(
             f"{icons.red_dot} The '{source_capacity}' capacity does not exist."
         )
-    source_capacity_id = dfC_filt["Id"].iloc[0].upper()
+    source_capacity_id = dfC_filt["Capacity Id"].iloc[0].upper()
 
-    dfC_filt = dfC[dfC["Display Name"] == target_capacity]
+    dfC_filt = dfC[dfC["Capacity Name"] == target_capacity]
     if len(dfC_filt) == 0:
         raise ValueError(
             f"{icons.red_dot} The '{target_capacity}' capacity does not exist."
         )
-    target_capacity_id = dfC_filt["Id"].iloc[0].upper()
+    target_capacity_id = dfC_filt["Capacity Id"].iloc[0].upper()
 
     client = fabric.FabricRestClient()
     response_get = client.get("v1/admin/capacities/delegatedTenantSettingOverrides")
@@ -621,3 +619,106 @@ def migrate_spark_settings(source_capacity: str, target_capacity: str):
     print(
         f"{icons.green_dot} The spark settings have been migrated from the '{source_capacity}' capacity to the '{target_capacity}' capacity."
     )
+
+
+@log
+def migrate_fabric_trial_capacity(
+    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,
+    source_capacity: str,
+    target_capacity: str,
+    target_capacity_sku: str = "F64",
+    target_capacity_admin_members: Optional[str | List[str]] = None,
+):
+    """
+    This function migrates a Fabric trial capacity to a Fabric capacity. If the 'target_capacity' does not exist, it is created with the relevant target capacity parameters (sku, region, admin members).
+
+    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.
+    source_capacity : str
+        The name of the Fabric trial capacity.
+    target_capacity : str
+        The name of the new Fabric capacity (F SKU). If this capacity does not exist, it will be created.
+    target_capacity_sku : str, default="F64"
+        If the target capacity does not exist, this property sets the SKU size for the target capacity.
+    target_capacity_admin_members : str, default=None
+        If the target capacity does not exist, this property sets the admin members for the target capacity.
+        Defaults to None which resolves to the admin members on the Trial SKU.
+    """
+
+    notebook_workspace_id = fabric.get_notebook_workspace_id()
+    dfW = fabric.list_workspaces(filter=f"id eq '{notebook_workspace_id}'")
+    notebook_capacity_id = dfW["Capacity Id"].iloc[0].lower()
+
+    dfC = list_capacities()
+    dfC_filt = dfC[dfC["Capacity Name"] == source_capacity]
+
+    if len(dfC_filt) == 0:
+        raise ValueError(
+            f"{icons.red_dot} The {source_capacity}' capacity does not exist."
+        )
+
+    source_capacity_sku = dfC_filt["Sku"].iloc[0]
+    if not source_capacity_sku.startswith("FT"):
+        raise ValueError(
+            f"{icons.red_dot} This function is for migrating Fabric trial capacites to Fabric capacities."
+        )
+
+    source_capacity_id = dfC_filt["Capacity Id"].iloc[0].lower()
+    if source_capacity_id == notebook_capacity_id:
+        print(
+            f"{icons.warning} The '{source_capacity}' capacity cannot be both the source capacity as well as the capacity in which the notebook is running."
+        )
+        return
+
+    target_capacity_region = dfC_filt["Region"].iloc[0]
+
+    # Use same admins as source capacity
+    if isinstance(target_capacity_admin_members, str):
+        target_capacity_admin_members = [target_capacity_admin_members]
+
+    if target_capacity_admin_members is None:
+        target_capacity_admin_members = dfC_filt["Admins"].iloc[0]
+
+    dfC_filt = dfC[dfC["Capacity Name"] == target_capacity]
+    if len(dfC_filt) == 0:
+        create_fabric_capacity(
+            capacity_name=target_capacity,
+            azure_subscription_id=azure_subscription_id,
+            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_group=resource_group,
+            region=target_capacity_region,
+            admin_members=target_capacity_admin_members,
+            sku=target_capacity_sku,
+        )
+
+    assign_workspaces_to_capacity(
+        source_capacity=source_capacity,
+        target_capacity=target_capacity,
+        workspace=None,
+    )
+
+    # This migrates all the capacity settings
+    migrate_settings(
+        source_capacity=source_capacity,
+        target_capacity=target_capacity,
+    )