From 23c61ef49ef918b29952150d4a6e61b99d60ac64 Mon Sep 17 00:00:00 2001 From: Julius Unverfehrt Date: Mon, 12 Sep 2022 15:14:19 +0200 Subject: [PATCH] make env var parser discrete --- image_prediction/transformer/transformers/response.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/image_prediction/transformer/transformers/response.py b/image_prediction/transformer/transformers/response.py index 91244c9..4703411 100644 --- a/image_prediction/transformer/transformers/response.py +++ b/image_prediction/transformer/transformers/response.py @@ -4,7 +4,7 @@ import os from functools import lru_cache from operator import itemgetter -from funcy import filter, juxt, first, rest, compose +from funcy import juxt, first, rest, compose from image_prediction.config import CONFIG from image_prediction.exceptions import ParsingError @@ -131,11 +131,9 @@ def build_image_info(data: dict) -> dict: @lru_cache(maxsize=None) def parse_env_var(prefix, table=None): table = table or os.environ - head, tail = juxt(first, compose(list, rest))(filter(prefix, table)) + head = first(filter(lambda s: s == prefix, table)) if not head: logger.warning(f"Found no environment variable with prefix '{prefix}'.") - elif tail: - logger.warning(f"Found multiple candidates for environment variable with prefix '{prefix}'.") else: try: return parse_env_var_value(table[head])