from PLC.Method import Method
from PLC.Parameter import Parameter, Mixed
from PLC.Nodes import Node, Nodes
-from PLC.NodeNetworks import NodeNetwork, NodeNetworks
-from PLC.Auth import PasswordAuth
+from PLC.Interfaces import Interface, Interfaces
+from PLC.Auth import Auth
from PLC.POD import udp_pod
class RebootNode(Method):
roles = ['admin', 'pi', 'tech']
accepts = [
- PasswordAuth(),
+ Auth(),
Mixed(Node.fields['node_id'],
Node.fields['hostname'])
]
def call(self, auth, node_id_or_hostname):
# Get account information
- nodes = Nodes(self.api, [node_id_or_hostname]).values()
+ nodes = Nodes(self.api, [node_id_or_hostname])
if not nodes:
raise PLCInvalidArgument, "No such node"
# Only use the hostname as a backup, try to use the primary ID
# address instead.
host = node['hostname']
- nodenetworks = NodeNetworks(self.api, node['nodenetwork_ids']).values()
- for nodenetwork in nodenetworks:
- if nodenetwork['is_primary'] == 1:
- host = nodenetwork['ip']
+ interfaces = Interfaces(self.api, node['interface_ids'])
+ for interface in interfaces:
+ if interface['is_primary'] == 1:
+ host = interface['ip']
break
try:
except socket.error, e:
# Ignore socket errors
pass
-
+
+ self.event_objects = {'Node': [node['node_id']]}
+ self.message = "RebootNode called"
+
return 1