jupyter notebook 版本管理

Jupyter Notebook本身不适合用Git进行版本控制，可以将notebook转成python脚本

生成配置文件

运行命令，生成配置文件~/.jupyter/ipython_notebook_config.py

jupyter notebook --generate-config

返回配置文件所在路径，如

Writing default config to: C:\Users\simon.jupyter\jupyter_notebook_config.py

编辑配置文件

在生成的文件中添加，便于通过 post-save 的钩子自动将 .ipynb 文件转换成 .py 文件，生成的.py文件就可以用git进行版本管理了，jupyter中可以用!git status的形式运行系统命令。

### If you want to auto-save .html and .py versions of your notebook:
# modified from: https://github.com/ipython/ipython/issues/8009
# Solution2: https://jupyter-notebook.readthedocs.io/en/stable/extending/savehooks.html
import os
from subprocess import check_call
import re

def clear_prompt(dir_path, nb_fname, log_func):
    """remove the number in '# In[ ]:'"""
    name, ext = os.path.splitext(nb_fname)
    pattern = re.compile(r'^# In\[\d+\]:')

    for n_ext in ['.py', '.txt']:
        script_name = os.path.join(dir_path, name+n_ext)
        if os.path.exists(script_name):
            new_lines = []
            with open(script_name, 'rt') as f:
                lines = f.readlines()
            for line in lines:
                new_line = re.sub(pattern, '# In[ ]:', line)
                new_lines.append(new_line)
            with open(script_name, 'wt') as f:
                f.writelines(new_lines)
            log_func('Remove number in "# In[ ]:"! File Name: %s' % script_name)
            break

def post_save(model, os_path, contents_manager):
    """post-save hook for converting notebooks to .py scripts"""
    if model['type'] != 'notebook':
        return # only do this for notebooks
    d, fname = os.path.split(os_path)
    check_call(['jupyter', 'nbconvert', '--to', 'script', fname], cwd=d)  # '--no-prompt',
    log = contents_manager.log
    # log.info('Filename:%s'%fname)
    clear_prompt(d, fname, log.info)
    # check_call(['ipython', 'nbconvert', '--to', 'html', fname], cwd=d)

c.FileContentsManager.post_save_hook = post_save

参考

1. Jupyter Notebook版本控制
2. Python更改ipython notebook的默认工作路径