PLC/Methods/ReportRunlevel.py

   1 # $Id$
   2 # $URL$
   3 from PLC.Faults import *
   4 from PLC.Method import Method
   5 from PLC.Parameter import Parameter, Mixed
   6 from PLC.Auth import Auth, BootAuth, SessionAuth
   7 from PLC.Nodes import Node, Nodes
   8
   9 can_update = ['run_level']
  10
  11 class ReportRunlevel(Method):
  12     """
  13         report runlevel
  14     """
  15     roles = ['node', 'admin']
  16
  17     accepts = [
  18         Mixed(BootAuth(), SessionAuth(), Auth()),
  19         {'run_level': Node.fields['run_level'],
  20          },
  21         Mixed(Node.fields['node_id'],
  22               Node.fields['hostname'])
  23         ]
  24
  25     returns = Parameter(int, '1 if successful')
  26
  27     def call(self, auth, report_fields, node_id_or_hostname=None):
  28
  29         if not isinstance(self.caller, Node):
  30             # check admin
  31             if 'admin' not in self.caller['roles']:
  32                 raise PLCPermissionDenied, "Not allowed to update node run_level"
  33
  34             nodes = Nodes(self.api, [node_id_or_hostname])
  35             if not nodes:
  36                 raise PLCInvalidArgument, "No such node"
  37         else:
  38             nodes  = [self.caller]
  39
  40         node = nodes[0]
  41         # avoid logging this even too often
  42         # avoid logging occurrences where run_level does not change
  43         former_level=None
  44         if 'run_level' in node: former_level=node['run_level']
  45
  46         node.update_last_contact()
  47         for field in can_update:
  48             if field in report_fields:
  49                 node.update({field : report_fields[field]})
  50
  51         node.sync(commit=True)
  52
  53         # skip logging in this case
  54         if former_level and 'run_level' in node and node['run_level'] == former_level:
  55             pass
  56         else:
  57             # handle the 'run_level' key
  58             message="run level " + node['hostname'] + ":"
  59             if 'run_level' in report_fields:
  60                 message += str(former_level) + "->" + report_fields['run_level']
  61             message += ", ".join(  [ k + "->" + v for (k,v) in report_fields.items() if k not in ['run_level'] ] )
  62
  63         return 1