summaryrefslogtreecommitdiff
path: root/src/op_mode/image_installer.py
diff options
context:
space:
mode:
authorJohn Estabrook <jestabro@vyos.io>2024-04-22 13:18:17 -0500
committerMergify <37929162+mergify[bot]@users.noreply.github.com>2024-04-23 04:18:26 +0000
commit8349d3e6a8cad1c282beb6a0552edf30cd8d2788 (patch)
treede59419f90907b0fe20ef3228acd932cdd971d61 /src/op_mode/image_installer.py
parent92baf98bf80682cdec7adcf19bdf0e89ab597ff1 (diff)
downloadvyos-1x-8349d3e6a8cad1c282beb6a0552edf30cd8d2788.tar.gz
vyos-1x-8349d3e6a8cad1c282beb6a0552edf30cd8d2788.zip
image-tools: T6260: remove persistence image directory if no space error
(cherry picked from commit c2fc2dba32ba861684f5e34635f810c56d551d51)
Diffstat (limited to 'src/op_mode/image_installer.py')
-rwxr-xr-xsrc/op_mode/image_installer.py11
1 files changed, 11 insertions, 0 deletions
diff --git a/src/op_mode/image_installer.py b/src/op_mode/image_installer.py
index c401a8c20..185d12c49 100755
--- a/src/op_mode/image_installer.py
+++ b/src/op_mode/image_installer.py
@@ -26,6 +26,7 @@ from os import environ
from typing import Union
from urllib.parse import urlparse
from passlib.hosts import linux_context
+from errno import ENOSPC
from psutil import disk_partitions
@@ -886,6 +887,16 @@ def add_image(image_path: str, vrf: str = None, username: str = '',
if set_as_default:
grub.set_default(image_name, root_dir)
+ except OSError as e:
+ # if no space error, remove image dir and cleanup
+ if e.errno == ENOSPC:
+ cleanup(mounts=[str(iso_path)],
+ remove_items=[f'{root_dir}/boot/{image_name}'])
+ else:
+ # unmount an ISO and cleanup
+ cleanup([str(iso_path)])
+ exit(f'Error: {e}')
+
except Exception as err:
# unmount an ISO and cleanup
cleanup([str(iso_path)])