martINI: martini/main.py annotate

annotate martini/main.py @ 2:d172c078f4a2

add -O option for specifying output file

author	k0s <k0scist@gmail.com>
date	Tue, 08 Dec 2009 16:23:50 -0500
parents	5627074cd79b
children	f7f335561ceb

rev	line source
0 3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	1 #!/usr/bin/env python
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	2
1 5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	3 usage = "%s file1 [file2] [...] --section1 option1=value1 option2=value2 --section2 option3=value3"
0 3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	4
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	5 import config
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	6 import os
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	7 import sys
2 d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	8 from optparse import OptionParser
0 3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	9
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	10 sep = '--' # section separator
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	11
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	12 def parse_args(*args):
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	13 """
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	14 convert command line options to files, sections, and options
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	15 returns a tuple:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	16 ( [files], { 'section': { 'option': 'value', 'option2': 'value2' } } )
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	17 """
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	18
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	19 # find the files
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	20 # XXX note this prohibits files starting with --
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	21 index = None
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	22 for index, value in enumerate(args):
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	23 if value.startswith(sep):
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	24 break
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	25 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	26 return (args, [])
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	27
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	28 files = args[:index]
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	29 args = args[index:]
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	30
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	31 # find the sections
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	32 ini = []
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	33 for arg in args:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	34 if arg.startswith(sep):
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	35 arg = arg[len(sep):]
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	36 assert arg
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	37 section = []
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	38 ini.append((arg, section))
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	39 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	40 section.append(arg)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	41
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	42 return (files, ini)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	43
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	44 def parse_options(*args):
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	45 files, sections = parse_args(*args)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	46 ini = {}
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	47 for section, options in sections:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	48 ini[section] = {}
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	49 for option in options:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	50 key, value = option.split('=', 1)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	51 ini[section][key] = value
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	52
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	53 return (files, ini)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	54
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	55 def set(args=None):
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	56
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	57 # process arguments
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	58 if args is None:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	59 args = sys.argv[1:]
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	60 files, sections = parse_options(*args)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	61
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	62 # display usage information
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	63 if not files:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	64 print 'Usage:'
1 5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	65 print usage % sys.argv[0]
0 3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	66 sys.exit(0)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	67
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	68 # process the files
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	69 for f in files:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	70 if f == '-':
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	71 fp = sys.stdin
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	72 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	73 fp = file(f)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	74 munger = config.ConfigMunger(fp, sections)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	75
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	76 if f == '-':
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	77 fp = sys.stdout
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	78 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	79 fp.close()
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	80 fp = file(f, "w")
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	81 munger.write(fp=fp)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	82
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	83 def get(args=None):
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	84
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	85 # process arguments
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	86 if args is None:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	87 args = sys.argv[1:]
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	88 files, sections = parse_args(*args)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	89
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	90 # display usage information
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	91 if not files:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	92 print 'Usage:'
1 5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	93 print usage % sys.argv[0]
0 3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	94 sys.exit(0)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	95
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	96 # process the files
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	97 for f in files:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	98 if f == '-':
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	99 fp = sys.stdin
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	100 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	101 fp = file(f)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	102 munger = config.ConfigMunger(fp)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	103 for section, options in sections:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	104 if section in munger.sections():
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	105 if options:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	106 for option in options:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	107 value = munger.get(section, option)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	108 if value is not None:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	109 print value
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	110 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	111 config.ConfigMunger({section: munger[section]}).write()
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	112
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	113 def delete(args=None):
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	114
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	115 # process arguments
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	116 if args is None:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	117 args = sys.argv[1:]
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	118 files, sections = parse_args(*args)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	119
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	120 # display usage information
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	121 if not files:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	122 print 'Usage:'
1 5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	123 print usage % sys.argv[0]
0 3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	124 sys.exit(0)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	125
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	126 # process the files
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	127 for f in files:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	128 if f == '-':
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	129 fp = sys.stdin
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	130 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	131 fp = file(f)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	132 conf = config.ConfigMunger(fp).dict()
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	133 for section, options in sections:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	134 if section in conf:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	135 if options:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	136 for option in options:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	137 if option in conf[section]:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	138 conf[section].pop(option)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	139 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	140 conf.pop(section)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	141 if f == '-':
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	142 fp = sys.stdout
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	143 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	144 fp.close()
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	145 fp = file(f, 'w')
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	146 config.ConfigMunger(conf).write(fp)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	147
1 5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	148 def munge(args=None):
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	149
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	150 # process arguments
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	151 if args is None:
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	152 args = sys.argv[1:]
2 d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	153
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	154 parser = OptionParser()
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	155 parser.add_option('-O', dest='output',
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	156 help="name of file to output (stdout if not specified)")
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	157 options, args = parser.parse_args(args)
1 5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	158 files, sections = parse_options(*args)
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	159
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	160 # display usage information
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	161 if not files:
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	162 print 'Usage:'
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	163 print usage % sys.argv[0]
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	164 sys.exit(0)
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	165
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	166 # munge the files
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	167 conf = list(files)
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	168 conf.append(sections)
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	169 munger = config.ConfigMunger(*conf)
2 d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	170 if options.output:
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	171 fp = file(options.output, 'w')
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	172 else:
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	173 fp = sys.stdout
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	174 munger.write(fp)
0 3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	175
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	176 if __name__ == '__main__':
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	177 set(sys.argv[1:])

Mercurial > hg > martINI

annotate martini/main.py @ 2:d172c078f4a2