summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
-rw-r--r--src/live/ogOperations.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/src/live/ogOperations.py b/src/live/ogOperations.py
index a576389..1b2c3dd 100644
--- a/src/live/ogOperations.py
+++ b/src/live/ogOperations.py
@@ -41,12 +41,15 @@ class OgLiveOperations:
def _refresh_payload_disk(self, cxt, part_setup, num_disk):
part_setup['disk'] = str(num_disk)
part_setup['disk_type'] = 'DISK'
- part_setup['code'] = '2' if cxt.label.name == 'gpt' else '1'
part_setup['partition'] = '0'
part_setup['filesystem'] = ''
part_setup['os'] = ''
part_setup['size'] = str(cxt.nsectors * cxt.sector_size // 1024)
part_setup['used_size'] = '0'
+ if not cxt.label:
+ part_setup['code'] = '0'
+ else:
+ part_setup['code'] = '2' if cxt.label.name == 'gpt' else '1'
def _refresh_payload_partition(self, cxt, pa, part_setup, disk):
parttype = cxt.partition_to_string(pa, fdisk.FDISK_FIELD_TYPEID)