Skip to content
Snippets Groups Projects
main.py 1.28 KiB
Newer Older
Bartłomiej Koptyra's avatar
Bartłomiej Koptyra committed
"""Implementation of anonymizer service."""
import nlp_ws
Michał Pogoda's avatar
Michał Pogoda committed
from src.worker_old import Worker
import logging
import nlp_ws
Bartłomiej Koptyra's avatar
Bartłomiej Koptyra committed

_log = logging.getLogger(__name__)
Bartłomiej Koptyra's avatar
Bartłomiej Koptyra committed


class AnonymizerWorker(nlp_ws.NLPWorker):
    """Class implementing TextFixerWorker worker."""
Bartłomiej Koptyra's avatar
Bartłomiej Koptyra committed

    @classmethod
    def static_init(cls, config):
        """Initialize process."""
        cls._configuration = config.get("tool").get("configuration", "ccl")
        cls._default_language = config.get("tool").get("default_language", "pl")
        cls._default_replacer = config.get("tool").get("default_replacer", "tag")
Bartłomiej Koptyra's avatar
Bartłomiej Koptyra committed

        _log.info(
            "AnonymizerWorker initialized with configuration: %s, default language: %s, default replacer: %s",
            cls._configuration,
            cls._default_language,
            cls._default_replacer,
        )
Bartłomiej Koptyra's avatar
Bartłomiej Koptyra committed

    def __init__(self):
        self._worker = Worker(
            configuration=self._configuration,
            default_language=self._default_language,
            default_replacer=self._default_replacer,
        )
Bartłomiej Koptyra's avatar
Bartłomiej Koptyra committed

    def process(self, input_file, task_options, output_file):
        self._worker.process(input_file, task_options, output_file)
Bartłomiej Koptyra's avatar
Bartłomiej Koptyra committed


if __name__ == "__main__":
    nlp_ws.NLPService.main(AnonymizerWorker, pause_at_exit=False)