A file calls the function of the B file to get the file name of the A file

Question

Such as the title. What should I do with python3?

test1.py

import os
def get_file_name():
    return os.path.basename(__file__)

test2.py

from test1 import get_file_name
file_name = get_file_name()
print(file_name)

now, I want to get the file name of test2

Answer 1

Change test1.py to:

import os, inspect
def get_file_name():
    return os.path.basename(inspect.stack()[1].filename)