Get total word count of all markdown cells in all .ipynb files in a given directory

wordcount.py

import io
import glob
from IPython.nbformat import current

word_count = 0
directory = "dirpath"

for direc in glob.glob("{}*.ipynb".format(directory)):
    with io.open(direc, 'r', encoding='utf-8') as f:
        nb = current.read(f, 'json')

    for cell in nb.worksheets[0].cells:
        if cell.cell_type == "markdown":
            word_count += len(cell['source'].replace('#', '').lstrip().split(' '))
print(word_count)