PLC/Methods/ReportRunlevel.py

   1 # $Id$
   2 from PLC.Faults import *
   3 from PLC.Method import Method
   4 from PLC.Parameter import Parameter, Mixed
   5 from PLC.Auth import Auth, BootAuth, SessionAuth
   6 from PLC.Nodes import Node, Nodes
   7
   8 can_update = ['run_level']
   9
  10 class ReportRunlevel(Method):
  11     """
  12         report runlevel
  13     """
  14     roles = ['node', 'admin']
  15
  16     accepts = [
  17         Mixed(BootAuth(), SessionAuth(), Auth()),
  18         {'run_level': Node.fields['run_level'],
  19          },
  20         Mixed(Node.fields['node_id'],
  21               Node.fields['hostname'])
  22         ]
  23
  24     returns = Parameter(int, '1 if successful')
  25
  26     def call(self, auth, report_fields, node_id_or_hostname=None):
  27
  28         if not isinstance(self.caller, Node):
  29             # check admin
  30             if 'admin' not in self.caller['roles']:
  31                 raise PLCPermissionDenied, "Not allowed to update node run_level"
  32
  33             nodes = Nodes(self.api, [node_id_or_hostname])
  34             if not nodes:
  35                 raise PLCInvalidArgument, "No such node"
  36         else:
  37             nodes  = [self.caller]
  38
  39         node = nodes[0]
  40
  41         node.update_last_contact()
  42         for field in can_update:
  43             if field in report_fields:
  44                 node.update({field : report_fields[field]})
  45
  46         node.sync(commit=True)
  47
  48         self.message = "Node Runlevel Report : %s" % ", ".join(report_fields.keys())
  49
  50         return 1