Update hieraedit for python3

Change-Id: Ibd8991eb466416f77a2decc7b0a280d8e4124942
This commit is contained in:
James E. Blair 2019-03-26 15:32:23 -07:00
parent 2ccf1ddb7a
commit 2db41fc488

View File

@ -56,7 +56,7 @@ parser.add_argument('value', help='the value', nargs='?')
parser.add_argument('-f', dest='file', help='file to read in as value') parser.add_argument('-f', dest='file', help='file to read in as value')
args = parser.parse_args() args = parser.parse_args()
data = yaml.load(open(args.yaml)) data = yaml.safe_load(open(args.yaml))
changed = False changed = False
if args.value: if args.value:
@ -65,12 +65,12 @@ if args.value:
if args.file: if args.file:
data[args.key] = open(args.file).read() data[args.key] = open(args.file).read()
changed = True changed = True
print data[args.key] print(data[args.key])
if changed: if changed:
dn = os.path.dirname(args.yaml) dn = os.path.dirname(args.yaml)
(out, fn) = tempfile.mkstemp(dir=dn) (out, fn) = tempfile.mkstemp(dir=dn)
os.write(out, yaml.dump(data, default_flow_style=False, Dumper=MyDumper)) os.write(out, yaml.dump(data, default_flow_style=False, Dumper=MyDumper).encode('utf8'))
os.close(out) os.close(out)
os.chown(fn, pwd.getpwnam('puppet').pw_uid, grp.getgrnam('puppet').gr_gid) os.chown(fn, pwd.getpwnam('root').pw_uid, grp.getgrnam('admin').gr_gid)
os.rename(fn, args.yaml) os.rename(fn, args.yaml)