PLC/Methods/AdmGetSites.py

   1 import os
   2
   3 from PLC.Method import Method
   4 from PLC.Parameter import Parameter, Mixed
   5 from PLC.Auth import PasswordAuth
   6 from PLC.Sites import Site, Sites
   7
   8 class AdmGetSites(Method):
   9     """
  10     Return an array of structs containing details about all sites. If
  11     site_id_list is specified, only the specified sites will be
  12     queried.
  13
  14     If return_fields is specified, only the specified fields will be
  15     returned, if set. Otherwise, the default set of fields returned is:
  16
  17     """
  18
  19     roles = ['admin', 'pi', 'user', 'tech']
  20
  21     accepts = [
  22         PasswordAuth(),
  23         [Mixed(Site.fields['site_id'],
  24                Site.fields['login_base'])],
  25         Parameter([str], 'List of fields to return')
  26         ]
  27
  28     # Filter out deleted fields
  29     can_return = lambda (field, value): field not in ['deleted']
  30     return_fields = dict(filter(can_return, Site.all_fields.items()))
  31     returns = [return_fields]
  32
  33     def __init__(self, *args, **kwds):
  34         Method.__init__(self, *args, **kwds)
  35         # Update documentation with list of default fields returned
  36         self.__doc__ += os.linesep.join(Site.default_fields.keys())
  37
  38     def call(self, auth, site_id_or_login_base_list = None, return_fields = None):
  39         # Make sure that only valid fields are specified
  40         if return_fields is None:
  41             return_fields = self.return_fields
  42         elif filter(lambda field: field not in self.return_fields, return_fields):
  43             raise PLCInvalidArgument, "Invalid return field specified"
  44
  45         # Get site information
  46         sites = Sites(self.api, site_id_or_login_base_list, return_fields)
  47
  48         # Filter out undesired or None fields (XML-RPC cannot marshal
  49         # None) and turn each site into a real dict.
  50         valid_return_fields_only = lambda (key, value): \
  51                                    key in return_fields and value is not None
  52         sites = [dict(filter(valid_return_fields_only, site.items())) \
  53                  for site in sites.values()]
  54
  55         return sites