From 4df7f4eafbd0038f3b1947b430bc5f88ce3f21cb Mon Sep 17 00:00:00 2001
From: Stephen Soltesz <soltesz@cs.princeton.edu>
Date: Sat, 25 Jul 2009 05:19:48 +0000
Subject: [PATCH] added file missing from last commit to use 'runlevel' rather
 than 'bootstate'

---
 source/steps/UpdateRunLevelWithPLC.py | 65 +++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 source/steps/UpdateRunLevelWithPLC.py

diff --git a/source/steps/UpdateRunLevelWithPLC.py b/source/steps/UpdateRunLevelWithPLC.py
new file mode 100644
index 0000000..7e703a8
--- /dev/null
+++ b/source/steps/UpdateRunLevelWithPLC.py
@@ -0,0 +1,65 @@
+#!/usr/bin/python
+
+# Copyright (c) 2003 Intel Corporation
+# All rights reserved.
+#
+# Copyright (c) 2004-2006 The Trustees of Princeton University
+# All rights reserved.
+
+from Exceptions import *
+import BootAPI
+import notify_messages
+
+
+def Run( vars, log ):
+    """
+    Change this nodes run level at PLC.
+
+    Replaces the behavior of UpdateBootStateWithPLC.  Where previously, the
+    boot_state of a node would be altered by the BM, now the run_level is
+    updated, and the boot_state is preserved as a record of a User's
+    preference.
+
+    The current value of the RUN_LEVEL key in vars is used.
+    Optionally, notify the contacts of the run level change.
+    If this is the case, the following keys/values
+    should be set in vars before calling this step:
+    STATE_CHANGE_NOTIFY= 1
+    STATE_CHANGE_NOTIFY_MESSAGE= "<notify message>"
+    The second value is a message to send the users from notify_messages.py
+
+    Return 1 if succesfull, a BootManagerException otherwise.
+    """
+
+    log.write( "\n\nStep: Updating node run level at PLC.\n" )
+
+    update_vals= {}
+    update_vals['run_level']= vars['RUN_LEVEL']
+    try:
+        BootAPI.call_api_function( vars, "ReportRunlevel", (update_vals,) )
+        log.write( "Successfully updated run level for this node at PLC\n" )
+    except BootManagerException, e:
+        log.write( "Unable to update run level for this node at PLC: %s.\n" % e )
+
+    notify = vars.get("STATE_CHANGE_NOTIFY",0)
+
+    if notify:
+        message= vars['STATE_CHANGE_NOTIFY_MESSAGE']
+        include_pis= 0
+        include_techs= 1
+        include_support= 0
+
+        sent= 0
+        try:
+            sent= BootAPI.call_api_function( vars, "BootNotifyOwners",
+                                             (message,
+                                              include_pis,
+                                              include_techs,
+                                              include_support) )
+        except BootManagerException, e:
+            log.write( "Call to BootNotifyOwners failed: %s.\n" % e )
+
+        if sent == 0:
+            log.write( "Unable to notify site contacts of state change.\n" )
+
+    return 1
-- 
2.43.0