#!/usr/bin/python
#
# Copyright (C) 2012, Red Hat, Inc.
# Alan Pevec <apevec@redhat.com>
#
# Licensed 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.
#

import iniparse
import sys

parameter = value = None

def usage():
  sys.stderr.write(sys.argv[0]+
                     " --set|--del config_file section [parameter] [value]\n")
  sys.exit(1)

try:
  mode=sys.argv[1]
  if mode not in ('--set', '--del'):
      usage()
  cfgfile=sys.argv[2]
  section=sys.argv[3]
  try:
    parameter=sys.argv[4]
    value=sys.argv[5]
  except IndexError:
     if mode == '--set':
         usage()
except IndexError:
  usage()

conf=iniparse.ConfigParser()
conf.read(cfgfile)

if mode == '--set':
    if not conf.has_section(section):
        conf.add_section(section)
        value += '\n'
    conf.set(section, parameter, value)
else:
    if parameter is None:
        conf.remove_section(section)
    elif value is None:
        conf.remove_option(section, parameter)

with open(cfgfile, 'w') as f:
    conf.write(f)
