klbr/khoj
1
0
Fork 0
mirror of https://github.com/khoaliber/khoj.git synced 2026-03-02 21:19:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Extract code to load config from YAML file into new utils.yaml module

Browse Source
This commit is contained in:
Debanjum Singh Solanky
2022-08-10 20:07:00 +03:00
parent daef276fd1
commit d2c7b28172
2 changed files with 18 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/utils/cli.py
View File

@@ -2,12 +2,9 @@
import argparse
import pathlib
# External Packages
import yaml
# Internal Packages
from src.utils.helpers import get_absolute_path, resolve_absolute_path
from src.utils.rawconfig import FullConfig
from src.utils.helpers import resolve_absolute_path
from src.utils.yaml import load_config_from_file
def cli(args=None):
@@ -25,12 +22,6 @@ def cli(args=None):
if not resolve_absolute_path(args.config_file).exists():
args.config = None
else:
# Read Config from YML file
config_from_file = None
with open(get_absolute_path(args.config_file), 'r', encoding='utf-8') as config_file:
config_from_file = yaml.safe_load(config_file)
# Parse, Validate Config in YML file
args.config = FullConfig.parse_obj(config_from_file)
args.config = load_config_from_file(args.config_file)
return args

15
src/utils/yaml.py Normal file
View File

@@ -0,0 +1,15 @@
# External Packages
import yaml
# Internal Packages
from src.utils.helpers import get_absolute_path
from src.utils.rawconfig import FullConfig
def load_config_from_file(yaml_config_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)
# Parse, Validate Config in YML file
return FullConfig.parse_obj(config_from_file)