Style only

tenable · Feb 16, 2025 · 052cdb1 · 052cdb1
1 parent 5b96898
commit 052cdb1
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/tenable/one/assets/api.py b/tenable/one/assets/api.py
@@ -17,7 +17,7 @@
 
 class AssetsAPI(APIEndpoint):
 
-    def list_properties(self, asset_classes: Optional[list[AssetClass]] = None):
+    def list_properties(self, asset_classes: Optional[list[AssetClass]] = None) -> AssetProperties:
         """
          Retrieve assets properties
 
@@ -33,7 +33,7 @@ def list_properties(self, asset_classes: Optional[list[AssetClass]] = None):
                 The asset properties.
 
          Examples:
-             >>> tone_asset_properties = tone.assets.list_properties()
+             >>> tone_asset_properties = tone.assets.list_properties().property_to_asset_field
              >>> for asset_property in tone_asset_properties:
              ...     pprint(asset_property)
 
@@ -43,4 +43,4 @@ def list_properties(self, asset_classes: Optional[list[AssetClass]] = None):
             asset_classes: str = ",".join([asset_class.value for asset_class in asset_classes])
             payload["asset_classes"] = asset_classes
         asset_properties_response: dict[str, dict] = self._get(path="one/api/v1/assets/properties", params=payload)
-        return AssetProperties(asset_properties={**asset_properties_response})
+        return AssetProperties(property_to_asset_field={**asset_properties_response})
diff --git a/tenable/one/assets/schema.py b/tenable/one/assets/schema.py
@@ -71,4 +71,4 @@ class AssetField(BaseModel):
 
 
 class AssetProperties(BaseModel):
-    asset_properties: dict[str, AssetField]
+    property_to_asset_field: dict[str, AssetField]
diff --git a/tests/one/assets/test_assets_api.py b/tests/one/assets/test_assets_api.py
@@ -138,4 +138,4 @@ def test_properties_list(api, asset_properties_response):
     # Act
     asset_properties_result: AssetProperties = api.assets.list_properties(asset_classes=asset_classes)
     # Assert
-    assert asset_properties_result == AssetProperties(asset_properties={**asset_properties_response})
+    assert asset_properties_result == AssetProperties(property_to_asset_field={**asset_properties_response})
Original file line number	Diff line number	Diff line change
Expand Up		@@ -71,4 +71,4 @@ class AssetField(BaseModel):


		class AssetProperties(BaseModel):
		asset_properties: dict[str, AssetField]
		property_to_asset_field: dict[str, AssetField]