feat: Added collecting network resources to the collector

someengineering · Oct 26, 2023 · 5517c0d · 5517c0d
1 parent b7905a7
commit 5517c0d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/plugins/azure/resoto_plugin_azure/collector.py b/plugins/azure/resoto_plugin_azure/collector.py
@@ -12,6 +12,7 @@
     AzureResource,
     resources as base_resources,
 )
+from resoto_plugin_azure.resource.network import resources as network_resources
 from resotolib.baseresources import Cloud, GraphRoot
 from resotolib.core.actions import CoreFeedback
 from resotolib.graph import Graph
@@ -27,7 +28,7 @@ def resource_with_params(clazz: Type[AzureResource], params: Set[str], includes_
     return cp.issubset(params) and (not includes_all or params.issubset(cp))
 
 
-all_resources: List[Type[AzureResource]] = base_resources + compute_resources
+all_resources: List[Type[AzureResource]] = base_resources + compute_resources + network_resources
 global_resources = [r for r in all_resources if resource_with_params(r, {"subscriptionId"})]
 regional_resources = [r for r in all_resources if resource_with_params(r, {"subscriptionId", "location"}, True)]
 

diff --git a/plugins/azure/resoto_plugin_azure/resource/compute.py b/plugins/azure/resoto_plugin_azure/resource/compute.py
@@ -2150,7 +2150,7 @@ class AzureVirtualMachineNetworkInterfaceConfiguration:
 
 @define(eq=False, slots=False)
 class AzureNetworkProfile:
-    kind: ClassVar[str] = "azure_vm_network_profile"
+    kind: ClassVar[str] = "azure_virtual_machine_network_profile"
     mapping: ClassVar[Dict[str, Bender]] = {
         "network_api_version": S("networkApiVersion"),
         "network_interface_configurations": S("networkInterfaceConfigurations")