This commit is contained in:
brent s
2019-10-31 23:20:05 -04:00
parent 799ef58667
commit 9e5ff48926
8 changed files with 34 additions and 33 deletions

View File

@@ -4,10 +4,10 @@ import subprocess
##
import psutil
##
from aif.disk.block import Partition
from aif.disk.luks import LUKS
from aif.disk.lvm import LV as LVMVolume
from aif.disk.mdadm import Array as MDArray
import aif.disk.block
import aif.disk.luks
import aif.disk.lvm
import aif.disk.mdadm
# I wish there was a better way of doing this.
# https://unix.stackexchange.com/a/98680
@@ -44,7 +44,11 @@ for i in os.listdir(_mod_dir):
class FS(object):
def __init__(self, fs_xml, sourceobj):
self.xml = fs_xml
if not isinstance(sourceobj, (Partition, LUKS, LVMVolume, MDArray)):
if not isinstance(sourceobj, (aif.disk.block.Disk,
aif.disk.block.Partition,
aif.disk.luks.LUKS,
aif.disk.lvm.LV,
aif.disk.mdadm.Array)):
raise ValueError(('sourceobj must be of type '
'aif.disk.block.Partition, '
'aif.disk.luks.LUKS, '