获取库函数调用者的主文件名-Python教程-PHP中文网

获取库函数调用者的主文件名

碧海醫心

发布： 2025-08-20 18:22:14

原创

348人浏览过

获取库函数调用者的主文件名

本文将介绍如何在Python库函数中获取调用它的主文件名。假设你有一个库文件，并且多个不同的Python脚本都导入并调用了这个库文件中的一个函数。你希望这个函数能够返回调用它的主脚本的文件名。

import sys, ntpath

def get_my_name():
    return ntpath.basename(sys.argv[0])

登录后复制

代码解释:

sys.argv: sys.argv 是一个包含命令行参数的列表。 sys.argv[0] 始终是正在执行的脚本的路径。
ntpath.basename(): ntpath.basename() 函数从路径中提取基本文件名。即使在非 Windows 平台上，ntpath 也能正确处理路径，并提取文件名。

使用示例:

假设我们有三个文件: A.py, B.py 和 C.py。

A.py:

import C
print(C.get_my_name())

登录后复制

B.py:

NameGPT名称生成器

免费AI公司名称生成器，AI在线生成企业名称，注册公司名称起名大全。

查看详情

import C
print(C.get_my_name())

登录后复制

C.py:

import sys, ntpath

def get_my_name():
    return ntpath.basename(sys.argv[0])

登录后复制

当我们运行 A.py 时，输出将会是 A.py。当我们运行 B.py 时，输出将会是 B.py。

注意事项:

这种方法依赖于 sys.argv[0] 的值。在某些特殊情况下，例如脚本是通过其他方式启动的，sys.argv[0] 可能不会包含预期的脚本路径。
如果脚本是通过绝对路径运行的，ntpath.basename() 将会返回包含 .py 扩展名的完整文件名。如果你想要去除 .py 扩展名，可以使用字符串操作，例如 ntpath.basename(sys.argv[0]).replace(".py", "")。
在某些环境中，sys.argv[0] 可能包含完整的路径信息。如果只需要文件名，始终使用 ntpath.basename() 来提取。

总结:

使用 sys.argv[0] 和 ntpath.basename() 是一种简单有效的方法，可以在Python库函数中获取调用者的主文件名。这种方法避免了复杂的源码查找，并且能够适用于不同的调用场景。记住，根据实际情况处理 sys.argv[0] 的值，并根据需要进行适当的字符串处理，以获得期望的文件名。

以上就是获取库函数调用者的主文件名的详细内容，更多请关注php中文网其它相关文章！