Calling a function that lives in `a.py`, that gets the filename of `b.py`

Question:

I have two files a.py and b.py.

a.py is where my functions live, and b.py is where my function calls live.

From b.py, I need to call a function that lives in a.py, that gets the filename of b.py.

a.py

import os

# functions
def foo():
    filename = os.path.basename(os.path.realpath(__file__))
    """ and then some other stuff """

b.py

from a import *

# function calls
foo()

But obviously __file__ in a.py is:

a.py

Is there a way to make this work without doing this:

a.py

import os

# functions
def foo():
    """ some other stuff """

b.py

from a import *

filename = os.path.basename(os.path.realpath(__file__))

# function calls
foo()

Which is completely possible, just much less elegant.

Asked By: mathbike

||

Answers:

If you only need the filename you can just import sys and use sys.argv[0]. If you want to use the filename and do some other things you can pass sys.argv[0] into foo.

If you really want to have all the functionality live in a.py you can try this.

Answered By: KyIe LiebIer
Categories: questions Tags:
Answers are sorted by their score. The answer accepted by the question owner as the best is marked with
at the top-right corner.