A collection of classes and commands for automated command line scripting using Python.
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

91 lines
2.9 KiB

# Imports
from commonkit import parse_jinja_template, read_file
from configparser import ConfigParser, ParsingError
import logging
from .base import BaseLoader
log = logging.getLogger(__name__)
# Exports
__all__ = (
"INILoader",
)
# Classes
class INILoader(BaseLoader):
"""Load commands from an INI file."""
def load(self):
"""Load the INI file.
:rtype: bool
"""
if not self.exists:
log.warning("Input file does not exist: %s" % self.path)
return False
content = self.read_file()
if content is None:
return False
ini = ConfigParser()
try:
ini.read_string(content)
except ParsingError as e:
log.error("Failed to parse %s as an INI file: %s" % (self.path, e))
return False
for comment in ini.sections():
args = list()
command_name = None
count = 0
kwargs = self.options.copy()
kwargs['comment'] = comment
for key, value in ini.items(comment):
# Is there a reason for this?
# if key.startswith("_"):
# continue
# The first key/value pair is the command name and arguments.
if count == 0:
command_name = key
# kwargs['name'] = command_name
# Explanations and screenshots aren't processed like commands, so the text need not be surrounded
# by double quotes.
if command_name in ("explain", "screenshot"):
args.append(value)
count += 1
continue
# Arguments surrounded by quotes are considered to be one argument. All others are split into a
# list to be passed to the command factory. It is also possible that this is a call where no
# arguments are present, so the whole thing is wrapped to protect against an index error. A TypeError is raised in
# cases where a command is provided with no positional arguments; we interpret this as True.
try:
if value[0] == '"':
args.append(value.replace('"', ""))
else:
args = value.split(" ")
except IndexError:
pass
# except TypeError:
# noinspection PyTypeChecker
# args.append(True)
else:
_key, _value = self._get_key_value(key, value)
kwargs[_key] = _value
count += 1
self.commands.append((command_name, args, kwargs))
self.is_loaded = True
return True