Skip to content

Commit eb9bc0f

Browse files
authored
Remove empty attachments from Markdown cell (#27)
1 parent e509672 commit eb9bc0f

3 files changed

Lines changed: 18 additions & 0 deletions

File tree

src/clean_notebook/clean.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,6 +49,9 @@ def clean_single_notebook(
4949
if not cell["source"] and not keep_empty:
5050
nb["cells"].remove(cell)
5151
cleaned = True
52+
if "attachments" in cell and len(cell["attachments"]) == 0:
53+
del cell["attachments"]
54+
cleaned = True
5255

5356
if not nb["cells"]:
5457
print(f"Notebook '{file}' does not have any valid cells.")

tests/data/clean_vscode.ipynb

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,13 @@
1919
"source": [
2020
"print(\"saved from vscode\")"
2121
]
22+
},
23+
{
24+
"cell_type": "markdown",
25+
"metadata": {},
26+
"source": [
27+
"# Markdown"
28+
]
2229
}
2330
],
2431
"metadata": {

tests/data/dirty_vscode.ipynb

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,14 @@
3030
"source": [
3131
"print(\"saved from vscode\")"
3232
]
33+
},
34+
{
35+
"attachments": {},
36+
"cell_type": "markdown",
37+
"metadata": {},
38+
"source": [
39+
"# Markdown"
40+
]
3341
}
3442
],
3543
"metadata": {

0 commit comments

Comments
 (0)