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
8 can_update = ['run_level']
10 class ReportRunlevel(Method):
14 roles = ['node', 'admin']
17 Mixed(BootAuth(), SessionAuth(), Auth()),
18 {'run_level': Node.fields['run_level'],
20 Mixed(Node.fields['node_id'],
21 Node.fields['hostname'])
24 returns = Parameter(int, '1 if successful')
26 def call(self, auth, report_fields, node_id_or_hostname=None):
28 if not isinstance(self.caller, Node):
30 if 'admin' not in self.caller['roles']:
31 raise PLCPermissionDenied, "Not allowed to update node run_level"
33 nodes = Nodes(self.api, [node_id_or_hostname])
35 raise PLCInvalidArgument, "No such node"
40 # avoid logging this even too often
41 # avoid logging occurrences where run_level does not change
43 if 'run_level' in node: former_level=node['run_level']
45 node.update_last_contact()
46 for field in can_update:
47 if field in report_fields:
48 node.update({field : report_fields[field]})
50 node.sync(commit=True)
52 # skip logging in this case
53 if former_level and 'run_level' in node and node['run_level'] == former_level:
56 # handle the 'run_level' key
57 message="run level " + node['hostname'] + ":"
58 if 'run_level' in report_fields:
59 message += former_level + "->" + report_fields['run_level']
60 message += ", ".join( [ k + "->" + v for (k,v) in report_fields.items() if k not in ['run_level'] ] )