- re-enable return_fields specification

[plcapi.git] / PLC / Methods / GetSliceAttributeTypes.py

diff --git a/PLC/Methods/GetSliceAttributeTypes.py b/PLC/Methods/GetSliceAttributeTypes.py
index 0495523..5511e58 100644 (file)
--- a/PLC/Methods/GetSliceAttributeTypes.py
+++ b/PLC/Methods/GetSliceAttributeTypes.py
@@ -6,9 +6,12 @@ from PLC.SliceAttributeTypes import SliceAttributeType, SliceAttributeTypes
 
 class GetSliceAttributeTypes(Method):
     """
-    Return an array of structs containing details about all possible
-    slice and node attributes. If attribute_id_or_name_list is
-    specified, only the specified attributes will be queried.
+    Returns an array of structs containing details about slice
+    attribute types. If attribute_type_filter is specified and
+    is an array of slice attribute type identifiers, or a
+    struct of slice attribute type attributes, only slice attribute
+    types matching the filter will be returned. If return_fields is
+    specified, only the specified details will be returned.
     """
 
     roles = ['admin', 'pi', 'user', 'tech']
@@ -17,10 +20,11 @@ class GetSliceAttributeTypes(Method):
         Auth(),
         Mixed([Mixed(SliceAttributeType.fields['attribute_type_id'],
                      SliceAttributeType.fields['name'])],
-              Filter(SliceAttributeType.fields))
+              Filter(SliceAttributeType.fields)),
+        Parameter([str], "List of fields to return", nullok = True)
         ]
 
     returns = [SliceAttributeType.fields]
 
-    def call(self, auth, attribute_type_filter = None):
-        return SliceAttributeTypes(self.api, attribute_type_filter).values()
+    def call(self, auth, attribute_type_filter = None, return_fields = None):
+        return SliceAttributeTypes(self.api, attribute_type_filter, return_fields)