How to pass parameters to python function using argparse?

Question

I have a foo.py using argparse to get command line parameters for the main() function.

""" foo.py """
import argparse

def main():
    parser = argparse.ArgumentParser(description='test')
    parser.add_argument('--all', '-a', action='store_true', help='all')
    args = parser.parse_args()

    if args.all:
        pass

if __name__ == '__main__':
    main()

And I have to test this main() function on another Python script bar.py. My question is how to pass parameters in bar.py. My current solution is changing the sys.argv. Looking for better solution.

""" bar.py """
import sys
import foo

if __name__ == '__main__':
    sys.argv.append('-a')
    foo.main()

Are you trying to get the arguments from foo.py into bar.py? Why not just move the parser into bar.py? — TheF1rstPancake
– TheF1rstPancake, Commented Sep 26, 2017 at 5:54

payala · Accepted Answer · 2018-12-13 18:16:45Z

5

You can modify main function to receive a list of args.

""" foo.py """
import argparse

def main(passed_args=None):
    parser = argparse.ArgumentParser(description='test')
    parser.add_argument('--all', '-a', action='store_true', help='all')
    args = parser.parse_args(passed_args)

    if args.all:
        pass

if __name__ == '__main__':
    main()

""" bar.py """
import sys
import foo

if __name__ == '__main__':
    foo.main(["-a"])

edited Dec 13, 2018 at 18:16

payala

1,40715 silver badges28 bronze badges

answered Sep 26, 2017 at 5:57

Sraw

20.6k11 gold badges61 silver badges93 bronze badges

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

2 Comments

hpaulj Over a year ago

You don't need the if test. If passed_args is None parse_args uses sys.argv[1:].

northtree Over a year ago

@hpaulj Thanks, guys.

Collectives™ on Stack Overflow

How to pass parameters to python function using argparse?

1 Answer 1

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related