From 62eb66b8cab945c8d8eb8d59f39c3fb2c0588e48 Mon Sep 17 00:00:00 2001 From: Debanjum Singh Solanky Date: Wed, 10 Aug 2022 22:28:51 +0300 Subject: [PATCH] Rename load_config_from_file to more descriptive parse_config_from_file --- src/utils/cli.py | 4 ++-- src/utils/yaml.py | 21 ++++++++++++++++++--- 2 files changed, 20 insertions(+), 5 deletions(-) diff --git a/src/utils/cli.py b/src/utils/cli.py index b24cc1bf..a77b35a7 100644 --- a/src/utils/cli.py +++ b/src/utils/cli.py @@ -4,7 +4,7 @@ import pathlib # Internal Packages from src.utils.helpers import resolve_absolute_path -from src.utils.yaml import load_config_from_file +from src.utils.yaml import parse_config_from_file def cli(args=None): @@ -22,6 +22,6 @@ def cli(args=None): if not resolve_absolute_path(args.config_file).exists(): args.config = None else: - args.config = load_config_from_file(args.config_file) + args.config = parse_config_from_file(args.config_file) return args \ No newline at end of file diff --git a/src/utils/yaml.py b/src/utils/yaml.py index 223f99fd..c7a7fd99 100644 --- a/src/utils/yaml.py +++ b/src/utils/yaml.py @@ -5,11 +5,26 @@ import yaml from src.utils.helpers import get_absolute_path from src.utils.rawconfig import FullConfig + +def save_config_to_file(yaml_config, yaml_config_file): + "Write config to YML file" + with open(get_absolute_path(yaml_config_file), 'w', encoding='utf-8') as config_file: + yaml.dump(yaml_config, config_file, allow_unicode=True) + + def load_config_from_file(yaml_config_file): - # Read Config from YML file + "Read config from YML file" config_from_file = None with open(get_absolute_path(yaml_config_file), 'r', encoding='utf-8') as config_file: config_from_file = yaml.safe_load(config_file) + return config_from_file - # Parse, Validate Config in YML file - return FullConfig.parse_obj(config_from_file) + +def parse_config_from_string(yaml_config): + "Parse and validate config in YML string" + return FullConfig.parse_obj(yaml_config) + + +def parse_config_from_file(yaml_config_file): + "Parse and validate config in YML file" + return parse_config_from_string(load_config_from_file(yaml_config_file))