From 674eee8a45c8c492afde327208f0c1d1fe3fd2a4 Mon Sep 17 00:00:00 2001
From: Soren Hansen <soren@canonical.com>
Date: Tue, 25 Aug 2009 17:38:05 +0200
Subject: Replace static content_type_handler dictionary with a decorator. LP:
 #376740

---
 ec2-run-user-data.py | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

(limited to 'ec2-run-user-data.py')

diff --git a/ec2-run-user-data.py b/ec2-run-user-data.py
index 51e0d68d..5af04249 100755
--- a/ec2-run-user-data.py
+++ b/ec2-run-user-data.py
@@ -25,8 +25,14 @@ import tempfile
 
 import ec2init
 
-content_type_handlers = { 'text/x-shellscript' : handle_shell_script,
-                          'text/x-ebs-mount-description' : handle_ebs_mount_description }
+content_type_handlers = {}
+
+def handler(mimetype):
+    return lambda f: register_handler(mimetype, f)
+
+def register_handler(mimetype, func):
+    content_type_handlers[mimetype] = func
+    return func
 
 def main():
     ec2 = ec2init.EC2Init()
@@ -52,6 +58,7 @@ def handle_unknown_payload(payload):
     if payload.startswith('#!'):
         content_type_handlers['text/x-shellscript'](payload)
 
+@handler('text/x-ebs-mount-description')
 def handle_ebs_mount_description(payload):
     (volume_description, path) = payload.split(':')
     (identifier_type, identifier) = volume_description.split('=')
@@ -65,6 +72,7 @@ def handle_ebs_mount_description(payload):
     else:
         return
 
+@handler('text/x-shellscript')
 def handle_shell_script(payload):
     (fd, path) = tempfile.mkstemp()
     fp = os.fdopen(fd, 'a')
-- 
cgit v1.2.3