annif.transformer.transformer

Complexity

Total Complexity

11

Size/Duplication

Total Lines	53
Duplicated Lines	0 %

Importance

Changes

0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	IdentityTransformer.transform_text()	0	3	1
A	Transformer.__init__()	0	3	1
A	IdentityTransformer.__init__()	0	2	1
A	IdentityTransformer.transform_corpus()	0	3	1
A	Transformer._init_transformers()	0	12	3
A	Transformer.transform_text()	0	4	2
A	Transformer.transform_corpus()	0	4	2

"""Common functionality for input transforming."""

from annif.corpus import TransformingDocumentCorpus
from annif.exception import ConfigurationException


class IdentityTransformer():
    """"""""  # TODO

    name = 'pass_through'

    def __init__(self, project):
        self.project = project

    def transform_text(self, text):
        """"""  # TODO
        return text

    def transform_corpus(self, corpus):
        """"""  # TODO
        return TransformingDocumentCorpus(corpus, self.transform_text)


class Transformer():
    """"""  # TODO

    def __init__(self, transformer_classes, args, project):
        self.project = project
        self.transformers = self._init_transformers(transformer_classes, args)

    def _init_transformers(self, transformer_classes, args):
        transformers = []
        for trans, (posargs, kwargs) in zip(transformer_classes, args):
            try:
                transformers.append(
                    trans(self.project, *posargs, **kwargs))
            except (ValueError, TypeError):
                raise ConfigurationException(
                    f"Invalid arguments to {trans.name} transformer: "
                    f"{posargs}, {kwargs})",
                    project_id=self.project.project_id)
        return transformers

    def transform_text(self, text):
        for trans in self.transformers:
            text = trans.transform_text(text)
        return text

    def transform_corpus(self, corpus):
        for trans in self.transformers:
            corpus = trans.transform_corpus(corpus)
        return corpus