#!/usr/bin/python3

import argparse

import sys

import blivet

POOL_NAME = "blivet_test"

parser = argparse.ArgumentParser()

parser.add_argument('-d', '--disks', default='')

args = parser.parse_args()

b = blivet.Blivet()

b.reset()

disks = list()

for disk_id in args.disks.split(','):

    disk = b.devicetree.resolve_device(disk_id)

    if not disk.is_disk:

        sys.stderr.write("specified disk '%s' is not a disk\n" % disk_id)

        sys.exit(1)

    disks.append(disk)

    b.initialize_disk(disk)

if len(disks) > 1:

    container_raid_level = "raid1"

    total_size = min(d.size for d in disks)

else:

    container_raid_level = None

    total_size = sum(d.size for d in disks)

lv1 = b.factory_device(size=total_size*0.8, disks=disks,

                       name="lv1", container_name=POOL_NAME,

                       fstype='xfs', device_type=blivet.devicefactory.DEVICE_TYPE_LVM)

lv2 = b.factory_device(disks=disks, name="lv2",

                       container_name=POOL_NAME, container_raid_level='raid1',

                       fstype='ext4', device_type=blivet.devicefactory.DEVICE_TYPE_LVM)

b.do_it()