diff --git a/rwx/grub/__init__.py b/rwx/grub/__init__.py index b931ee3..a11cf7d 100644 --- a/rwx/grub/__init__.py +++ b/rwx/grub/__init__.py @@ -1,17 +1,17 @@ import cmd import ps -cmd.need('grub-mkimage') +cmd.need("grub-mkimage") -COMPRESSION = 'xz' +COMPRESSION = "xz" ENV_BYTES = 1024 -ENV_COMMENT = '#' -ENV_HEADER = f'''{ENV_COMMENT} GRUB Environment Block -''' +ENV_COMMENT = "#" +ENV_HEADER = f"""{ENV_COMMENT} GRUB Environment Block +""" MODULES = { - 'i386-pc': [ - ('biosdisk',), - ('ntldr',), + "i386-pc": [ + ("biosdisk",), + ("ntldr",), ] } @@ -19,14 +19,14 @@ MODULES = { def make_image(image_format: str, image_path: str, modules: list[str], memdisk_path: str, pubkey_path: str = None) -> None: args = [ - ('grub-mkimage',), - ('--compress', COMPRESSION), - ('--format', image_format), - ('--output', image_path), - ('--memdisk', memdisk_path), + ("grub-mkimage",), + ("--compress", COMPRESSION), + ("--format", image_format), + ("--output", image_path), + ("--memdisk", memdisk_path), ] if pubkey_path: - args.append(('--pubkey', pubkey_path)) + args.append(("--pubkey", pubkey_path)) args.extend(modules) if modules := MODULES.get(image_format, None): args.extend(modules)