CodeMirror/bulk-downloader-for-reddit
1
0
Fork 0
mirror of synced 2024-05-20 12:12:40 +12:00
Code Issues Releases Wiki Activity

Extract YAML function

Browse source
This commit is contained in:
Serene-Arc 2022-07-22 15:44:19 +10:00
parent 5f443fddff
commit cb3415c62f
1 changed files with 19 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
bdfr/configuration.py
View file

@ -2,6 +2,7 @@
# coding=utf-8
from argparse import Namespace
from pathlib import Path
from typing import Optional
import logging
@ -54,13 +55,7 @@ class Configuration(Namespace):
def process_click_arguments(self, context: click.Context):
if context.params.get('opts') is not None:
with open(context.params['opts']) as f:
opts = yaml.load(f, Loader=yaml.FullLoader)
for arg_key, val in opts.items():
if not hasattr(self, arg_key):
logger.error(f'Ignoring an unknown YAML argument: {arg_key}')
continue
setattr(self, arg_key, val)
self.parse_yaml_options(context.params['opts'])
for arg_key in context.params.keys():
if not hasattr(self, arg_key):
logger.warning(f'Ignoring an unknown CLI argument: {arg_key}')
@ -70,3 +65,20 @@ class Configuration(Namespace):
# don't overwrite with an empty value
continue
setattr(self, arg_key, val)
def parse_yaml_options(self, file_path: str):
yaml_file_loc = Path(file_path)
if not yaml_file_loc.exists():
logger.error(f'No YAML file found at {yaml_file_loc}')
return
with open(yaml_file_loc) as f:
try:
opts = yaml.load(f, Loader=yaml.FullLoader)
except yaml.YAMLError as e:
logger.error(f'Could not parse YAML options file: {e}')
return
for arg_key, val in opts.items():
if not hasattr(self, arg_key):
logger.error(f'Ignoring an unknown YAML argument: {arg_key}')
continue
setattr(self, arg_key, val)