annif.transformer

Complexity

Total Complexity

6

Size/Duplication

Total Lines	46
Duplicated Lines	0 %

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
A	parse_specs()	0	16	3
A	get_transform()	0	10	3

"""Functionality for obtaining text transformation from string specification"""

import re
from . import transformer
from . import inputlimiter
from .transformer import IdentityTransform
from annif.util import parse_args
from annif.exception import ConfigurationException

__all__ = ["IdentityTransform"]


def parse_specs(transform_specs):
    """Parse a transformation specification into a list of tuples, e.g.
    'transf_1(x),transf_2(y=42),transf_3' is parsed to
    [(transf_1, [x], {}), (transf_2, [], {y: 42}), (transf_3, [], {})]."""

    parsed = []
    # Split by commas not inside parentheses
    parts = re.split(r',\s*(?![^()]*\))', transform_specs)
    for part in parts:
        match = re.match(r'(\w+)(\((.*)\))?', part)
        if match is None:
            continue
        transform = match.group(1)
        posargs, kwargs = parse_args(match.group(3))
        parsed.append((transform, posargs, kwargs))
    return parsed


def get_transform(transform_specs, project):
    transform_defs = parse_specs(transform_specs)
    transform_classes = []
    args = []
    for trans, posargs, kwargs in transform_defs:
        if trans not in _transforms:
            raise ConfigurationException(f"No such transform {trans}")
        transform_classes.append(_transforms[trans])
        args.append((posargs, kwargs))
    return transformer.TransformChain(transform_classes, args, project)


_transforms = {
    transformer.IdentityTransform.name: transformer.IdentityTransform,
    inputlimiter.InputLimiter.name: inputlimiter.InputLimiter}