martINI: martini/main.py annotate

annotate martini/main.py @ 4:f7f335561ceb

make special usage strings for each program; again, really should abstract this

author	k0s <k0scist@gmail.com>
date	Tue, 02 Mar 2010 20:27:25 -0500
parents	d172c078f4a2
children	13d85b455533

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
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	3 import config
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	4 import os
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	5 import sys
2 d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	6 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	7
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	8 sep = '--' # section separator
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 def parse_args(*args):
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 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	13 returns a tuple:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	14 ( [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	15 """
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	16
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	17 # find the files
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	18 # 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	19 index = None
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	20 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	21 if value.startswith(sep):
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	22 break
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	23 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	24 return (args, [])
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	25
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	26 files = args[:index]
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	27 args = args[index:]
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	28
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	29 # find the sections
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	30 ini = []
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	31 for arg in args:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	32 if arg.startswith(sep):
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	33 arg = arg[len(sep):]
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	34 assert arg
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	35 section = []
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	36 ini.append((arg, section))
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	37 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	38 section.append(arg)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	39
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	40 return (files, ini)
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 def parse_options(*args):
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	43 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	44 ini = {}
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	45 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	46 ini[section] = {}
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	47 for option in options:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	48 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	49 ini[section][key] = value
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	50
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	51 return (files, ini)
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 def set(args=None):
4 f7f335561ceb make special usage strings for each program; again, really should abstract this k0s <k0scist@gmail.com> parents: 2 diff changeset	54
f7f335561ceb make special usage strings for each program; again, really should abstract this k0s <k0scist@gmail.com> parents: 2 diff changeset	55 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	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:'
4 f7f335561ceb make special usage strings for each program; again, really should abstract this k0s <k0scist@gmail.com> parents: 2 diff changeset	65 print usage % os.path.basename(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
4 f7f335561ceb make special usage strings for each program; again, really should abstract this k0s <k0scist@gmail.com> parents: 2 diff changeset	85 usage = "%s file1 [file2] [...] --section1 option1 option2 --section2 option3"
f7f335561ceb make special usage strings for each program; again, really should abstract this k0s <k0scist@gmail.com> parents: 2 diff changeset	86
0 3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	87 # process arguments
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	88 if args is None:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	89 args = sys.argv[1:]
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	90 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	91
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	92 # display usage information
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	93 if not files:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	94 print 'Usage:'
4 f7f335561ceb make special usage strings for each program; again, really should abstract this k0s <k0scist@gmail.com> parents: 2 diff changeset	95 print usage % os.path.basename(sys.argv[0])
0 3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	96 sys.exit(0)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	97
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	98 # process the files
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	99 for f in files:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	100 if f == '-':
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	101 fp = sys.stdin
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	102 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	103 fp = file(f)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	104 munger = config.ConfigMunger(fp)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	105 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	106 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	107 if options:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	108 for option in options:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	109 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	110 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	111 print value
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	112 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	113 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	114
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	115 def delete(args=None):
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	116
4 f7f335561ceb make special usage strings for each program; again, really should abstract this k0s <k0scist@gmail.com> parents: 2 diff changeset	117 usage = "%s file1 [file2] [...] --section1 option1 option2 --section2 option3"
f7f335561ceb make special usage strings for each program; again, really should abstract this k0s <k0scist@gmail.com> parents: 2 diff changeset	118
0 3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	119 # process arguments
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	120 if args is None:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	121 args = sys.argv[1:]
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	122 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	123
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	124 # display usage information
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	125 if not files:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	126 print 'Usage:'
4 f7f335561ceb make special usage strings for each program; again, really should abstract this k0s <k0scist@gmail.com> parents: 2 diff changeset	127 print usage % os.path.basename(sys.argv[0])
0 3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	128 sys.exit(0)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	129
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	130 # process the files
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	131 for f in files:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	132 if f == '-':
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	133 fp = sys.stdin
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	134 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	135 fp = file(f)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	136 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	137 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	138 if section in conf:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	139 if options:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	140 for option in options:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	141 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	142 conf[section].pop(option)
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 conf.pop(section)
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	145 if f == '-':
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	146 fp = sys.stdout
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	147 else:
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	148 fp.close()
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	149 fp = file(f, 'w')
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	150 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	151
1 5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	152 def munge(args=None):
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	153
4 f7f335561ceb make special usage strings for each program; again, really should abstract this k0s <k0scist@gmail.com> parents: 2 diff changeset	154 usage = "%s file1 [file2] [...]"
f7f335561ceb make special usage strings for each program; again, really should abstract this k0s <k0scist@gmail.com> parents: 2 diff changeset	155
1 5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	156 # process arguments
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	157 if args is None:
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	158 args = sys.argv[1:]
2 d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	159
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	160 parser = OptionParser()
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	161 parser.add_option('-O', dest='output',
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	162 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	163 options, args = parser.parse_args(args)
1 5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	164 files, sections = parse_options(*args)
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 # display usage information
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	167 if not files:
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	168 print 'Usage:'
4 f7f335561ceb make special usage strings for each program; again, really should abstract this k0s <k0scist@gmail.com> parents: 2 diff changeset	169 print usage % os.path.basename(sys.argv[0])
1 5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	170 sys.exit(0)
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	171
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	172 # munge the files
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	173 conf = list(files)
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	174 conf.append(sections)
5627074cd79b * adding ini-munge command k0s <k0scist@gmail.com> parents: 0 diff changeset	175 munger = config.ConfigMunger(*conf)
2 d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	176 if options.output:
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	177 fp = file(options.output, 'w')
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	178 else:
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	179 fp = sys.stdout
d172c078f4a2 add -O option for specifying output file k0s <k0scist@gmail.com> parents: 1 diff changeset	180 munger.write(fp)
0 3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	181
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	182 if __name__ == '__main__':
3c3522ce6e3a initial import of martINI from https://svn.openplans.org/svn/standalone/martINI/ k0s <k0scist@gmail.com> parents: diff changeset	183 set(sys.argv[1:])

Mercurial > hg > martINI

annotate martini/main.py @ 4:f7f335561ceb