Fix crash when using cfg on empty/new configs

Fixes #19.
This commit is contained in:
rlaphoenix 2023-02-07 20:44:34 +00:00
parent f7683173f8
commit 1fa3ba61c8
1 changed files with 4 additions and 0 deletions

View File

@ -45,6 +45,10 @@ def cfg(ctx: click.Context, key: str, value: str, unset: bool, list_: bool) -> N
if not data:
log.warning(f"{config_path} has no configuration data, yet")
# yaml.load() returns `None` if the input data is blank instead of a usable object
# force a usable object by making one and removing the only item within it
data = yaml.load("""__TEMP__: null""")
del data["__TEMP__"]
if list_:
yaml.dump(data, sys.stdout)