checking in some XSD work

aif/disk/mdadm.py

@@ -1,11 +1,59 @@
+import copy
+import subprocess
+##
+import mdstat
+##
+from aif.disk.block import Disk
+from aif.disk.block import Partition
+
+
+SUPPORTED_LEVELS = (0, 1, 4, 5, 6)
+
 class Member(object):
     def __init__(self, member_xml, partobj):
         self.xml = member_xml
         self.device = partobj
+        if not isinstance(self.device, (Partition, Disk, Array)):
+            raise ValueError(('partobj must be of type aif.disk.block.Partition, '
+                              'aif.disk.block.Disk, or aif.disk.mdadm.Array'))
         self.devpath = self.device.devpath
-        pass
+
+    def prepare(self):
+        # TODO: logging
+        subprocess.run(['mdadm', '--misc', '--zero-superblock', self.devpath])
+        return()
 
 class Array(object):
     def __init__(self, array_xml):
-        self.devpath = None
-        pass
+        self.xml = array_xml
+        self.id = array_xml.attrib['id']
+        self.level = int(array_xml.attrib['level'])
+        if self.level not in SUPPORTED_LEVELS:
+            raise ValueError('RAID level must be one of: {0}'.format(', '.join(SUPPORTED_LEVELS)))
+        self.devname = self.xml.attrib['name']
+        self.devpath = '/dev/md/{0}'.format(self.devname)
+        self.updateStatus()
+        self.members = []
+
+    def addMember(self, memberobj):
+        if not isinstance(memberobj, Member):
+            raise ValueError('memberobj must be of type aif.disk.mdadm.Member')
+
+    def assemble(self):
+        cmd = ['mdadm', '--assemble', self.devpath]
+        for m in self.members:
+            cmd.append(m.devpath)
+        subprocess.run(cmd)
+
+    def stop(self):
+        # TODO: logging
+        subprocess.run(['mdadm', '--stop', self.devpath])
+        return()
+
+    def updateStatus(self):
+        _info = mdstat.parse()
+        for k, v in _info['devices'].items():
+            if k != self.devname:
+                del(_info['devices'][k])
+        self.info = copy.deepcopy(_info)
+        return()