From 857dab6b5dfc69cf398e6f8f873b4ef472b9e635 Mon Sep 17 00:00:00 2001
From: Scott Moser <smoser@nelson>
Date: Wed, 6 Jan 2010 01:40:29 -0500
Subject: support compressed user data, use cPickle instead of pickle

---
 ec2init/DataSource.py | 20 +++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

(limited to 'ec2init/DataSource.py')

diff --git a/ec2init/DataSource.py b/ec2init/DataSource.py
index da6170fd..b940c564 100644
--- a/ec2init/DataSource.py
+++ b/ec2init/DataSource.py
@@ -1,8 +1,26 @@
 
+import ec2init
+
 class DataSource:
+    userdata = None
+    metadata = None
+    userdata_raw = None
+
     def __init__(self):
        pass
 
+    def store_user_data_raw(self):
+        fp=fopen(user_data_raw,"wb")
+        fp.write(self.userdata_raw)
+        fp.close()
+    
+    def store_user_data(self):
+        fp=fopen(user_data,"wb")
+        fp.write(self.userdata)
+        fp.close()
+
     def get_user_data(self):
-        raise Exception("get_user_data Not-implemented")
+        if self.userdata == None:
+            self.userdata = ec2init.preprocess_user_data(self.userdata_raw)
 
+        return self.userdata
-- 
cgit v1.2.3