perftests/etc/json_config_tool.py

#!/usr/bin/env python
#
# Licensed to the Apache Software Foundation (ASF) under one
# or more contributor license agreements.  See the NOTICE file
# distributed with this work for additional information
# regarding copyright ownership.  The ASF licenses this file
# to you under the Apache License, Version 2.0 (the
# "License"); you may not use this file except in compliance
# with the License.  You may obtain a copy of the License at
#
#   http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
# KIND, either express or implied.  See the License for the
# specific language governing permissions and limitations
# under the License.
#

from StringIO import StringIO
import json
import sys
import re

objname = None
attrname = None
attrvalue = None

try:
  import argparse

  parser = argparse.ArgumentParser(description='Adds (or updates) a attribute name/value pair of an existing object within an Apache Qpid Broker-J config.json')
  parser.add_argument("objectname", help='Name of the object e.g. httpManagement')
  parser.add_argument("attrname", help='Name of the attribute to add or update e.g. httpBasicAuthenticationEnabled')
  parser.add_argument("attrvalue", help='Value of the attribute e.g. true')
  args = parser.parse_args()

  objname = args.objectname
  attrname = args.attrname
  attrvalue = args.attrvalue
except ImportError:
  objname = sys.argv[1]
  attrname = sys.argv[2]
  attrvalue = sys.argv[3]
  pass


def transform(data):
  if isinstance(data, tuple):
    (k, v) = data;
    return k, transform(v)
  elif isinstance(data, list):
    return [transform(i) for i in data]
  elif isinstance(data, dict):
    for k, v in data.items():
      data[k] = transform(v)
    if "name" in data and data["name"] == objname:
      data[attrname] = attrvalue
    return data
  else:
    return data

# Expects a config.json to be provided on stdin, write the resulting json to stdout.

lines = []
for line in sys.stdin:
  lines.append(line.rstrip())

# naive strip C style comments - this deals with the Apache licence comment present on the default config
input = re.sub("/\*.*?\*/", " ", "".join(lines), re.S)

data = json.load(StringIO(input))
data = dict([transform((k, v)) for k,v in data.items()])

json.dump(data, sys.stdout, sort_keys=True, indent=2)