Fix GRUB2._autogen AttributeError

GRUB2._autogen should and can be determined on initialization, rather than when retrieving the list of kernels (which may or may not happen before the attribute is accessed). Signed-off-by: Konstantinos Smanis <[email protected]> Closes: #6 Signed-off-by: Michał Górny <[email protected]>
projg2 · Sep 22, 2020 · 25349e8 · 25349e8
1 parent b0cace4
commit 25349e8
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ecleankernel/bootloader/grub2.py b/ecleankernel/bootloader/grub2.py
@@ -20,11 +20,13 @@ class GRUB2(GRUB):
     kernel_re = r'^\s*linux\s*(\([^)]+\))?(?P<path>\S+)'
     def_path = ('/boot/grub/grub.cfg', '/boot/grub2/grub.cfg')
 
+    def __init__(self) -> None:
+        super().__init__()
+        self._autogen = self._content.startswith(grub2_autogen_header)
+
     def _get_kernels(self,
                      content: str
                      ) -> typing.Iterable[str]:
-        self._autogen = content.startswith(grub2_autogen_header)
-
         if self._autogen:
             logging.debug('Config is autogenerated, ignoring')
             return ()