209 lines
7.5 KiB
Python
209 lines
7.5 KiB
Python
# pylint: disable=too-few-public-methods
|
|
|
|
#
|
|
# The Qubes OS Project, http://www.qubes-os.org
|
|
#
|
|
# Copyright (C) 2016 Bahtiar `kalkin-` Gadimov <bahtiar@gadimov.de>
|
|
#
|
|
# This program is free software; you can redistribute it and/or modify
|
|
# it under the terms of the GNU Lesser General Public License as published by
|
|
# the Free Software Foundation; either version 2.1 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public License along
|
|
# with this program; if not, see <http://www.gnu.org/licenses/>.
|
|
|
|
'''Manages Qubes pools and their options'''
|
|
|
|
from __future__ import print_function
|
|
|
|
import argparse
|
|
import sys
|
|
|
|
import qubesadmin
|
|
import qubesadmin.exc
|
|
import qubesadmin.tools
|
|
|
|
|
|
class _Info(qubesadmin.tools.PoolsAction):
|
|
''' Action for argument parser that displays pool info and exits. '''
|
|
|
|
def __init__(self, option_strings, help='print pool info and exit',
|
|
**kwargs):
|
|
# pylint: disable=redefined-builtin
|
|
super().__init__(option_strings, help=help, **kwargs)
|
|
|
|
def __call__(self, parser, namespace, values, option_string=None):
|
|
setattr(namespace, 'command', 'info')
|
|
super().__call__(parser, namespace, values, option_string)
|
|
|
|
|
|
def pool_info(pool):
|
|
''' Prints out pool name and config '''
|
|
data = [("name", pool.name)]
|
|
data += [i for i in sorted(pool.config.items()) if i[0] != 'name']
|
|
qubesadmin.tools.print_table(data)
|
|
|
|
|
|
def list_pools(app):
|
|
''' Prints out all known pools and their drivers '''
|
|
result = [('NAME', 'DRIVER')]
|
|
for pool in app.pools.values():
|
|
result += [(pool.name, pool.driver)]
|
|
qubesadmin.tools.print_table(result)
|
|
|
|
|
|
class _Remove(argparse.Action):
|
|
''' Action for argument parser that removes a pool '''
|
|
|
|
def __init__(self, option_strings, dest=None, default=None, metavar=None):
|
|
super().__init__(option_strings=option_strings,
|
|
dest=dest,
|
|
metavar=metavar,
|
|
default=default,
|
|
help='remove pool')
|
|
|
|
def __call__(self, parser, namespace, name, option_string=None):
|
|
setattr(namespace, 'command', 'remove')
|
|
setattr(namespace, 'name', name)
|
|
|
|
|
|
class _Add(argparse.Action):
|
|
''' Action for argument parser that adds a pool. '''
|
|
|
|
def __init__(self, option_strings, dest=None, default=None, metavar=None):
|
|
super().__init__(option_strings=option_strings,
|
|
dest=dest,
|
|
metavar=metavar,
|
|
default=default,
|
|
nargs=2,
|
|
help='add pool')
|
|
|
|
def __call__(self, parser, namespace, values, option_string=None):
|
|
name, driver = values
|
|
setattr(namespace, 'command', 'add')
|
|
setattr(namespace, 'name', name)
|
|
setattr(namespace, 'driver', driver)
|
|
|
|
|
|
class _Set(qubesadmin.tools.PoolsAction):
|
|
''' Action for argument parser that sets pool options. '''
|
|
|
|
def __init__(self, option_strings, dest=None, default=None, metavar=None):
|
|
super().__init__(option_strings=option_strings,
|
|
dest=dest,
|
|
metavar=metavar,
|
|
default=default,
|
|
help='modify pool (use -o to specify '
|
|
'modifications)')
|
|
|
|
def __call__(self, parser, namespace, name, option_string=None):
|
|
setattr(namespace, 'command', 'set')
|
|
super().__call__(parser, namespace, name, option_string)
|
|
|
|
|
|
class _Options(argparse.Action):
|
|
''' Action for argument parser that parsers options. '''
|
|
|
|
def __init__(self, option_strings, dest, default, metavar='options'):
|
|
super().__init__(
|
|
option_strings=option_strings,
|
|
dest=dest,
|
|
metavar=metavar,
|
|
default=default,
|
|
help='comma-separated list of driver options')
|
|
|
|
def __call__(self, parser, namespace, options, option_string=None):
|
|
setattr(namespace, 'options',
|
|
dict([option.split('=', 1) for option in options.split(',')]))
|
|
|
|
|
|
def get_parser():
|
|
''' Parses the provided args '''
|
|
parser = qubesadmin.tools.QubesArgumentParser(description=__doc__)
|
|
parser.add_argument('-o', action=_Options, dest='options', default={})
|
|
group = parser.add_mutually_exclusive_group()
|
|
group.add_argument('-l',
|
|
'--list',
|
|
dest='command',
|
|
const='list',
|
|
action='store_const',
|
|
help='list all pools and exit (default action)')
|
|
group.add_argument('-i', '--info', metavar='POOLNAME', dest='pools',
|
|
action=_Info, default=[])
|
|
group.add_argument('-a',
|
|
'--add',
|
|
action=_Add,
|
|
dest='command',
|
|
metavar=('NAME', 'DRIVER'))
|
|
group.add_argument('-r', '--remove', metavar='NAME', action=_Remove)
|
|
group.add_argument('-s', '--set', metavar='POOLNAME', dest='pool',
|
|
action=_Set, default=[])
|
|
group.add_argument('--help-drivers',
|
|
dest='command',
|
|
const='list-drivers',
|
|
action='store_const',
|
|
help='list all drivers with their options and exit')
|
|
return parser
|
|
|
|
|
|
def main(args=None, app=None):
|
|
'''Main routine of :program:`qvm-pools`.
|
|
|
|
:param list args: Optional arguments to override those delivered from \
|
|
command line.
|
|
'''
|
|
parser = get_parser()
|
|
try:
|
|
args = parser.parse_args(args, app=app)
|
|
except qubesadmin.exc.QubesException as e:
|
|
parser.print_error(str(e))
|
|
return 1
|
|
|
|
if args.command is None or args.command == 'list':
|
|
list_pools(args.app)
|
|
elif args.command == 'list-drivers':
|
|
result = [('DRIVER', 'OPTIONS')]
|
|
for driver in sorted(args.app.pool_drivers):
|
|
params = args.app.pool_driver_parameters(driver)
|
|
driver_options = ', '.join(params)
|
|
result += [(driver, driver_options)]
|
|
qubesadmin.tools.print_table(result)
|
|
elif args.command == 'add':
|
|
try:
|
|
args.app.add_pool(name=args.name, driver=args.driver,
|
|
**args.options)
|
|
except qubesadmin.exc.QubesException as e:
|
|
parser.error('failed to add pool %s: %s\n' % (args.name, str(e)))
|
|
elif args.command == 'remove':
|
|
try:
|
|
args.app.remove_pool(args.name)
|
|
except KeyError:
|
|
parser.print_error('no such pool %s\n' % args.name)
|
|
except qubesadmin.exc.QubesException as e:
|
|
parser.error('failed to remove pool %s: %s\n' % (args.name, str(e)))
|
|
elif args.command == 'info':
|
|
for pool in args.pools:
|
|
pool_info(pool)
|
|
elif args.command == 'set':
|
|
pool = args.pool[0]
|
|
for opt, value in args.options.items():
|
|
if not hasattr(type(pool), opt):
|
|
parser.error('setting pool option %s is not supported' % (
|
|
pool.name))
|
|
try:
|
|
setattr(pool, opt, value)
|
|
except qubesadmin.exc.QubesException as e:
|
|
parser.error('failed to set pool %s option %s: %s\n' % (
|
|
pool.name, opt, str(e)))
|
|
return 0
|
|
|
|
|
|
if __name__ == '__main__':
|
|
sys.exit(main())
|