fixes found testing in ec2

1 files changed, 3 insertions, 8 deletions

diff --git a/ec2init/__init__.py b/ec2init/__init__.py
index d9d24969..a0ef24c2 100644
--- a/ec2init/__init__.py
+++ b/ec2init/__init__.py
@@ -80,7 +80,7 @@ class EC2Init:
                 s = source()
                 if s.get_data():
                     self.datasource = s
-                    return
+                    return True
             except Exception as e:
                 pass
         raise Exception("Could not find data source")
@@ -93,13 +93,8 @@ class EC2Init:
         self.store_userdata()
 
     def store_userdata(self):
-        f = open(userdata_raw,"wb")
-        f.write(self.datasource.get_userdata_raw())
-        f.close()
-
-        f = open(userdata,"wb")
-        f.write(self.get_userdata())
-        f.close()
+        write_file(userdata_raw, self.datasource.get_userdata_raw(), 0644)
+        write_file(userdata, self.datasource.get_userdata(), 0644)
 
     def get_cfg_option_bool(self, key, default=None):
         val = self.config.get(key, default)