Using sys.argv from another .py file - python

Question

I have a file (test.py) that receives sys.argv from the console/bash:

import sys

def main():
    ans = int(sys.argv[1])**int(sys.argv[1])
    with open('test.out', 'w') as fout:
        fout.write(str(ans))

if __name__ == '__main__':
    main()

Usually, I could just do $ python test.py 2 to produce the test.out file. But I need to call the main() function from test.py from another script.

I could do as below in (call.py) but is there any other way to run pass an argument to sys.argv to main() in `test.py?

import os

number = 2
os.system('python test.py '+str(number))

Please note that I CANNOT modify test.py and I also have a main() in call.py which does other things.

thefourtheye · Accepted Answer · 2014-05-02 05:09:04Z

2

You can use your program as it is. Because, irrespective of the file invoked by python, all the python files will get the command line arguments passed.

But you can make the main function accept sys.argv as the default parameter. So, main will always take the sys.argv by default. When you pass a different list, it will take the first element and process it.

test.py

import sys

def main(args = sys.argv):
    ans = int(args[1])**int(args[1])
    with open('test.out', 'w') as fout:
        fout.write(str(ans))

call.py

import sys, test
test.main()

edited May 2, 2014 at 5:09

answered May 2, 2014 at 4:59

thefourtheye

241k53 gold badges466 silver badges505 bronze badges

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

Comments

Kirk Strauser · Accepted Answer · 2014-05-02 05:06:10Z

2

Write that like:

import sys

def main(num):
    ans = int(num)**int(num)
    with open('test.out', 'w') as fout:
        fout.write(str(ans))

if __name__ == '__main__':
    main(sys.argv[1])

so that your main() function doesn't have to know about sys.argv - it just handles the parameters being passed in to it.

answered May 2, 2014 at 5:06

Kirk Strauser

31.1k5 gold badges53 silver badges69 bronze badges

Comments

user3591723 · Accepted Answer · 2014-05-02 06:42:11Z

2

Without modifying test.py you can still run it just as you have it, just do call.py:

import test

test.main()

then $ python call.py ARG will still work. Since you've already imported sys in test, you don't need to reimport it unless you want to use sys in call.py. Note that sys.argv[0]=='call.py' not test.py if use test through call.

edited May 2, 2014 at 6:42

answered May 2, 2014 at 5:04

user3591723

1,2841 gold badge10 silver badges23 bronze badges

Comments

Droppy · Accepted Answer · 2014-05-02 05:12:29Z

0

Create a function to do your calculation and file writing, which can be called from any other module:

power.py:

import sys

def power(arg):    
    ans = arg ** arg
    with open('test.out', 'w') as fout:
        fout.write(str(ans))

if __name__ == '__main__':
    power(int(sys.argv[1]))

other.py:

import power

power.power(2)

answered May 2, 2014 at 5:12

Droppy

9,7311 gold badge22 silver badges27 bronze badges

Collectives™ on Stack Overflow

Using sys.argv from another .py file - python

4 Answers 4

test.py

call.py

Comments

Comments

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

test.py

call.py

Comments

Comments

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related