PLC/Methods/GetNodes.py

   1 from PLC.Faults import *
   2 from PLC.Method import Method
   3 from PLC.Parameter import Parameter, Mixed
   4 from PLC.Filter import Filter
   5 from PLC.Nodes import Node, Nodes
   6 from PLC.Persons import Person, Persons
   7 from PLC.Auth import Auth
   8
   9 class GetNodes(Method):
  10     """
  11     Returns an array of structs containing details about nodes. If
  12     node_filter is specified and is an array of node identifiers or
  13     hostnames, or a struct of node attributes, only nodes matching the
  14     filter will be returned. If return_fields is specified, only the
  15     specified details will be returned.
  16
  17     Some fields may only be viewed by admins.
  18     """
  19
  20     roles = ['admin', 'pi', 'user', 'tech', 'node', 'anonymous']
  21
  22     accepts = [
  23         Auth(),
  24         Mixed([Mixed(Node.fields['node_id'],
  25                      Node.fields['hostname'])],
  26               Filter(Node.fields)),
  27         Parameter([str], "List of fields to return", nullok = True),
  28         ]
  29
  30     returns = [Node.fields]
  31
  32
  33     def call(self, auth, node_filter = None, return_fields = None):
  34
  35         # Must query at least slice_ids_whitelist
  36         if return_fields is not None:
  37             added_fields = set(['slice_ids_whitelist']).difference(return_fields)
  38             return_fields += added_fields
  39         else:
  40             added_fields =[]
  41
  42         # Get node information
  43         nodes = Nodes(self.api, node_filter, return_fields)
  44
  45         # Remove admin only fields
  46         if not isinstance(self.caller, Person) or \
  47            'admin' not in self.caller['roles']:
  48             slice_ids = set()
  49             if self.caller:
  50                 slice_ids.update(self.caller['slice_ids'])
  51             # if node has whitelist, make sure the user has a slice on the whitelist
  52             for node in nodes[:]:
  53                 if 'slice_ids_whitelist' in node and \
  54                    node['slice_ids_whitelist'] and \
  55                    not slice_ids.intersection(node['slice_ids_whitelist']):
  56                     nodes.remove(node)
  57             for node in nodes:
  58                 for field in ['boot_nonce', 'key', 'session', 'root_person_ids']:
  59                     if field in node:
  60                         del node[field]
  61
  62         # remove added fields if not specified
  63         if added_fields:
  64             for node in nodes:
  65                 for field in added_fields:
  66                     del node[field]
  67
  68         return nodes