#!/usr/bin/python3

import yaml
import sys

values_file = sys.argv[1]
service_name = sys.argv[2]
new_version = sys.argv[3]

with open(values_file) as file:
    ymldoc = yaml.safe_load(file)
    ymldoc[service_name]['image']['tag'] = new_version
    ymldoc[service_name]['helm']['version'] = new_version

with open(values_file, "w") as ostream:
    yaml.dump(ymldoc, ostream, default_flow_style=False, sort_keys=False)