Changeset b709861 for fedd/federation/emulab_segment.py

Timestamp:

Nov 30, 2011 4:45:15 PM (12 years ago)

Author:

Ted Faber <faber@…>

Branches:

compt_changes, info-ops, master

Children:

1ae1aa2

Parents:

22a1a77

Message:

Rebooting works

File:

• fedd/federation/emulab_segment.py (modified) (4 diffs)

fedd/federation/emulab_segment.py

@@ -14 +14 @@
 from federation.util import fedd_ssl_context
 from federation import service_error
+from federation.operation_status import operation_status
 
 class emulab_segment:
@@ -32 +33 @@
         self.debug = getattr(self, 'debug', False)
         self.node = { }
+        self.status = [ ]
 
     def emulab_call(self, method, params):
@@ -103 +105 @@
         if self.debug:
             if self.log:
-                self.log.debug("[make_null_experiment]: (debug) Creating experiment")
+                self.log.debug("[make_null_experiment]: " + \
+                        "(debug) Creating experiment")
             return True
         else:
@@ -243 +246 @@
                         "Cannot get node mapping of segment:%s/%s" % (pid, eid))
 
+    def do_operation(self, op, lnode, pnode, params):
+        """
+        Carry out operation on node in the given experiment.
+        """
+        def get_param(params, name):
+            if params is None:
+                return None
+            for d in params:
+                if 'attribute' in d and d['attribute'] == name:
+                    return d.get('value', None)
+            else:
+                return None
+
+        op = op.lower()
+
+        if op == 'restore':
+            state = get_param(params, 'state')
+            if state is None:
+                self.status.append(operation_status(lnode, 
+                        operation_status.bad_param, 'No state to restore'))
+                return False
+            elif state == 'initial': 
+                self.status.append(operation_status(lnode,
+                        operation_status.unsupp, 'Image load unsupported'))
+                return False
+            elif state == 'boot': 
+                p = {'nodes': pnode, 'wait': False}
+                code, result = self.emulab_call('node.reboot', p)
+                if code == 0: 
+                    self.status.append(operation_status(lnode,
+                        operation_status.success, 'rebooting'))
+                    return True
+                else:
+                    self.status.append(operation_status(lnode,
+                            operation_status.federant, 'Error code: %d' % code))
+                    return False
+            else: 
+                self.status.append(operation_status(lnode, 
+                        operation_status.bad_param, 
+                        "Unsupported state %s" % state))
+                return False
+        else: 
+            self.status.append(operation_status(lnode, operation_status.unsupp))
+            return False
+