fba/helpers/config.py

   1 # Fedi API Block - An aggregator for fetching blocking data from fediverse nodes
   2 # Copyright (C) 2023 Free Software Foundation
   3 #
   4 # This program is free software: you can redistribute it and/or modify
   5 # it under the terms of the GNU Affero General Public License as published
   6 # by the Free Software Foundation, either version 3 of the License, or
   7 # (at your option) any later version.
   8 #
   9 # This program is distributed in the hope that it will be useful,
  10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 # GNU Affero General Public License for more details.
  13 #
  14 # You should have received a copy of the GNU Affero General Public License
  15 # along with this program.  If not, see <https://www.gnu.org/licenses/>.
  16
  17 import logging
  18 import json
  19 import sys
  20
  21 logging.basicConfig(level=logging.INFO)
  22 logger = logging.getLogger(__name__)
  23
  24 with open("config.json") as f:
  25     logger.debug("Loading configuration file ...")
  26     _config = json.loads(f.read())
  27     _config["max_crawl_depth"] = min(_config["max_crawl_depth"], (sys.getrecursionlimit() - 50))
  28
  29 def get(key: str) -> any:
  30     logger.debug("key[%s]='%s' - CALLED!", type(key), key)
  31
  32     if not isinstance(key, str):
  33         raise ValueError(f"Parameter key[]='{type(key)}' is not of type 'str'")
  34     elif key == "":
  35         raise ValueError("Parameter 'key' is empty")
  36     elif not key in _config:
  37         raise KeyError(f"key='{key}' does not exist in _config array")
  38
  39     logger.debug("_config[%s][%s]='%s' - EXIT!", key, type(_config[key]), _config[key] if not key.endswith("_api_key") else "***")
  40     return _config[key]