如何在文件B中访问文件A的名称，其中文件B导入到文件A中

Question

我有一个文件已导入到我的作业中，该文件具有多种功能，其中之一是

def PDF(input_file:str, output_file:str):
    CODE TO MAKE A PDF OUT OF THE FILE I NEED TO TURN IN

我希望能够在不带参数的情况下运行 PDF，并让它选择已导入的文件并创建一个同名的输出文件，但文件扩展名为 .pdf。

我希望文件 A 中的代码尽可能少，这样每次我需要使用文件 B 时，额外的代码就会最少。

我尝试使用 name 和 file 以及类似于

的代码

def get_caller_filename():
    frame = inspect.currentframe().f_back
    file_name = frame.f_code.co_filename
    return file_name

但它们都返回文件 B（我正在导入的文件）。

提前谢谢您！！

Answer 1

import os
import inspect

def PDF(input_file=None, output_file=None):
    if input_file is None:
        frame = inspect.currentframe().f_back
        input_file = frame.f_code.co_filename

    if output_file is None:
        output_file = os.path.splitext(input_file)[0] + '.pdf'

    print(f"Created PDF file: {output_file}")

# Simple call

# PDF() no args