From a92d328874408288fefc33c5a849a1f83c68cc38 Mon Sep 17 00:00:00 2001 From: "Joshua M. Boniface" Date: Sun, 27 Aug 2023 11:35:21 -0400 Subject: [PATCH] Fix missing YAML load --- client-cli/pvc/cli/cli.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/client-cli/pvc/cli/cli.py b/client-cli/pvc/cli/cli.py index 1e1d9aa4..298c3832 100644 --- a/client-cli/pvc/cli/cli.py +++ b/client-cli/pvc/cli/cli.py @@ -27,6 +27,7 @@ from os import environ, makedirs, path from pkg_resources import get_distribution from lxml.etree import fromstring, tostring from re import sub +from yaml import load as yload from pvc.cli.helpers import * from pvc.cli.waiters import * @@ -4510,7 +4511,7 @@ def cli_provisioner_userdata_add(name, filename): userdata = filename.read() filename.close() try: - yaml.load(userdata, Loader=yaml.SafeLoader) + yload(userdata, Loader=yaml.SafeLoader) except Exception as e: echo("Error: Userdata document is malformed") cleanup(False, e) @@ -4599,7 +4600,7 @@ def cli_provisioner_userdata_modify(name, filename, editor): filename.close() try: - yaml.load(userdata, Loader=yaml.SafeLoader) + yload(userdata, Loader=yaml.SafeLoader) except Exception as e: echo("Error: Userdata document is malformed") cleanup(False, e)