X-Git-Url: http://git.onelab.eu/?p=plcapi.git;a=blobdiff_plain;f=trunk%2FPLC%2FMethods%2FAddNodeToPCU.py;fp=trunk%2FPLC%2FMethods%2FAddNodeToPCU.py;h=c0d5eff9db93b5d3e5cae40b26dd604717667893;hp=0000000000000000000000000000000000000000;hb=5a4c1b1278ffa01e630fde47f7c54888ed20a576;hpb=cee5ab52df1c9f38b6eaff2dd354cb22f59028c7

diff --git a/trunk/PLC/Methods/AddNodeToPCU.py b/trunk/PLC/Methods/AddNodeToPCU.py
new file mode 100644
index 0000000..c0d5eff
--- /dev/null
+++ b/trunk/PLC/Methods/AddNodeToPCU.py
@@ -0,0 +1,74 @@
+from PLC.Faults import *
+from PLC.Method import Method
+from PLC.Parameter import Parameter, Mixed
+from PLC.Nodes import Node, Nodes
+from PLC.PCUs import PCU, PCUs
+from PLC.Sites import Site, Sites
+from PLC.Auth import Auth
+
+class AddNodeToPCU(Method):
+    """
+    Adds a node to a port on a PCU. Faults if the node has already
+    been added to the PCU or if the port is already in use.
+
+    Non-admins may only update PCUs at their sites.
+
+    Returns 1 if successful, faults otherwise.
+    """
+
+    roles = ['admin', 'pi', 'tech']
+
+    accepts = [
+        Auth(),
+	Mixed(Node.fields['node_id'],
+              Node.fields['hostname']),
+        PCU.fields['pcu_id'],
+        Parameter(int, 'PCU port number')
+        ]
+
+    returns = Parameter(int, '1 if successful')
+
+    def call(self, auth, node_id_or_hostname, pcu_id, port):
+	 # Get node
+        nodes = Nodes(self.api, [node_id_or_hostname])
+        if not nodes:
+            raise PLCInvalidArgument, "No such node"
+        node = nodes[0]
+
+        if node['peer_id'] is not None:
+            raise PLCInvalidArgument, "Not a local node"
+
+        # Get PCU
+        pcus = PCUs(self.api, [pcu_id])
+        if not pcus:
+            raise PLCInvalidArgument, "No such PCU"
+        pcu = pcus[0]
+
+        if 'admin' not in self.caller['roles']:
+            ok = False
+            sites = Sites(self.api, self.caller['site_ids'])
+            for site in sites:
+                if pcu['pcu_id'] in site['pcu_ids']:
+                    ok = True
+                    break
+            if not ok:
+                raise PLCPermissionDenied, "Not allowed to update that PCU"
+	
+	# Add node to PCU
+        if node['node_id'] in pcu['node_ids']:
+            raise PLCInvalidArgument, "Node already controlled by PCU"
+
+        if node['site_id'] != pcu['site_id']:
+            raise PLCInvalidArgument, "Node is at a different site than this PCU"
+
+        if port in pcu['ports']:
+            raise PLCInvalidArgument, "PCU port already in use"
+
+        pcu.add_node(node, port)
+
+	# Logging variables
+	self.event_objects = {'Node': [node['node_id']],
+			      'PCU': [pcu['pcu_id']]}
+	self.message = 'Node %d added to pcu %d on port %d' % \
+		(node['node_id'], pcu['pcu_id'], port)
+        return 1