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.
173 lines
7.3 KiB
173 lines
7.3 KiB
import re
|
|
|
|
# https://stackoverflow.com/questions/249791/regex-for-quoted-string-with-escaping-quotes
|
|
# https://stackoverflow.com/questions/21105360/regex-find-comma-not-inside-quotes
|
|
from collections import namedtuple
|
|
from typing import Dict, List, Optional, Container, Any, Union, Callable
|
|
|
|
_param_re = re.compile(
|
|
r'(([a-zA-Z]+)(!=|>=|<=|>|<|==|=)(("(?:[^"\\]|\\.)*"|\'(?:[^\'\\]|\\.)*\'|[^,\s]+)(,("(?:[^"\\]|\\.)*"|\'(?:[^\'\\]|\\.)*\'|[^,\s]+))*))')
|
|
# The intention of having both = and == is that they might have different behavior
|
|
# What that means depends on the usage
|
|
_param_operator_re = re.compile(r'!=|==|=|>|<|>=|<=')
|
|
_param_argument_re = re.compile(r'("(?:[^"\\]|\\.)*"|\'(?:[^\'\\]|\\.)*\'|[^,\s]+)')
|
|
_param_string_re = re.compile(r'("(?:[^"\\]|\\.)*"|\'(?:[^\'\\]|\\.)*\')')
|
|
_param_re_with_post_space = re.compile(
|
|
r'([a-zA-Z]+)(!=|==|=|>|<|>=|<=)(("(?:[^"\\]|\\.)*"|\'(?:[^\'\\]|\\.)*\'|[^,\s]+)(,("(?:[^"\\]|\\.)*"|\'(?:[^\'\\]|\\.)*\'|[^,\s]+))*) ?')
|
|
|
|
NamedArgument = namedtuple('NamedArgument', 'name operator value')
|
|
ArgumentValue = namedtuple('ArgumentValue', 'value operator')
|
|
|
|
|
|
def _parse_named_argument(arg):
|
|
groups = _param_re.fullmatch(arg).groups()
|
|
name = groups[1]
|
|
operator = groups[2]
|
|
values = [value[1:-1] if _param_string_re.fullmatch(value) else value for value in
|
|
_param_argument_re.findall(groups[3])]
|
|
return NamedArgument(name, operator, values)
|
|
|
|
|
|
def parse_arguments(arg):
|
|
named_arguments_parsed = [_parse_named_argument(na[0]) for na in _param_re.findall(arg)]
|
|
text_argument = _param_re_with_post_space.sub('', arg)
|
|
named_arguments = {}
|
|
for na in named_arguments_parsed:
|
|
if na.name not in named_arguments:
|
|
named_arguments[na.name] = []
|
|
named_arguments[na.name].append(ArgumentValue(na.value, na.operator))
|
|
return ParsedArguments(text_argument.strip(), named_arguments)
|
|
|
|
|
|
class ArgumentError(Exception):
|
|
pass
|
|
|
|
|
|
class ParsedArguments:
|
|
text_argument: str
|
|
named_arguments: Dict[str, List[ArgumentValue]]
|
|
|
|
def __init__(self, text, named_arguments):
|
|
self.text_argument = text
|
|
self.named_arguments = named_arguments
|
|
self.used = set()
|
|
|
|
def single(self, names: Union[List[str], str], default: Any = None, allowed_operators: Optional[Container] = None,
|
|
is_list=False, numeric=False, converter: Union[dict, Callable] = lambda n: n):
|
|
if allowed_operators is None:
|
|
allowed_operators = {'>', '<', '>=', '<=', '!=', '==', '='}
|
|
if not isinstance(default, tuple) and default is not None:
|
|
default = ArgumentValue(default, '=')
|
|
if not isinstance(names, list):
|
|
names = [names]
|
|
for name in names:
|
|
self.used.add(name)
|
|
name = f'{names[0]} ({", ".join(names[1:])})' if len(names) > 1 else names[0]
|
|
value = [arg for args in (self.named_arguments.get(name) for name in names) if args for arg in args]
|
|
if not value:
|
|
return default
|
|
if len(value) != 1:
|
|
raise ArgumentError(f'Expected only one value for parameter "{name}".')
|
|
value = value[0]
|
|
if value.operator not in allowed_operators:
|
|
raise ArgumentError(
|
|
f'Allowed operators for parameter "{name}" are {", ".join(str(o) for o in allowed_operators)}.')
|
|
if numeric:
|
|
try:
|
|
value = ArgumentValue([float(v) for v in value.value], value.operator)
|
|
except ValueError:
|
|
raise ArgumentError(f'Expected numerical arguments for parameter "{name}".')
|
|
try:
|
|
if isinstance(converter, dict):
|
|
value = ArgumentValue([converter[v] for v in value.value], value.operator)
|
|
else:
|
|
value = ArgumentValue([converter(v) for v in value.value], value.operator)
|
|
except Exception:
|
|
raise ArgumentError(f'Invalid value for parameter "{name}".')
|
|
if not is_list:
|
|
if len(value.value) != 1:
|
|
raise ArgumentError(f'List not allowed for parameter "{name}".')
|
|
value = ArgumentValue(value.value[0], value.operator)
|
|
return value
|
|
|
|
def repeatable(self, names: Union[List[str], str], default: Any = None, allowed_operators: Optional[Container] = None,
|
|
is_list=False, numeric=False, converter: Union[dict, Callable] = lambda n: n):
|
|
if allowed_operators is None:
|
|
allowed_operators = {'>', '<', '>=', '<=', '!=', '==', '='}
|
|
if not isinstance(default, tuple) and default is not None:
|
|
default = [ArgumentValue(default, '=')]
|
|
if default is None:
|
|
default = []
|
|
if not isinstance(names, list):
|
|
names = [names]
|
|
for name in names:
|
|
self.used.add(name)
|
|
name = f'{names[0]} ({", ".join(names[1:])})' if len(names) > 1 else names[0]
|
|
values = [arg for args in (self.named_arguments.get(name) for name in names) if args for arg in args]
|
|
if not values:
|
|
return default
|
|
if any(value.operator not in allowed_operators for value in values):
|
|
raise ArgumentError(
|
|
f'Allowed operators for parameter "{name}" are {", ".join(str(o) for o in allowed_operators)}.')
|
|
if numeric:
|
|
try:
|
|
values = [ArgumentValue([float(v) for v in value.value], value.operator) for value in values]
|
|
except ValueError:
|
|
raise ArgumentError(f'Expected numerical arguments for parameter "{name}".')
|
|
try:
|
|
if isinstance(converter, dict):
|
|
values = [ArgumentValue([converter[v] for v in value.value], value.operator) for value in values]
|
|
else:
|
|
values = [ArgumentValue([converter(v) for v in value.value], value.operator) for value in values]
|
|
except Exception:
|
|
raise ArgumentError(f'Invalid value for parameter "{name}".')
|
|
if not is_list:
|
|
if any(len(value.value) != 1 for value in values):
|
|
raise ArgumentError(f'List not allowed for parameter "{name}".')
|
|
values = [ArgumentValue(value.value[0], value.operator) for value in values]
|
|
return values
|
|
|
|
def has_unused(self):
|
|
return any(name not in self.used for name in self.named_arguments.keys())
|
|
|
|
def require_all_arguments_used(self):
|
|
def quote(s):
|
|
return f'"{s}"'
|
|
|
|
if self.has_unused():
|
|
raise ArgumentError(
|
|
f'Unkown arguments with names {", ".join(quote(v) for v in self.named_arguments.keys() if v not in self.used)}.')
|
|
|
|
|
|
_operators = {
|
|
'=': lambda a, b: a == b,
|
|
'==': lambda a, b: a == b,
|
|
'!=': lambda a, b: a != b,
|
|
'>': lambda a, b: a > b,
|
|
'<': lambda a, b: a < b,
|
|
'>=': lambda a, b: a >= b,
|
|
'<=': lambda a, b: a <= b,
|
|
}
|
|
|
|
_list_operators = {
|
|
'=': lambda a, b: any(a == v for v in b),
|
|
'==': lambda a, b: all(a == v for v in b),
|
|
'!=': lambda a, b: all(a != v for v in b),
|
|
'>': lambda a, b: all(a > v for v in b),
|
|
'<': lambda a, b: all(a < v for v in b),
|
|
'>=': lambda a, b: all(a >= v for v in b),
|
|
'<=': lambda a, b: all(a <= v for v in b),
|
|
}
|
|
|
|
|
|
def operator_for(operator: str):
|
|
return _operators[operator]
|
|
|
|
|
|
def list_operator_for(operator: str):
|
|
return _list_operators[operator]
|
|
|
|
|
|
if __name__ == '__main__':
|
|
a = (parse_arguments(r'sort=default rating>=13.5 a name="a",b," asf,ds ",\'sdf\',dsf'))
|
|
print(a)
|
|
|