如何在Python包的无参函数中获取调用它的Jupyter Notebook文件路径？

要实现无需传参自动获取当前调用convertMe()的Jupyter Notebook路径，你可以利用Jupyter内核提供的API直接读取环境信息，替代inspect.stack()的方法。下面是两种可靠的实现方案：

方案一：利用Jupyter内核的get_ipython()接口（推荐） #

在Jupyter环境中，get_ipython()函数可以访问内核的会话信息，其中包含当前Notebook的路径。这个方法兼容性较好，同时支持Jupyter Notebook和JupyterLab。

修改你的mypackage.py代码如下：

import os

def convertMe():
    try:
        # 获取Jupyter内核实例
        ip = get_ipython()
        # 从会话头信息中提取Notebook相对路径
        notebook_rel_path = ip.session.parent_header['path']
        # 转换为绝对路径，方便后续处理
        notebook_abs_path = os.path.abspath(notebook_rel_path)
        
        # 这里加入你的转换逻辑，比如读取Notebook内容并转换
        print(f"成功获取当前Notebook路径：{notebook_abs_path}")
        # 示例：使用nbformat读取Notebook（需先安装nbformat）
        # import nbformat
        # nb = nbformat.read(notebook_abs_path, as_version=4)
        # 转换代码...
        
    except NameError:
        # 不在Jupyter环境中调用时抛出错误
        raise RuntimeError("convertMe() 只能在Jupyter Notebook/JupyterLab环境中使用")
    except KeyError:
        # 未保存的Notebook没有路径信息
        raise RuntimeError("无法获取Notebook路径，请先保存你的Notebook文件")

关键说明： #

• 未保存的Notebook：如果用户还没保存过Notebook，parent_header['path']字段会不存在，所以需要捕获KeyError并提示用户保存。
• 路径转换：parent_header['path']返回的是相对路径，用os.path.abspath()转换为绝对路径后，后续处理会更方便。
• 兼容性：该方法在Jupyter Notebook 5.x+、JupyterLab 2.x+版本中测试有效，覆盖大多数常用场景。

方案二：针对经典Jupyter Notebook的专属方法（兼容性有限） #

如果你只需要支持经典的Jupyter Notebook（非JupyterLab），可以直接访问内核的notebook属性获取路径：

import os

def convertMe():
    try:
        ip = get_ipython()
        # 仅适用于经典Jupyter Notebook
        notebook_abs_path = ip.shell.notebook.notebook_path
        
        print(f"当前Notebook路径：{notebook_abs_path}")
        # 转换逻辑...
        
    except NameError:
        raise RuntimeError("convertMe() 只能在Jupyter Notebook环境中使用")
    except AttributeError:
        # JupyterLab或其他环境中该属性不存在
        raise RuntimeError("当前环境不支持该方法，请使用方案一")

这个方法更直接，但不适用于JupyterLab，所以除非你明确只支持经典Notebook，否则优先选择方案一。

额外优化：兼容手动传参 #

如果你的包需要支持非Jupyter环境，可以添加手动传参的备选逻辑，兼顾自动获取和灵活调用：

import os

def convertMe(notebook_path=None):
    if notebook_path is None:
        # 尝试自动获取路径
        try:
            ip = get_ipython()
            notebook_rel_path = ip.session.parent_header['path']
            notebook_path = os.path.abspath(notebook_rel_path)
        except (NameError, KeyError):
            raise RuntimeError("无法自动获取Notebook路径，请手动传入路径或在Jupyter环境中调用")
    
    # 转换逻辑...

内容的提问来源于stack exchange，提问作者Acedon