Fix AdjustExpandRootScript

Also, only add it when it is required

bootstrapvz/common/tasks/initd.py

@@ -1,6 +1,5 @@
 from bootstrapvz.base import Task
 from .. import phases
-from ..exceptions import TaskError
 from ..tools import log_check_call
 from . import assets
 import os.path
@@ -57,14 +56,9 @@ class AdjustExpandRootScript(Task):
 
 	@classmethod
 	def run(cls, info):
-		if 'expand-root' not in info.initd['install']:
+		import os.path
-			raise TaskError('The expand-root script was not installed')
+		from ..tools import sed_i
-
+		script = os.path.join(info.root, 'etc/init.d/expand-root')
-		from bootstrapvz.base.fs.partitionmaps.none import NoPartitions
+		root_idx = info.volume.partition_map.root.get_index()
-		if not isinstance(info.volume.partition_map, NoPartitions):
+		device_path = 'device_path="/dev/xvda{idx}"'.format(idx=root_idx)
-			import os.path
+		sed_i(script, '^device_path="/dev/xvda$', device_path)
-			from ..tools import sed_i
-			script = os.path.join(info.root, 'etc/init.d.expand-root')
-			root_idx = info.volume.partition_map.root.get_index()
-			device_path = 'device_path="/dev/xvda{idx}"'.format(idx=root_idx)
-			sed_i(script, '^device_path="/dev/xvda$', device_path)

bootstrapvz/providers/ec2/__init__.py

@@ -60,11 +60,13 @@ def resolve_tasks(taskset, manifest):
 	                initd.RemoveHWClock,
 	                tasks.initd.AddEC2InitScripts,
 	                initd.InstallInitScripts,
-	                initd.AdjustExpandRootScript,
 
 	                tasks.ami.RegisterAMI,
 	                ])
 
+	if manifest.volume['partitions']['type'] != 'none':
+		taskset.add(initd.AdjustExpandRootScript)
+
 	if manifest.system['bootloader'] == 'pvgrub':
 		taskset.add(boot.AddGrubPackage)
 		taskset.add(tasks.boot.ConfigurePVGrub)