importing a python script from another script and running it with arguments

Question

I have a python script which has been packaged up as a command line script (dbtoyaml.py in Pyrseas since you ask).

I am running another python script from which I want to call this script. Is there no way to import the module and artificially populate the required arguments from my second script to avoid changing any of the pyrseas code at all?

from pyrseas import dbtoyaml -- My initial script, which also takes arguments dbtoyaml.main(['-m','-H MYHOSTNAME' .... other options])

Hasn't yet worked for me.

I get a strange error:

usage: checkSchemaChanges.py [-h] [-H HOST] [-p PORT] [-U USERNAME] [-W]
                         [-c CONFIG] [-r REPOSITORY] [-o OUTPUT]
                         [--version] [-m] [-O] [-x] [-n SCHEMA]
                         [-N SCHEMA] [-t TABLE] [-T TABLE]
                         dbname
checkSchemaChanges.py: error: unrecognized arguments: MYHOSTNAME mydatabaseuser

Which is a mixture of my new script (checkSchemaChanges.py, and MYHOSTNAME and mydatabaseuser at the bottom) and the parameters from dbtoyaml, which are all correct.

Could it be the double set of parameters which is confusing argparse?

Waylan · Accepted Answer · 2015-02-10 21:07:25Z

17

When I'm writing a command line script I oftentimes will specifically design my script so this is possible. The key is to parse the args separate from the main function.

For example the main function might look like this:

def main(**kwargs):
    # the body of the script goes here

Then elsewhere in the module I will configure the arg parser, parse the args and pass the result into the main script:

def run():
    parser = ... # configure parser here
    configs = parse_args(parser)
    main(**configs)

That way, if someone wants to call the script from within Python, they can do so (it also makes testing much easier):

import somescript
somescript.main(option='value', option2='value2')

Unfortunately, it appears that the authors of the script you are using did not do anything like that. As stated in another answer you can overwrite sys.argv, then import the script. While that may feel hacky, it should be less resource intensive than opening a new process and calling the command separately.

answered Feb 10, 2015 at 21:07

Waylan

44k14 gold badges93 silver badges118 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

user5359531 Over a year ago

I've started doing this and it has made my scripts much more flexible and reusable. Great suggestion.

Joran Beasley · Accepted Answer · 2015-02-10 19:47:58Z

9

this seems like not the best way to do it but you can probably set sys.argv

import sys
sys.argv += ['-m','-H MYHOSTNAME' .... other options]
from pyrseas import dbtoyaml
dbtoyaml.main()

but really I have no idea what dbtoyaml.py lookslike or is

answered Feb 10, 2015 at 19:47

Joran Beasley

114k13 gold badges167 silver badges187 bronze badges

1 Comment

Stev_k Over a year ago

That, worked, but I had to replace sys.argv, not add to it, i.e. sys.argv = ... thanks

Collectives™ on Stack Overflow

importing a python script from another script and running it with arguments

2 Answers 2

1 Comment

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related