fixed yaml warning

This commit is contained in:
Krystian Dużyński 2023-04-21 22:35:21 +02:00
parent fee79e60e2
commit fc1fafad4c

View File

@ -20,7 +20,10 @@ def main():
args = argparser.parse_args() args = argparser.parse_args()
dry_run = args.dry_run dry_run = args.dry_run
cfg = yaml.load(open(args.config, "rt"), Loader=yaml.FullLoader) if hasattr(yaml, "SafeLoader"):
cfg = yaml.load(open(args.config, "rt"), Loader=yaml.SafeLoader)
else:
cfg = yaml.load(open(args.config, "rt"))
host = cfg["host"] host = cfg["host"]
has_flash = cfg.get("has_flash", False) has_flash = cfg.get("has_flash", False)