Last active
April 20, 2020 23:59
-
-
Save snickell/b29a189c56ad1833a8ee8061b34e12d0 to your computer and use it in GitHub Desktop.
Gets a reference to the current Jupyter Notebook and Server when run inside a Cell (also works for JupyterLab), can map kernel relative paths to files_urls
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from pathlib import Path | |
import re | |
import requests | |
from urllib.parse import urljoin | |
import ipykernel | |
from notebook.notebookapp import list_running_servers | |
def files_url_for(kernel_path): | |
_server = server() | |
path_of_base_url = _server['notebook_dir'] | |
return urljoin( | |
_files_base_url(_server), | |
str(Path(kernel_path).resolve().relative_to(path_of_base_url)) | |
) | |
def notebook(): | |
return _current_jupyter()['notebook'] | |
def server(): | |
return _current_jupyter()['server'] | |
def _files_base_url(server): | |
return urljoin( | |
server['url'], | |
# TODO: this is a hacky whacky assumption... 🤦♀️ | |
# what's the least unprincipled way to do this? | |
f'files/' | |
) | |
def _current_jupyter(): | |
kernel_id = _current_kernel_id() | |
return next(( | |
{ | |
'notebook': notebook, | |
'server': server | |
} | |
for notebook, server in _all_notebooks() | |
if notebook['kernel']['id'] == kernel_id | |
)) | |
def _api_url(server, api_name): | |
return urljoin(server['url'], f'api/{api_name}') | |
def _all_notebooks(): | |
def get_session(server): | |
session_url = urljoin(server['url'], 'api/sessions') | |
token_param = { | |
'token': server.get('token', '') | |
} | |
sessions_url = _api_url(server, 'sessions') | |
return requests.get(sessions_url, token_param).json() | |
return ( | |
( notebook, server ) | |
for notebooks, server in ( | |
( get_session(server), server ) for server in list_running_servers() | |
) | |
for notebook in notebooks | |
) | |
def _current_kernel_id(): | |
connections = ipykernel.connect.get_connection_file() | |
return re.search('kernel-(.*).json', connections).group(1) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.