如何在EMR集群的PySpark Jupyter Notebook中动态获取编辑器ID？

以下是几种无需硬编码编辑器ID的实现方案：

• 通过IPython内核接口提取
  利用Jupyter内置的get_ipython()工具，直接从当前内核的会话信息中解析ID，适配大多数本地Jupyter环境：

  from IPython import get_ipython
  
  def get_editor_id():
      ip = get_ipython()
      if ip:
          # 从会话ID中提取编辑器标识，不同环境可能需调整拆分逻辑
          session_id = ip.kernel.session.session
          editor_id = session_id.split('-')[0]
          return editor_id
      return None
  
  # 调用获取ID
  editor_id = get_editor_id()
  

  注：JupyterLab和经典Notebook的会话结构可能略有差异，需根据实际返回值微调解析规则。

• 从Notebook文件元数据读取
  本地运行的Notebook会以.ipynb文件存储，元数据中通常包含编辑器ID，直接读取文件解析即可：

  import json
  import os
  from IPython import get_ipython
  
  def get_editor_id_from_file():
      ip = get_ipython()
      if ip:
          nb_path = ip.notebook_name
          if os.path.exists(nb_path):
              with open(nb_path, 'r', encoding='utf-8') as f:
                  nb_data = json.load(f)
                  # 从元数据中提取ID，字段名可能随环境变化
                  return nb_data.get('metadata', {}).get('id')
      return None
  
  # 调用获取ID
  editor_id = get_editor_id_from_file()
  

  注：云端Jupyter环境（如Colab）可能无法直接访问Notebook文件，该方法不适用。

• 通过Jupyter REST API获取
  如果Jupyter服务器开启了API服务，可调用会话接口获取当前Notebook的ID：

  import requests
  from IPython import get_ipython
  
  def get_editor_id_via_api():
      # 替换为你的Jupyter服务器地址和认证token
      base_url = 'http://localhost:8888'
      token = 'your_jupyter_auth_token'
  
      ip = get_ipython()
      if not ip:
          return None
  
      response = requests.get(f"{base_url}/api/sessions", params={'token': token})
      if response.status_code == 200:
          sessions = response.json()
          current_kernel_id = ip.kernel.session.session
          # 匹配当前内核对应的会话，提取编辑器ID
          for session in sessions:
              if session['kernel']['id'] == current_kernel_id:
                  return session['notebook']['id']
      return None
  
  # 调用获取ID
  editor_id = get_editor_id_via_api()
  

  注：需确保服务器允许API访问，云端环境需调整base_url为对应服务地址。

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