1 from PLC.Faults import *
2 from PLC.Method import Method
3 from PLC.Parameter import Parameter, Mixed
4 from PLC.Nodes import Node, Nodes
5 from PLC.PCUs import PCU, PCUs
6 from PLC.Auth import PasswordAuth
8 class AddNodeToPCU(Method):
10 Adds a node to a port on a PCU. Faults if the node has already
11 been added to the PCU or if the port is already in use.
13 Non-admins may only update PCUs at their sites.
15 Returns 1 if successful, faults otherwise.
18 roles = ['admin', 'pi', 'tech']
22 Mixed(Node.fields['node_id'],
23 Node.fields['hostname']),
25 Parameter(int, 'PCU port number')
28 returns = Parameter(int, '1 if successful')
30 def call(self, auth, node_id_or_hostname, pcu_id, port):
32 nodes = Nodes(self.api, [node_id_or_hostname])
34 raise PLCInvalidArgument, "No such node"
36 node = nodes.values()[0]
39 pcus = PCUs(self.api, [pcu_id])
41 raise PLCInvalidArgument, "No such PCU"
43 pcu = pcus.values()[0]
45 if 'admin' not in self.caller['roles']:
47 sites = Sites(self.api, self.caller['site_ids']).values()
49 if pcu['pcu_id'] in site['pcu_ids']:
53 raise PLCPermissionDenied, "Not allowed to update that PCU"
56 if node['node_id'] in pcu['node_ids']:
57 raise PLCInvalidArgument, "Node already controlled by PCU"
59 if port in pcu['ports']:
60 raise PLCInvalidArgument, "PCU port already in use"
62 pcu.add_node(node, port)