Automate Cleanup of Unused Azure Resources (AVD Included)

#!/usr/bin/env python3
"""
Azure Unused Resources Finder
Identifies unused Azure resources across subscriptions
"""

from azure.identity import DefaultAzureCredential
from azure.mgmt.compute import ComputeManagementClient
from azure.mgmt.network import NetworkManagementClient
from azure.mgmt.resource import SubscriptionClient
from datetime import datetime, timedelta
import csv

def get_subscriptions(credential):
    """Get all Azure subscriptions"""
    subscription_client = SubscriptionClient(credential)
    return [sub.subscription_id for sub in subscription_client.subscriptions.list()]

def find_unattached_disks(compute_client):
    """Find all unattached managed disks"""
    unattached = []
    for disk in compute_client.disks.list():
        if disk.disk_state == 'Unattached':
            cost_estimate = disk.disk_size_gb * 0.05  # Rough estimate: $0.05/GB/month
            unattached.append({
                'type': 'Disk',
                'name': disk.name,
                'resource_group': disk.id.split('/')[4],
                'size_gb': disk.disk_size_gb,
                'estimated_monthly_cost': f'${cost_estimate:.2f}',
                'location': disk.location
            })
    return unattached

def find_orphaned_nics(network_client):
    """Find network interfaces not attached to VMs"""
    orphaned = []
    for nic in network_client.network_interfaces.list_all():
        if not nic.virtual_machine:
            orphaned.append({
                'type': 'NIC',
                'name': nic.name,
                'resource_group': nic.id.split('/')[4],
                'location': nic.location,
                'estimated_monthly_cost': '$5.00'
            })
    return orphaned

def find_unused_public_ips(network_client):
    """Find public IPs not associated with any resource"""
    unused = []
    for ip in network_client.public_ip_addresses.list_all():
        if not ip.ip_configuration:
            unused.append({
                'type': 'PublicIP',
                'name': ip.name,
                'resource_group': ip.id.split('/')[4],
                'ip_address': ip.ip_address or 'Not assigned',
                'estimated_monthly_cost': '$3.50',
                'location': ip.location
            })
    return unused

def main():
    print("🔍 Azure Unused Resources Finder")
    print("=" * 50)

    credential = DefaultAzureCredential()
    subscriptions = get_subscriptions(credential)

    all_unused = []
    total_estimated_cost = 0

    for sub_id in subscriptions:
        print(f"\n📋 Scanning subscription: {sub_id}")

        compute_client = ComputeManagementClient(credential, sub_id)
        network_client = NetworkManagementClient(credential, sub_id)

        # Find unused resources
        disks = find_unattached_disks(compute_client)
        nics = find_orphaned_nics(network_client)
        ips = find_unused_public_ips(network_client)

        all_unused.extend(disks + nics + ips)

        print(f"  Found: {len(disks)} disks, {len(nics)} NICs, {len(ips)} IPs")

    # Write to CSV
    if all_unused:
        timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
        filename = f'azure_unused_resources_{timestamp}.csv'

        with open(filename, 'w', newline='') as f:
            writer = csv.DictWriter(f, fieldnames=all_unused[0].keys())
            writer.writeheader()
            writer.writerows(all_unused)

        print(f"\n✅ Report saved: {filename}")
        print(f"📊 Total unused resources: {len(all_unused)}")
    else:
        print("\n✨ No unused resources found!")

if __name__ == "__main__":
    main()