add option to strip ipynb output

n-wbrown · Jul 24, 2024 · 8db6c1b · 8db6c1b
1 parent 6f849f1
commit 8db6c1b
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/cookiecutter.json b/cookiecutter.json
@@ -11,5 +11,6 @@
     "auto_git_setup": ["no", "yes"],
     "git_remote_name": "origin",
     "readme_format": ["md", "rst"],
-    "simple": ["yes", "no"]
+    "simple": ["yes", "no"],
+    "strip_ipynb_outputs": ["yes", "no"]
 }
diff --git a/hooks/post_gen_project.sh b/hooks/post_gen_project.sh
@@ -1,5 +1,8 @@
 #!/bin/bash
 git init
+{% if cookiecutter.strip_ipynb_outputs == "yes" %}
+git config filter.strip-notebook-output.clean 'jupyter nbconvert --ClearOutputPreprocessor.enabled=True --to=notebook --stdin --stdout --log-level=ERROR'
+{% endif %}
 {% if cookiecutter.auto_git_setup == "yes" %}
 git add -A
 git remote add {{ cookiecutter.git_remote_name }} [email protected]:{{ cookiecutter.github_repo_group }}/{{ cookiecutter.repo_name }}.git

diff --git a/{{ cookiecutter.folder_name }}/.gitattributes b/{{ cookiecutter.folder_name }}/.gitattributes
@@ -1 +1,5 @@
 .git_archival.txt  export-subst
+
+{% if cookiecutter.strip_ipynb_outputs=="yes"%}
+*.ipynb filter=strip-notebook-output  
+{%endif%}