elbepack: hdimg: simplify grub device.map handling

Use some helper functions to simplify the code. Signed-off-by: Thomas Weißschuh <[email protected]>
Linutronix · May 10, 2024 · eaa431d · eaa431d
1 parent 5c5954f
commit eaa431d
Showing 1 changed file with 6 additions and 16 deletions.
diff --git a/elbepack/hdimg.py b/elbepack/hdimg.py
@@ -161,11 +161,9 @@ def install(self, target, user_args):
                     stack.enter_context(
                         mount(bindmnt, imagemntfs.fname(bindmnt), bind=True))
 
-                do(f'mkdir -p "{imagemntfs.fname("boot/grub")}"')
-
-                devmap = open(imagemntfs.fname('boot/grub/device.map'), 'w')
-                devmap.write(f'(hd0) {loopdev}\n')
-                devmap.close()
+                imagemntfs.mkdir_p('boot/grub')
+                imagemntfs.write_file('boot/grub/device.map', 0o644, f'(hd0) {loopdev}\n')
+                stack.callback(imagemntfs.remove, 'boot/grub/device.map')
 
                 chroot(imagemnt, ['update-grub2'])
 
@@ -188,9 +186,6 @@ def install(self, target, user_args):
             except subprocess.CalledProcessError as E:
                 logging.error('Fail installing grub device: %s', E)
 
-            finally:
-                os.unlink(imagemntfs.fname('boot/grub/device.map'))
-
 
 class grubinstaller97(grubinstaller_base):
 
@@ -221,11 +216,9 @@ def install(self, target, user_args):
                     stack.enter_context(
                         mount(bindmnt, imagemntfs.fname(bindmnt), bind=True))
 
-                do(f'mkdir -p "{imagemntfs.fname("boot/grub")}"')
-
-                devmap = open(imagemntfs.fname('boot/grub/device.map'), 'w')
-                devmap.write(f'(hd0) {loopdev}\n')
-                devmap.close()
+                imagemntfs.mkdir_p('boot/grub')
+                imagemntfs.write_file('boot/grub/device.map', 0o644, f'(hd0) {loopdev}\n')
+                stack.callback(imagemntfs.remove, 'boot/grub/device.map')
 
                 # Replace groot and kopt because else they will be given
                 # bad values
@@ -241,9 +234,6 @@ def install(self, target, user_args):
             except subprocess.CalledProcessError as E:
                 logging.error('Fail installing grub device: %s', E)
 
-            finally:
-                os.unlink(imagemntfs.fname('boot/grub/device.map'))
-
 
 class simple_fstype:
     def __init__(self, typ):