Source code for cybsi.api.enrichment.api

from ..internal import BaseAPI, BaseAsyncAPI
from .analyzers import AnalyzersAPI
from .config_rules import ConfigRulesAPI
from .external_dbs import ExternalDBsAPI
from .task_queue import TaskQueueAPI, TaskQueueAsyncAPI
from .tasks import TasksAPI, TasksAsyncAPI




class EnrichmentAPI(BaseAPI):
    """Enrichment API."""

    @property
    def config_rules(self) -> "ConfigRulesAPI":
        """Get config rules route."""
        return ConfigRulesAPI(self._connector)

    @property
    def tasks(self) -> "TasksAPI":
        """Get enrichment tasks route."""
        return TasksAPI(self._connector)

    @property
    def task_queue(self) -> "TaskQueueAPI":
        """Get task queue route."""
        return TaskQueueAPI(self._connector)

    @property
    def external_dbs(self) -> "ExternalDBsAPI":
        """Get external databases route."""
        return ExternalDBsAPI(self._connector)

    @property
    def analyzers(self) -> "AnalyzersAPI":
        """Get analyzers route."""
        return AnalyzersAPI(self._connector)





class EnrichmentAsyncAPI(BaseAsyncAPI):
    """Enrichment asynchronous API."""

    @property
    def tasks(self) -> "TasksAsyncAPI":
        """Get enrichment tasks route."""
        return TasksAsyncAPI(self._connector)

    @property
    def task_queue(self) -> "TaskQueueAsyncAPI":
        """Get task queue route."""
        return TaskQueueAsyncAPI(self._connector)