本文将介绍如何在Python库函数中获取调用该函数的主文件名。通过sys.argv[0]获取正在运行脚本的路径，并利用ntpath.basename提取文件名，实现动态获取调用者文件名，从而避免依赖源码浏览。
在开发Python项目时，有时需要在库函数中获取调用该函数的脚本文件名。例如，你可能希望根据不同的调用脚本执行不同的逻辑，或者仅仅是为了日志记录和调试。以下提供一种简洁有效的方法来实现这个需求。
使用sys.argv[0]和ntpath.basename
Python的sys模块提供了访问与Python解释器及其环境密切相关的变量和函数的功能。其中，sys.argv是一个包含命令行参数的列表，sys.argv[0]通常是正在运行的脚本的路径。
ntpath模块（或os.path，如果你的目标平台不一定是Windows）提供了处理文件路径的函数。ntpath.basename()函数可以从路径中提取文件名。
结合这两个模块，我们可以编写一个get_my_name()函数，该函数返回调用脚本的文件名。
import sys
import ntpath
def get_my_name():
"""
获取调用脚本的文件名。
Returns:
str: 调用脚本的文件名。
"""
return ntpath.basename(sys.argv[0])
# 示例
if __name__ == '__main__':
print(f"当前运行的脚本是: {get_my_name()}")示例用法
假设我们有三个文件：A.py、B.py和C.py。
C.py（库文件）：
import sys
import ntpath
def get_my_name():
"""
获取调用脚本的文件名。
Returns:
str: 调用脚本的文件名。
"""
return ntpath.basename(sys.argv[0])A.py:
import C print(C.get_my_name())
B.py:
import C print(C.get_my_name())
当运行A.py时，输出将是A.py。当运行B.py时，输出将是B.py。
注意事项
- sys.argv[0]的值取决于脚本的启动方式。在某些情况下，它可能包含完整路径，而在其他情况下，可能只包含文件名。ntpath.basename()确保我们始终获得文件名部分。
- 如果脚本是通过绝对路径调用的，sys.argv[0]将包含绝对路径。如果通过相对路径调用，sys.argv[0]将包含相对路径。
- 在某些特殊情况下，例如使用某些打包工具或在某些特定的执行环境中，sys.argv[0]可能不是预期的值。在这种情况下，可能需要考虑其他方法来获取文件名，例如检查调用栈。
总结
通过结合sys.argv[0]和ntpath.basename()，我们可以轻松地在Python库函数中获取调用脚本的文件名。这种方法简单、有效，并且不依赖于浏览源代码。请注意，在特殊情况下，可能需要考虑其他方法。
