Source code for dimcat.data.catalogs.inputs

from __future__ import annotations

import logging

from dimcat.data.catalogs.base import DimcatCatalog
from dimcat.data.resources.base import F
from dimcat.data.resources.dc import FeatureSpecs
from dimcat.data.resources.features import Metadata
from dimcat.data.resources.utils import feature_specs2config

module_logger = logging.getLogger(__name__)


class InputsCatalog(DimcatCatalog):
    def extract_feature(self, feature: FeatureSpecs) -> F:
        """Extracts the given features from all packages and combines them in a Feature resource."""
        package = self.get_package()
        return package.extract_feature(feature)

    def get_feature(self, feature: FeatureSpecs) -> F:
        """ToDo: Get features from all packages and merge them."""
        package = self.get_package()
        feature_config = feature_specs2config(feature)
        return package.get_feature(feature_config)

    def get_metadata(self) -> Metadata:
        """Returns a dataframe with all metadata."""
        package = self.get_package()
        return package.get_metadata()