From ec712e77bccbb9a9965ce7876c919e263ddc8753 Mon Sep 17 00:00:00 2001
From: Scott Moser <smoser@nelson>
Date: Wed, 6 Jan 2010 21:00:37 -0500
Subject: add UserDataHandler.py, it handles includes and normalizing of
userdata
---
ec2init/UserDataHandler.py | 82 ++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 82 insertions(+)
create mode 100644 ec2init/UserDataHandler.py
diff --git a/ec2init/UserDataHandler.py b/ec2init/UserDataHandler.py
new file mode 100644
index 00000000..ec265af5
--- /dev/null
+++ b/ec2init/UserDataHandler.py
@@ -0,0 +1,82 @@
+import email
+
+from email.mime.multipart import MIMEMultipart
+from email.mime.text import MIMEText
+
+def do_include(str,parts):
+ import urllib
+ # is just a list of urls, one per line
+ for line in str.splitlines():
+ if line == "#include": continue
+ content = urllib.urlopen(line).read()
+ process_includes(email.message_from_string(content),parts)
+
+def process_includes(msg,parts):
+ # parts is a dictionary of arrays
+ # parts['content']
+ # parts['names']
+ # parts['types']
+ for t in ( 'content', 'names', 'types' ):
+ if not parts.has_key(t):
+ parts[t]=[ ]
+ for part in msg.walk():
+ # multipart/* are just containers
+ if part.get_content_maintype() == 'multipart':
+ continue
+ ctype = part.get_content_type()
+ if ctype is None:
+ # No guess could be made, or the file is encoded (compressed), so
+ # use a generic bag-of-bits type.
+ ctype = 'application/octet-stream'
+
+ if ctype == 'text/x-include-url' or \
+ part.get_payload().startswith("#include"):
+ do_include(part.get_payload(),parts)
+ continue
+ filename = part.get_filename()
+ if not filename:
+ filename = 'part-%03d' % len(parts['content'])
+
+ parts['content'].append(part.get_payload())
+ parts['types'].append(ctype)
+ parts['names'].append(filename)
+
+def parts2mime(parts):
+ outer = MIMEMultipart()
+
+ i = 0
+ while i < len(parts['content']):
+ if parts['types'][i] is None:
+ # No guess could be made, or the file is encoded (compressed), so
+ # use a generic bag-of-bits type.
+ ctype = 'application/octet-stream'
+ else: ctype = parts['types'][i]
+ maintype, subtype = ctype.split('/', 1)
+ if maintype == 'text':
+ msg = MIMEText(parts['content'][i], _subtype=subtype)
+ else:
+ msg = MIMEBase(maintype, subtype)
+ msg.set_payload(parts['content'][i])
+ # Encode the payload using Base64
+ encoders.encode_base64(msg)
+ # Set the filename parameter
+ msg.add_header('Content-Disposition', 'attachment',
+ filename=parts['names'][i])
+ outer.attach(msg)
+
+ i=i+1
+ return(outer.as_string())
+
+# this is heavily wasteful, reads through userdata string input
+def preprocess_userdata(str):
+ parts = { }
+ process_includes(email.message_from_string(data),parts)
+ return(parts2mime(parts))
+
+if __name__ == "__main__":
+ import sys
+ data = file(sys.argv[1]).read()
+ parts = { }
+ process_includes(email.message_from_string(data),parts)
+ print "#found %s parts" % len(parts['content'])
+ print parts2mime(parts)
--
cgit v1.2.3