60 lines
1.8 KiB
Python
60 lines
1.8 KiB
Python
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
|
|
|
|
def prepare(self):
|
|
# TODO: logging
|
|
subprocess.run(['mdadm', '--misc', '--zero-superblock', self.devpath])
|
|
return()
|
|
|
|
class Array(object):
|
|
def __init__(self, array_xml):
|
|
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()
|