Adding support to conditionally replace a value

Bug: 231691643
Test: presubmit
Change-Id: I786ab152a94126ebc8a9d7dea0bb68e07d789a0b
This commit is contained in:
Alexei Nicoara
2022-07-11 12:38:50 +01:00
parent 2971a94e4c
commit 7d69b1d8e6
3 changed files with 97 additions and 0 deletions

View File

@@ -59,6 +59,13 @@ class Replace(str):
cur[key] = val
class ReplaceIfEqual(str):
def apply(self, obj, old_val, new_val):
cur, key = follow_path(obj, self)
if cur and cur[key] == int(old_val):
cur[key] = new_val
class Remove(str):
def apply(self, obj):
cur, key = follow_path(obj, self)
@@ -91,6 +98,11 @@ def main():
help='replace value of the key specified by path. If path doesn\'t exist, no op.',
metavar=('path', 'value'),
nargs=2, dest='patch', action='append')
parser.add_argument("-se", "--replace-if-equal", type=ReplaceIfEqual,
help='replace value of the key specified by path to new_value if it\'s equal to old_value.' +
'If path doesn\'t exist or the value is not equal to old_value, no op.',
metavar=('path', 'old_value', 'new_value'),
nargs=3, dest='patch', action='append')
parser.add_argument("-r", "--remove", type=Remove,
help='remove the key specified by path. If path doesn\'t exist, no op.',
metavar='path',