diff options
Diffstat (limited to 'scripts/lib/wic/engine.py')
-rw-r--r-- | scripts/lib/wic/engine.py | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/scripts/lib/wic/engine.py b/scripts/lib/wic/engine.py index fe036f60e9..f0c5ff0aaf 100644 --- a/scripts/lib/wic/engine.py +++ b/scripts/lib/wic/engine.py | |||
@@ -416,6 +416,7 @@ class Disk: | |||
416 | 416 | ||
417 | # calculate expanded partitions sizes | 417 | # calculate expanded partitions sizes |
418 | sizes = {} | 418 | sizes = {} |
419 | num_auto_resize = 0 | ||
419 | for num, part in enumerate(parts['partitiontable']['partitions'], 1): | 420 | for num, part in enumerate(parts['partitiontable']['partitions'], 1): |
420 | if num in expand: | 421 | if num in expand: |
421 | if expand[num] != 0: # don't resize partition if size is set to 0 | 422 | if expand[num] != 0: # don't resize partition if size is set to 0 |
@@ -425,10 +426,11 @@ class Disk: | |||
425 | sizes[num] = sectors | 426 | sizes[num] = sectors |
426 | elif part['type'] != 'f': | 427 | elif part['type'] != 'f': |
427 | sizes[num] = -1 | 428 | sizes[num] = -1 |
429 | num_auto_resize += 1 | ||
428 | 430 | ||
429 | for num, part in enumerate(parts['partitiontable']['partitions'], 1): | 431 | for num, part in enumerate(parts['partitiontable']['partitions'], 1): |
430 | if sizes.get(num) == -1: | 432 | if sizes.get(num) == -1: |
431 | part['size'] += free // len(sizes) | 433 | part['size'] += free // num_auto_resize |
432 | 434 | ||
433 | # write resized partition table to the target | 435 | # write resized partition table to the target |
434 | write_ptable(parts, target) | 436 | write_ptable(parts, target) |