kevinheyer/bootstrap-vz
1
0
Fork 0
mirror of https://github.com/kevingruesser/bootstrap-vz.git synced 2025-08-22 18:00:35 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

grub: Prettier type checking when writing grub defaults

Browse source
This commit is contained in:
Anders Ingemann 2016-06-05 13:20:34 +02:00
parent e030b9f84f
commit 2ffc4dd6c3
1 changed files with 9 additions and 15 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

24
bootstrapvz/common/tasks/grub.py
View file

@ -200,25 +200,19 @@ class WriteGrubConfig(Task):
def run(cls, info):
grub_config_contents = ''
for key, value in info.grub_config.items():
if value is None:
continue
if isinstance(value, str):
grub_config_contents += '{}="{}"\n'.format(key, value)
continue
if isinstance(value, int):
elif isinstance(value, int):
grub_config_contents += '{}={}\n'.format(key, value)
continue
if isinstance(value, bool):
elif isinstance(value, bool):
grub_config_contents += '{}="{}"\n'.format(key, str(value).lower())
continue
if isinstance(value, list):
if len(value) == 0:
continue
args_list = ' '.join(map(str, value))
grub_config_contents += '{}="{}"\n'.format(key, args_list)
continue
raise TaskError('Don\'t know how to handle type {}, '
'when creating grub config'.format(type(value)))
elif isinstance(value, list):
if len(value) > 0:
args_list = ' '.join(map(str, value))
grub_config_contents += '{}="{}"\n'.format(key, args_list)
elif value is not None:
raise TaskError('Don\'t know how to handle type {}, '
'when creating grub config'.format(type(value)))
grub_defaults = os.path.join(info.root, 'etc/default/grub')
with open(grub_defaults, 'w') as grub_defaults_handle:
grub_defaults_handle.write(grub_config_contents)