summaryrefslogtreecommitdiff
path: root/cloudinit/transforms/welcome.py
diff options
context:
space:
mode:
authorJoshua Harlow <harlowja@yahoo-inc.com>2012-06-16 12:14:54 -0700
committerJoshua Harlow <harlowja@yahoo-inc.com>2012-06-16 12:14:54 -0700
commit75b46cc80c6a9ab65c3c7f430ffaa4157d0db369 (patch)
tree710f157ce2bdd3cd4c593ef2f726dd6eaa177a51 /cloudinit/transforms/welcome.py
parent68e710207900ff8d933da6f79adc84bce980685d (diff)
downloadvyos-cloud-init-75b46cc80c6a9ab65c3c7f430ffaa4157d0db369.tar.gz
vyos-cloud-init-75b46cc80c6a9ab65c3c7f430ffaa4157d0db369.zip
Remove the need to have transforms start with "cc_".
1. Just let them have no prefix (ie blank) unless this isn't wanted later.
Diffstat (limited to 'cloudinit/transforms/welcome.py')
-rw-r--r--cloudinit/transforms/welcome.py64
1 files changed, 64 insertions, 0 deletions
diff --git a/cloudinit/transforms/welcome.py b/cloudinit/transforms/welcome.py
new file mode 100644
index 00000000..04691d21
--- /dev/null
+++ b/cloudinit/transforms/welcome.py
@@ -0,0 +1,64 @@
+# vi: ts=4 expandtab
+#
+# Copyright (C) 2012 Canonical Ltd.
+# Copyright (C) 2012 Hewlett-Packard Development Company, L.P.
+# Copyright (C) 2012 Yahoo! Inc.
+#
+# Author: Scott Moser <scott.moser@canonical.com>
+# Author: Juerg Haefliger <juerg.haefliger@hp.com>
+# Author: Joshua Harlow <harlowja@yahoo-inc.com>
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License version 3, as
+# published by the Free Software Foundation.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program. If not, see <http://www.gnu.org/licenses/>.
+
+from cloudinit.settings import PER_ALWAYS
+
+from cloudinit import templater
+from cloudinit import util
+from cloudinit import version
+
+import sys
+
+welcome_message_def = ("Cloud-init v. {{version}} starting stage {{stage}} at "
+ "{{timestamp}}. Up {{uptime}} seconds.")
+
+
+frequency = PER_ALWAYS
+
+
+def handle(_name, cfg, cloud, log, args):
+
+ welcome_msg = util.get_cfg_option_str(cfg, "welcome_msg")
+ if not welcome_msg:
+ tpl_fn = cloud.get_template_filename("welcome_msg")
+ if tpl_fn:
+ welcome_msg = util.load_file(tpl_fn)
+
+ if not welcome_msg:
+ welcome_msg = welcome_message_def
+
+ stage = "??"
+ if args:
+ stage = args[0]
+
+ tpl_params = {
+ 'stage': stage,
+ 'version': version.version_string(),
+ 'uptime': util.uptime(),
+ 'timestamp': util.time_rfc2822(),
+ }
+ try:
+ contents = templater.render_string(welcome_msg, tpl_params)
+ # TODO use log or sys.stderr??
+ sys.stderr.write("%s\n" % (contents))
+ except:
+ util.logexc(log, "Failed to render welcome message template")