local_jupyter_43_notebook_name.py

# solution for:
# https://stackoverflow.com/questions/12544056/how-to-i-get-the-current-ipython-notebook-name
#
# aimed at:
# IPython 4.3.0
'''
Library for getting the name of a Jupyter notebook from within an IPython
kernel launched for the notebook.
Probably doesn't work on OS X.
Usage:
    jupyter_notebook_name()
'''
import json
import os
import urllib2
import urlparse
import IPython
from IPython.lib import kernel
import subprocess
import notebook.notebookapp
import contextlib

def _iter_listening_sockaddrs(pid):
    for line in subprocess.check_output((
            'lsof',
            '-p', str(pid),
            '-a', '-i', 'TCP',
            '-a', '-s', 'TCP:LISTEN',
            '-P',
            '-Fn',
        )).split():
        if not line:
            continue
        if line.startswith('n'):
            yield line[1:]

def _sockaddr_to_port(sockaddr):
    return urlparse.urlparse('http://' + sockaddr + '/').port

def _parent_pid_listening_ports():
    parent_pid = os.getppid()
    listening_sockaddrs = _iter_listening_sockaddrs(parent_pid)
    return set(map(_sockaddr_to_port, listening_sockaddrs))

def _parent_jupyter_api_url():
    # Use the parent process's listening ports to rule out some servers.
    parent_ports = _parent_pid_listening_ports()

    api_urls = []
    for server in notebook.notebookapp.list_running_servers():
        url = server['url']
        token = server['token']
        parsed = urlparse.urlparse(url)
        if parsed.port not in parent_ports:
            continue
        parsed = parsed._replace(query='token=' + urllib2.quote(token))
        api_urls.append(parsed)

    if len(api_urls) != 1:
        raise Exception('Did not get exactly 1 API URL %(api_urls)s for the listening ports %(parent_ports)s' % locals())
    api_url, = api_urls
    return api_url

# https://stackoverflow.com/questions/12544056/how-do-i-get-the-current-ipython-notebook-name
def jupyter_notebook_name():
    connection_file_path = kernel.get_connection_file()
    connection_file = os.path.basename(connection_file_path)
    kernel_id = connection_file.split('-', 1)[1].split('.')[0]
    api_url = _parent_jupyter_api_url()

    with contextlib.closing(urllib2.urlopen(api_url._replace(path='/api/sessions').geturl())) as f:
        sessions = json.load(f)

    notebook_names = []
    for sess in sessions:
        if sess['kernel']['id'] == kernel_id:
            notebook_path = sess['notebook']['path'].encode('utf-8')
            notebook_filename = os.path.basename(notebook_path)
            notebook_name, _ipynb = os.path.splitext(notebook_filename)
            notebook_names.append(notebook_name)

    if len(notebook_names) != 1:
        raise Exception('Expected exactly one notebook for kernel %(kernel_id)s among the sessions %(sessions)r' % locals())
    notebook_name, = notebook_names
    return notebook_name