开发者

How to extend this search-and-replace python script to accept variables from command line?

开发者 https://www.devze.com 2023-03-22 01:51 出处:网络
Currently, I have a script which does the following. If I have text file with the lines: <<Name>> is at <<Location>>.

Currently, I have a script which does the following. If I have text file with the lines:

<<Name>> is at <<Location>>.
<<Name>> is feeling <<Emotion>>.

The script will take in the this input file as a command line argument and prompt the user for the variables:

Name? Bob
Location? work
Emotion? frustrated

Note that name is only asked once. The script also takes in an output file as an argument and will place in the file the following.

Bob is at work.
Bob is feeling frustrated.

Now I am trying to extend the script so that I can input the variables from the command line (as if I already knew what it was going to ask). So the command would be like (in this case):

python script.py infile outfile Bob work frustrated

And it would generate the same output. Ideally, the extension should prompt the user for 开发者_如何学Cremaining variables if there are more remaining after those put in the command line. So if I run the script as:

python script.py infile outfile Bob work

The script would still prompt:

Emotion?

Excess variables in the command line would be ignored. I am pretty new to Python, so while this seems pretty simple, I haven't had success updating my current script with this add-on. Attached is the script:

import argparse
from os import path
import re

replacements = {}
pattern = '<<([^>]*)>>'

def user_replace(match):
    ## Pull from replacements dict or prompt
    placeholder = match.group(1)
    if placeholder in replacements:
        return replacements[placeholder]
    ## .setdefault(key, value) returns the value if present, else sets it then returns
    return replacements.setdefault(placeholder, raw_input('%s? ' % placeholder))

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('infile', type=argparse.FileType('r'))
    parser.add_argument('outfile', type=argparse.FileType('w'))
    args = parser.parse_args()

    matcher = re.compile(pattern)

    for line in args.infile:
        new_line = matcher.sub(user_replace, line)
        args.outfile.write(new_line)

    args.infile.close()
    args.outfile.close()

if __name__ == '__main__':
    main()

Edit: The input file example above is arbitrary. In actuality, the input file could have any number of variables, repeated any number of times.


OK, if you want to dynamically generate the options:

parser = argparse.ArgumentParser()
parser.add_argument('infile', type=argparse.FileType('r'))
parser.add_argument('outfile', type=argparse.FileType('w'))

required, extra = parser.parse_known_args()
infile, outfile = required.infile, required.outfile

args = re.findall(pattern, infile.read())
infile.seek(0)

parser = argparse.ArgumentParser()
for arg in args:
    parser.add_argument('--' + arg.lower())

replacements = vars(parser.parse_args(extra))

This will give you a dictionary of all the arguments. Each argument value will be a string inside a list. Then you just do

def user_replace(match):
    """Pull from replacements dict or prompt"""
    placeholder = match.group(1)
    return (replacements[placeholder][0] 
             if placeholder in replacements else 
              raw_input('%s? ' % placeholder))

Edit: I've edited the code at the top to set the arguments. This way, the arguments are optional, and you can have any number. Their names will still be name, location, and emotion in replacements.

The user can now do:

python script.py infile outfile --name Bob --emotion 'extremely frustrated'

leaving out the ones they want to be prompted for, and enclosing strings with spaces in quotes.

Edit 2: Edited the top part so it dynamically gets the args from the textfile.

0

精彩评论

暂无评论...
验证码 换一张
取 消

关注公众号