From 9b63a88deda6a2b26fc36068caf54459ab50b676 Mon Sep 17 00:00:00 2001
From: Yann Dirson <yann.dirson@vates.tech>
Date: Wed, 3 Jul 2024 14:26:37 +0200
Subject: [PATCH] restore: fix access to ESP partition

Commit 29c7bb54201d81b84ec17ddca5328e306173c7a0 introduced assumption
that `mounts` has a `esp` key, which was not the case when called from
`restoreFromBackup`.  This makes sure the key is created.

Signed-off-by: Yann Dirson <yann.dirson@vates.tech>
---
 restore.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/restore.py b/restore.py
index 39c99361..392cd0d4 100644
--- a/restore.py
+++ b/restore.py
@@ -214,6 +214,7 @@ def restore_partitions():
             # restore boot loader
             if boot_config.src_fmt == 'grub2':
                 if efi_boot:
+                    mounts['esp'] = esp
                     backend.setEfiBootEntry(mounts, disk_device, boot_partnum, constants.INSTALL_TYPE_RESTORE, branding)
                 else:
                     if location == constants.BOOT_LOCATION_MBR: