# -*- coding: utf-8 -*-
"""The arguments helper interface."""
import locale
import sys
from plaso.lib import errors
[docs]
class ArgumentsHelper(object):
"""CLI arguments helper."""
NAME = 'baseline'
# Category further divides the registered helpers down after function,
# this can be something like: analysis, output, storage, etc.
CATEGORY = ''
DESCRIPTION = ''
_PREFERRED_ENCODING = 'UTF-8'
# pylint gets confused by the multi type return value.
# pylint: disable=missing-return-type-doc
@classmethod
def _ParseNumericOption(cls, options, argument_name, default_value=None):
"""Parses a numeric command line argument.
Args:
options (argparse.Namespace): parser options.
argument_name (str): name of the command line argument.
default_value (Optional[float|int]): default value of the command line
argument.
Returns:
float|int: command line argument value or the default value if the command
line argument is not set
Raises:
BadConfigOption: if the command line argument value is not a
floating-point or integer.
"""
argument_value = getattr(options, argument_name, None)
if argument_value is None:
return default_value
if not isinstance(argument_value, (float, int)):
raise errors.BadConfigOption((
f'Unsupported option: {argument_name:s} floating-point or integer '
f'type required.'))
return argument_value
# pylint: enable=missing-return-type-doc
@classmethod
def _ParseStringOption(cls, options, argument_name, default_value=None):
"""Parses a string command line argument.
Args:
options (argparse.Namespace): parser options.
argument_name (str): name of the command line argument.
default_value (Optional[str]): default value of the command line argument.
Returns:
str: command line argument value or the default value if the command line
argument is not set
Raises:
BadConfigOption: if the command line argument value cannot be converted
to a Unicode string.
"""
argument_value = getattr(options, argument_name, None)
if argument_value is None:
return default_value
if isinstance(argument_value, bytes):
encoding = sys.stdin.encoding
# Note that sys.stdin.encoding can be None.
if not encoding:
encoding = locale.getpreferredencoding()
if not encoding:
encoding = cls._PREFERRED_ENCODING
try:
argument_value = argument_value.decode(encoding)
except UnicodeDecodeError as exception:
raise errors.BadConfigOption((
f'Unable to convert option: {argument_name:s} to Unicode with '
f'error: {exception!s}'))
elif not isinstance(argument_value, str):
raise errors.BadConfigOption(
f'Unsupported option: {argument_name:s} string type required.')
return argument_value
[docs]
@classmethod
def AddArguments(cls, argument_group):
"""Adds command line arguments to an argument group.
This function takes an argument parser or an argument group object and adds
to it all the command line arguments this helper supports.
Args:
argument_group (argparse._ArgumentGroup|argparse.ArgumentParser):
argparse group.
"""
[docs]
@classmethod
def ParseOptions(cls, options, configuration_object):
"""Parses and validates options.
Args:
options (argparse.Namespace): parser options.
configuration_object (object): object to be configured by the argument
helper.
Raises:
BadConfigObject: when the configuration object is of the wrong type.
BadConfigOption: when a configuration parameter fails validation.
"""