113 lines
3.4 KiB
Python
113 lines
3.4 KiB
Python
import os
|
|
import subprocess
|
|
##
|
|
import psutil
|
|
##
|
|
import aif.disk.block_fallback as block
|
|
import aif.disk.luks_fallback as luks
|
|
import aif.disk.lvm_fallback as lvm
|
|
import aif.disk.mdadm_fallback as mdadm
|
|
import aif.utils
|
|
|
|
|
|
FS_FSTYPES = aif.utils.kernelFilesystems()
|
|
|
|
|
|
class FS(object):
|
|
def __init__(self, fs_xml, sourceobj):
|
|
self.xml = fs_xml
|
|
if not isinstance(sourceobj, (block.Disk,
|
|
block.Partition,
|
|
luks.LUKS,
|
|
lvm.LV,
|
|
mdadm.Array)):
|
|
raise ValueError(('sourceobj must be of type '
|
|
'aif.disk.block.Partition, '
|
|
'aif.disk.luks.LUKS, '
|
|
'aif.disk.lvm.LV, or'
|
|
'aif.disk.mdadm.Array'))
|
|
self.id = self.xml.attrib['id']
|
|
self.source = sourceobj
|
|
self.devpath = sourceobj.devpath
|
|
self.formatted = False
|
|
self.fstype = self.xml.attrib.get('type')
|
|
|
|
def format(self):
|
|
if self.formatted:
|
|
return ()
|
|
# This is a safeguard. We do *not* want to high-format a disk that is mounted.
|
|
aif.utils.checkMounted(self.devpath)
|
|
# TODO! Logging
|
|
cmd = ['mkfs',
|
|
'-t', self.fstype]
|
|
for o in self.xml.findall('opt'):
|
|
cmd.append(o.attrib['name'])
|
|
if o.text:
|
|
cmd.append(o.text)
|
|
cmd.append(self.devpath)
|
|
subprocess.run(cmd)
|
|
self.formatted = True
|
|
return()
|
|
|
|
|
|
class Mount(object):
|
|
def __init__(self, mount_xml, fsobj):
|
|
self.xml = mount_xml
|
|
self.id = self.xml.attrib['id']
|
|
if not isinstance(fsobj, FS):
|
|
raise ValueError('partobj must be of type aif.disk.filesystem.FS')
|
|
self.id = self.xml.attrib['id']
|
|
self.fs = fsobj
|
|
self.source = self.fs.devpath
|
|
self.target = os.path.realpath(self.xml.attrib['target'])
|
|
self.opts = {}
|
|
for o in self.xml.findall('opt'):
|
|
self.opts[o.attrib['name']] = o.text
|
|
self.mounted = False
|
|
|
|
def _parseOpts(self):
|
|
opts = []
|
|
for k, v in self.opts.items():
|
|
if v and v is not True: # Python's boolean determination is weird sometimes.
|
|
opts.append('{0}={1}'.format(k, v))
|
|
else:
|
|
opts.append(k)
|
|
return(opts)
|
|
|
|
def mount(self):
|
|
if self.mounted:
|
|
return()
|
|
os.makedirs(self.target, exist_ok = True)
|
|
opts = self._parseOpts()
|
|
# TODO: logging
|
|
cmd = ['/usr/bin/mount',
|
|
'--types', self.fs.fstype]
|
|
if opts:
|
|
cmd.extend(['--options', ','.join(opts)])
|
|
cmd.extend([self.source, self.target])
|
|
subprocess.run(cmd)
|
|
self.mounted = True
|
|
return()
|
|
|
|
def unmount(self, lazy = False, force = False):
|
|
self.updateMount()
|
|
if not self.mounted and not force:
|
|
return()
|
|
# TODO: logging
|
|
cmd = ['/usr/bin/umount']
|
|
if lazy:
|
|
cmd.append('--lazy')
|
|
if force:
|
|
cmd.append('--force')
|
|
cmd.append(self.target)
|
|
subprocess.run(cmd)
|
|
self.mounted = False
|
|
return()
|
|
|
|
def updateMount(self):
|
|
if self.source in [p.device for p in psutil.disk_partitions(all = True)]:
|
|
self.mounted = True
|
|
else:
|
|
self.mounted = False
|
|
return()
|