Write script and make it executable (#9)

* write script

* rename script

* Make blackbook script executable.

setup.py

@@ -16,4 +16,5 @@
     url="",
     license="The MIT License (MIT)",
     description="`Black` for Jupyter notebooks.",
+    entry_points={"console_scripts": "blackbook=blackbook.__main__:main"},
 )

src/blackbook/__init__.py

@@ -1 +1,28 @@
-from .main import gen_notebook_files_in_dir, format_notebook_content
+""""""
+
+from typing import Iterator
+
+import re
+import pathlib
+import json
+
+import black
+
+
+def gen_notebook_files_in_dir(path: pathlib.Path) -> Iterator[pathlib.Path]:
+
+    return path.glob("**/*.ipynb")
+
+def format_notebook_content(path: pathlib.Path) -> dict:
+    content = path.read_text()
+    nb = json.loads(content)
+
+    for cell in nb["cells"]:
+        try:
+            string = "".join(cell["source"])
+            formatted_string = black.format_str(string, line_length=black.DEFAULT_LINE_LENGTH)
+            cell["source"] = [s + "\n" for s in formatted_string.split("\n")][:-1]
+        except black.InvalidInput:
+            pass
+
+    return nb

src/blackbook/__main__.py

@@ -0,0 +1,18 @@
+import json
+import pathlib
+import sys
+
+import blackbook
+
+
+def main(path=None):
+    if path is None:
+        path = pathlib.Path(sys.argv[1])
+
+    for notebook_path in blackbook.gen_notebook_files_in_dir(path):
+        nb = blackbook.format_notebook_content(notebook_path)
+
+        notebook_path.write_text(json.dumps(nb))
+
+if __name__ == '__main__':
+    main()