visual-guidance-enable-byRG

autoscaleMultiple-IOPS/README.md

@@ -29,7 +29,7 @@ Last updated: 2025-05-13
 
 ## By Resource Group
 
-> Overall process: <br/> (./scripts/enable_autoscale_iops_byRG.py)
+> Overall process: <br/> 
 >
 > - Automatically retrieves your **Azure subscription ID** using the Azure CLI. <br/>
 > - List all Resource Groups in current subscription ID. <br/>
@@ -41,11 +41,11 @@ Review [the script](./scripts/enable_autoscale_iops_byRG.py), and download it to
 
 > Example: enabling Autoscale IOPS on two different servers, each hosted in same resource group and same subscription.
 
-<img width="550" alt="image" src="" />
+<https://github.com/user-attachments/assets/4c087afe-6fa1-40cb-bb2f-ef912edb974d>
 
 ## Across a Subscription
 
-> You can also enable autoscale IOPS across an entire subscription, it requires: <br/>
+> You can also enable autoscale IOPS across an entire subscription, overall process: <br/>
 >
 > - Listing all MySQL Flexible Servers in the subscription. <br/>
 > - For each server, retrieving its resource group.  <br/>

autoscaleMultiple-IOPS/scripts/enable_autoscale_iops_byRG.py

@@ -0,0 +1,152 @@
+"""
+Azure MySQL Flexible Server Auto IOPS Enabler
+
+This script connects to Azure using the Azure CLI and Azure REST API to:
+1. Retrieve the current subscription ID.
+2. List all Resource Groups in current subscription ID.
+3. List all MySQL Flexible Servers in a specified resource group.
+4. Check if each server is eligible for Auto IOPS scaling (only GeneralPurpose or BusinessCritical tiers).
+5. Optionally enable Auto IOPS scaling if not already enabled.
+6. Print a summary of which servers were updated or skipped.
+
+You can use this script to:
+- Enable Auto IOPS scaling across all eligible servers in a resource group.
+- Target a specific server by name.
+"""
+
+import subprocess
+import requests
+import shutil
+from azure.identity import AzureCliCredential
+
+# Constants
+SUPPORTED_TIERS = {"GeneralPurpose", "BusinessCritical"}
+
+def mask_value(value, start=4, end=4):
+    return value[:start] + '*' * (len(value) - start - end) + value[-end:]
+
+def get_subscription_id():
+    az_path = shutil.which("az")
+    if az_path is None:
+        raise FileNotFoundError("Azure CLI (az) not found in system PATH.")
+    result = subprocess.check_output([az_path, "account", "show", "--query", "id", "-o", "tsv"], text=True)
+    return result.strip()
+
+def get_access_token():
+    credential = AzureCliCredential()
+    token = credential.get_token("https://management.azure.com/.default")
+    return token.token
+
+def list_resource_groups(subscription_id, access_token):
+    url = f"https://management.azure.com/subscriptions/{subscription_id}/resourcegroups?api-version=2024-06-01-preview"
+    headers = {"Authorization": f"Bearer {access_token}"}
+    response = requests.get(url, headers=headers)
+    response.raise_for_status()
+    return response.json()
+
+def list_mysql_servers(subscription_id, resource_group, access_token):
+    url = (
+        f"https://management.azure.com/subscriptions/{subscription_id}/resourceGroups/"
+        f"{resource_group}/providers/Microsoft.DBforMySQL/flexibleServers?api-version=2024-06-01-preview"
+    )
+    headers = {"Authorization": f"Bearer {access_token}"}
+    response = requests.get(url, headers=headers)
+    response.raise_for_status()
+    return response.json()
+
+def get_server_config(subscription_id, resource_group, server_name, access_token):
+    url = (
+        f"https://management.azure.com/subscriptions/{subscription_id}/resourceGroups/{resource_group}/"
+        f"providers/Microsoft.DBforMySQL/flexibleServers/{server_name}?api-version=2024-06-01-preview"
+    )
+    headers = {"Authorization": f"Bearer {access_token}"}
+    response = requests.get(url, headers=headers)
+    response.raise_for_status()
+    return response.json()
+
+def enable_auto_io_scaling_put(subscription_id, resource_group, server_name, access_token):
+    url = (
+        f"https://management.azure.com/subscriptions/{subscription_id}/resourceGroups/{resource_group}/"
+        f"providers/Microsoft.DBforMySQL/flexibleServers/{server_name}?api-version=2024-06-01-preview"
+    )
+    headers = {
+        "Authorization": f"Bearer {access_token}",
+        "Content-Type": "application/json"
+    }
+    payload = {
+        "properties": {
+            "storage": {
+                "autoIoScaling": "Enabled"
+            }
+        }
+    }
+    response = requests.patch(url, headers=headers, json=payload)
+    print(f"PATCH Response: {response.status_code} - {response.text}")
+    response.raise_for_status()
+    return response.json()
+
+def main():
+    try:
+        print("Script started")
+        subscription_id = get_subscription_id()
+        print(f"Using subscription: {mask_value(subscription_id)}")
+
+        access_token = get_access_token()
+        print(f"Access Token: {mask_value(access_token[:20])}...")
+
+        resource_groups = list_resource_groups(subscription_id, access_token)
+        print("Available Resource Groups:")
+        for rg in resource_groups.get("value", []):
+            print(f"  - {rg['name']}")
+
+        resource_group = input("Enter the resource group name: ").strip()
+
+        servers = list_mysql_servers(subscription_id, resource_group, access_token)
+        print("Available MySQL Flexible Servers:")
+        for server in servers.get("value", []):
+            print(f"  - {server['name']}")
+
+        target_server = input("Enter a specific server name (or press Enter to apply to all): ").strip()
+
+        eligible_servers = []
+        ineligible_servers = []
+
+        for server in servers.get("value", []):
+            server_name = server["name"]
+            if target_server and server_name.lower() != target_server.lower():
+                continue
+
+            config = get_server_config(subscription_id, resource_group, server_name, access_token)
+            tier = config["sku"]["tier"]
+            auto_io = config["properties"]["storage"].get("autoIoScaling", "Disabled")
+
+            if tier not in SUPPORTED_TIERS:
+                ineligible_servers.append((server_name, tier))
+                continue
+
+            if auto_io == "Enabled":
+                print(f"[SKIP] Auto IOPS already enabled for {server_name}")
+                continue
+
+            print(f"[UPDATE] Enabling Auto IOPS for {server_name} (Tier: {tier})...")
+            enable_auto_io_scaling_put(subscription_id, resource_group, server_name, access_token)
+            eligible_servers.append(server_name)
+
+        print("\nSummary:")
+        if eligible_servers:
+            print("Auto IOPS enabled for:")
+            for name in eligible_servers:
+                print(f"  - {name}")
+        else:
+            print("No servers were updated.")
+
+        if ineligible_servers:
+            print("\nIneligible servers (unsupported tier):")
+            for name, tier in ineligible_servers:
+                print(f"  - {name} (Tier: {tier})")
+
+    except Exception as e:
+        print(f"Error: {e}")
+
+if __name__ == "__main__":
+    main()