import json
from pathlib import Path

CATALOG_PATH = Path(__file__).parent / "catalog.json"


class SemanticCatalog:

    def __init__(self):
        with open(CATALOG_PATH, "r") as f:
            self.catalog = json.load(f)

    def get_entity(self, entity_name):
        return self.catalog["entities"].get(entity_name)

    def get_all_entities(self):
        return list(
            self.catalog["entities"].keys()
        )


semantic_catalog = SemanticCatalog()