X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=PLC%2FMethods%2FGetPCUProtocolTypes.py;h=72fe26a230e9b97a9806a37d12e20e3872c90982;hb=f7ce7ce813d4c44502629820a3583f32a99a98f7;hp=a75a6b010e4b72f2a9ca852bdf2eaef9789ae805;hpb=9a608e111c7f8182f51753ea3f563ebc19eb4c71;p=plcapi.git diff --git a/PLC/Methods/GetPCUProtocolTypes.py b/PLC/Methods/GetPCUProtocolTypes.py index a75a6b0..72fe26a 100644 --- a/PLC/Methods/GetPCUProtocolTypes.py +++ b/PLC/Methods/GetPCUProtocolTypes.py @@ -16,27 +16,27 @@ class GetPCUProtocolTypes(Method): accepts = [ Auth(), - Mixed([PCUProtocolType.fields['pcu_type_id']], + Mixed([PCUProtocolType.fields['pcu_type_id']], Filter(PCUProtocolType.fields)), Parameter([str], "List of fields to return", nullok = True) ] returns = [PCUProtocolType.fields] - + def call(self, auth, protocol_type_filter = None, return_fields = None): - #Must query at least pcu_type_id - if return_fields is not None and 'pcu_protocol_type_id' not in return_fields: - return_fields.append('pcu_protocol_type_id') - added_fields = ['pcu_protocol_type_id'] - else: - added_fields = [] + #Must query at least pcu_type_id + if return_fields is not None and 'pcu_protocol_type_id' not in return_fields: + return_fields.append('pcu_protocol_type_id') + added_fields = ['pcu_protocol_type_id'] + else: + added_fields = [] + + protocol_types = PCUProtocolTypes(self.api, protocol_type_filter, return_fields) - protocol_types = PCUProtocolTypes(self.api, protocol_type_filter, return_fields) + for added_field in added_fields: + for protocol_type in protocol_types: + del protocol_type[added_field] - for added_field in added_fields: - for protocol_type in protocol_types: - del protocol_type[added_field] - - return protocol_types + return protocol_types