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
9 can_update = ['run_level']
11 class ReportRunlevel(Method):
15 roles = ['node', 'admin']
18 Mixed(BootAuth(), SessionAuth(), Auth()),
19 {'run_level': Node.fields['run_level'],
21 Mixed(Node.fields['node_id'],
22 Node.fields['hostname'])
25 returns = Parameter(int, '1 if successful')
27 def call(self, auth, report_fields, node_id_or_hostname=None):
29 if not isinstance(self.caller, Node):
31 if 'admin' not in self.caller['roles']:
32 raise PLCPermissionDenied, "Not allowed to update node run_level"
34 nodes = Nodes(self.api, [node_id_or_hostname])
36 raise PLCInvalidArgument, "No such node"
41 # avoid logging this even too often
42 # avoid logging occurrences where run_level does not change
44 if 'run_level' in node: former_level=node['run_level']
46 node.update_last_contact()
47 for field in can_update:
48 if field in report_fields:
49 node.update({field : report_fields[field]})
51 node.sync(commit=True)
53 # skip logging in this case
54 if former_level and 'run_level' in node and node['run_level'] == former_level:
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'] ] )