Support !import and !include in awx import -f yaml command

This commit is contained in:
Aggelos Kolaitis
2020-09-13 17:01:12 +03:00
parent b69fad83b1
commit 9918b2581c

View File

@@ -2,7 +2,7 @@ import yaml
import json import json
import os import os
from awxkit import api, config from awxkit import api, config, yaml_file
from awxkit.exceptions import ImportExportError from awxkit.exceptions import ImportExportError
from awxkit.utils import to_str from awxkit.utils import to_str
from awxkit.api.pages import Page from awxkit.api.pages import Page
@@ -140,7 +140,7 @@ class Import(CustomCommand):
if fmt == 'json': if fmt == 'json':
data = json.load(client.stdin) data = json.load(client.stdin)
elif fmt == 'yaml': elif fmt == 'yaml':
data = yaml.safe_load(client.stdin) data = yaml.load(client.stdin, Loader=yaml_file.Loader)
else: else:
raise ImportExportError("Unsupported format for Import: " + fmt) raise ImportExportError("Unsupported format for Import: " + fmt)