annif.corpus.combine

Complexity

Total Complexity

2

Size/Duplication

Total Lines	25
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	CombinedCorpus.documents()	0	4	1
A	CombinedCorpus.__init__()	0	2	1

"""Class for combining multiple corpora so they behave like a single corpus"""

from __future__ import annotations

import itertools
from typing import TYPE_CHECKING

from .types import DocumentCorpus

if TYPE_CHECKING:
    from annif.corpus.document import DocumentFile


class CombinedCorpus(DocumentCorpus):
    """Class for combining multiple corpora so they behave like a single
    corpus"""

    def __init__(self, corpora: list[DocumentFile]) -> None:
        self._corpora = corpora

    @property
    def documents(self) -> itertools.chain:
        return itertools.chain.from_iterable(
            [corpus.documents for corpus in self._corpora]
        )