Skip to content

Commit

Permalink
Update log level and execution statement
Browse files Browse the repository at this point in the history
  • Loading branch information
gtheodorio committed Jul 24, 2024
1 parent 04ff7be commit c51088e
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions Azure/azure_cspm_benchmark.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,14 +118,14 @@ def subscription_client(self):
for subscription in az.subscriptions:
row = {'tenant_id': subscription.tenant_id, 'subscription_id': subscription.subscription_id,
'aks_nodes': 0, 'vms': 0, 'aci_containers': 0}
log.info("Exploring Azure subscription: %s (id=%s)", subscription.display_name, subscription.subscription_id)
log.info("Processing Azure subscription: %s (id=%s)", subscription.display_name, subscription.subscription_id)

vmss_list = list(az.vmss_resources(subscription.subscription_id))

# (1) Process AKS
for aks in az.aks_resources(subscription.subscription_id):
for node_pool in az.container_vmss(aks):
log.debug("Identified node pool: '%s' within AKS: '%s' with %d node(s)",
log.info("Identified node pool: '%s' within AKS: '%s' with %d node(s)",
node_pool.name, aks.name, node_pool.count)
row['aks_nodes'] += node_pool.count

Expand All @@ -136,18 +136,18 @@ def subscription_client(self):
continue

vm_count = sum(1 for vm in az.vms_inside_vmss(vmss))
log.debug("Identified %d vm resource(s) inside Scale Set: '%s'", vm_count, vmss.name)
log.info("Identified %d vm resource(s) inside Scale Set: '%s'", vm_count, vmss.name)
row['vms'] += vm_count

# # (3) Process ACI
for aci in az.aci_resources(subscription.subscription_id):
container_count = sum(1 for container in az.container_aci(aci))
log.debug("Identified %d container resource(s) inside Container Group: '%s'", container_count, aci.name)
log.info("Identified %d container resource(s) inside Container Group: '%s'", container_count, aci.name)
row['aci_containers'] += container_count

# (4) Process VMs
vm_count = sum((1 for vm in az.vms_resources(subscription.subscription_id)))
log.debug('Identified %d vm resource(s) outside of Scale Sets', vm_count)
log.info('Identified %d vm resource(s) outside of Scale Sets', vm_count)
row['vms'] += vm_count
data.append(row)

Expand Down

0 comments on commit c51088e

Please sign in to comment.