From 49d3df468a94fef41a036dfa5d886449d180e006 Mon Sep 17 00:00:00 2001
From: Scott Moser <smoser@nelson>
Date: Thu, 7 Jan 2010 16:07:01 -0500
Subject: functional state now in kvm tests

supports user data as mime-multipart and handling types of
  text/x-shellscript  - execute at rc.local
  text/cloud-config   - feed to cloud config
  text/upstart-job    - add to /etc/init as upstart job
  text/x-include-url  - include urls as if they were in-line
---
 ec2init/DataSource.py | 18 ++++--------------
 1 file changed, 4 insertions(+), 14 deletions(-)

(limited to 'ec2init/DataSource.py')

diff --git a/ec2init/DataSource.py b/ec2init/DataSource.py
index 23e585e2..af5e9208 100644
--- a/ec2init/DataSource.py
+++ b/ec2init/DataSource.py
@@ -1,5 +1,6 @@
 
 import ec2init
+import UserDataHandler as ud
 
 class DataSource:
     userdata = None
@@ -9,21 +10,10 @@ class DataSource:
     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):
+    def get_userdata(self):
         if self.userdata == None:
-            self.userdata = ec2init.preprocess_user_data(self.userdata_raw)
-
+            self.userdata = ud.preprocess_userdata(self.userdata_raw)
         return self.userdata
 
-    def get_user_data_raw(self):
+    def get_userdata_raw(self):
         return(self.userdata_raw)
-- 
cgit v1.2.3