def testReplaceMulti(self): tmp_dict = deepcopy(self.testDict) DictProcessing.replace(self.testDict, "a", "some") self.assertEqual(self.testDict["a"], "some") self.assertEqual(self.testDict["dict"]["a"], "some") self.assertEqual(self.testDict["list"][2]["a"], "some") self.assertNotEqual(tmp_dict, self.testDict)
def purge(replaces, files, dry_run, simplify): for one_file in files: click.echo(f"Processing file: {one_file.name}") object_representation = yaml.safe_load(one_file) processor = DictProcessing(object_representation) for item in replaces: click.echo(f"\tTry to apply: {item}") selector_str, key, type_of_value, value = item.split(":", 3) selector_list = [] if not selector_str else selector_str.split("%") # retype the output object to proper type () value = getattr(builtins, type_of_value)(value) for matched in processor.match(selector=selector_list): click.echo(f"\t\tMatched {selector_list}") processor.replace(obj=matched, key=key, value=value) if simplify: processor.simplify() if not dry_run: click.echo(f"Writing content back to file: {one_file.name}") with open(one_file.name, mode="w") as outfile: outfile.write(yaml.safe_dump(object_representation))
def testReplaceNone(self): tmp_dict = deepcopy(self.testDict) DictProcessing.replace(self.testDict, "nonsense", "some") self.assertEqual(tmp_dict, self.testDict)