From fc1fafad4c43011f4d60f26c7695879a5b11aa5e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Krystian=20Du=C5=BCy=C5=84ski?= Date: Fri, 21 Apr 2023 22:35:21 +0200 Subject: [PATCH] fixed yaml warning --- mikrotik_configurator/__main__.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mikrotik_configurator/__main__.py b/mikrotik_configurator/__main__.py index 1528f7f..4c00eef 100644 --- a/mikrotik_configurator/__main__.py +++ b/mikrotik_configurator/__main__.py @@ -20,7 +20,10 @@ def main(): args = argparser.parse_args() 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"] has_flash = cfg.get("has_flash", False)