Rewrite the parameters to be readable by argpare

output

['-o', 'hello world']
['-o', 'hello world']
False
None
hello world
['-o', 'hello world', '-p:page']
['-o', 'hello world', '-p=page']
False
page
hello world
['-o', 'hello world', '-p:page', '--keep']
['-o', 'hello world', '-p=page', '--keep']
True
page
hello world
['--keep', '-o', 'hello world']
['--keep', '-o', 'hello world']
True
None
hello world
['--keep', '-o:hello', '-p', 'world']
['--keep', '-o=hello', '-p', 'world']
True
world
hello
['--keep', '-o:hello world', '-p', 'world']
['--keep', '-o=hello world', '-p', 'world']
True
world
hello world
['--keep', '-o:hello', '-p', 'scyscraper world']
['--keep', '-o=hello', '-p', 'scyscraper world']
True
scyscraper world
hello
['--keep', '-o', 'Wikipedia:Help', '-p', 'scyscraper world']
['--keep', '-o', 'Wikipedia:Help', '-p', 'scyscraper world']
True
scyscraper world
Wikipedia:Help

pt.py

#!/usr/bin/python
# -*- coding: utf-8  -*-
import argparse
import shlex

valid_warnings = ['foo', 'bar']

parser = argparse.ArgumentParser()
parser.add_argument('--keep', action='store_true')
parser.add_argument('-p', '--page', '-page')
parser.add_argument('-o', '--other', '-other')

def test(a):
    try:
        r = parser.parse_args(a)
        print(r.keep)
        print(r.page)
        print(r.other)
    except SystemError as e:
        print(e.message)

def rewrite(args):
    if isinstance(args, basestring):
        args = shlex.split(args)
    new_args = []
    for arg in args:
        if arg[0] == '-':
            separator = len(arg)
            if ' ' in arg:
                separator = arg.index(' ')
            if '=' in arg:
                separator = min(arg.index('='), separator)
            if ':' in arg:
                pos = arg.index(':')
                if separator > pos:
                    arg = arg[:pos] + '=' + arg[pos+1:]
        new_args += [arg]
    print args
    print new_args
    test(new_args)

rewrite(['-o', 'hello world'])
rewrite(['-o', 'hello world', '-p:page'])
rewrite(['-o', 'hello world', '-p:page', '--keep'])
rewrite('--keep -o "hello world"')
rewrite('--keep -o:hello -p world')
rewrite('--keep -o:"hello world" -p world')
rewrite('--keep -o:hello -p "scyscraper world"')
rewrite('--keep -o "Wikipedia:Help" -p "scyscraper world"')