#6697 beakerx commands refactor

README.md

@@ -57,7 +57,7 @@ conda create -y -n beakerx 'python>=3' nodejs pandas openjdk maven
 source activate beakerx
 conda install -y -c conda-forge ipywidgets
 (cd beakerx; pip install -e . --verbose)
-beakerx-install
+beakerx install
 ```
 
 ### Build and Install for Lab
@@ -67,7 +67,7 @@ conda create -y -n labx 'python>=3' nodejs pandas openjdk maven pytest
 source activate labx
 conda install -y -c conda-forge jupyterlab
 (cd beakerx; pip install -e . --verbose)
-beakerx-install
+beakerx install
 jupyter labextension install @jupyter-widgets/jupyterlab-manager
 (cd js/lab; jupyter labextension install .)
 ```

beakerx/beakerx/__init__.py

@@ -20,7 +20,7 @@
 from ._version import version_info, __version__
 from .handlers import load_jupyter_server_extension
 from .environment import *
-import json
+from .commands import parse
 
 def _jupyter_nbextension_paths():
     return [{
@@ -40,5 +40,11 @@ def _jupyter_nbextension_paths():
 def _jupyter_server_extension_paths():
     return [dict(module="beakerx")]
 
-
 beakerx = BeakerX()
+
+def run():
+    try:
+        parse()
+    except KeyboardInterrupt:
+        return 130
+    return 0

beakerx/beakerx/bkr2ipynb.py

@@ -86,16 +86,15 @@ def convertNotebook(notebook):
     nb = parseBkr(data)
     nbformat.write(nb, notebook.partition('.')[0] + '.ipynb')
 
-def main():
+def main(args):
+    for notebook in args.notebooks:
+        convertNotebook(notebook)
+
+if __name__ == "__main__":
     parser = argparse.ArgumentParser()
     parser.add_argument('notebooks', nargs='+',
-                        help="beaker notebooks to be converted. Enter *.bkr in case you want to convert all notebooks at once.")
+                    help="beaker notebooks to be converted. Enter *.bkr in case you want to convert all notebooks at once.")
     if len(sys.argv) == 1:
         parser.print_help()
     args = parser.parse_args()
-
-    for notebook in args.notebooks:
-        convertNotebook(notebook)
-
-if __name__ == "__main__":
-    main()
+    main(args)

beakerx/beakerx/commands.py

@@ -0,0 +1,44 @@
+import argparse
+import sys
+import subprocess
+import beakerx
+from .install import install, uninstall
+from .bkr2ipynb import main
+
+def install_subparser(subparser):
+    install_parser = subparser.add_parser('install', help='installs beakerx extension')
+    install_parser.set_defaults(func = install)
+    install_parser.add_argument("--prefix",
+                                help="location of the environment to install into",
+                                default=sys.prefix)
+    return subparser
+
+def uninstall_subparser(subparser):
+    uninstall_parser = subparser.add_parser('uninstall', help='uninstalls beakerx extension')
+    uninstall_parser.set_defaults(func=lambda x : uninstall())
+    return subparser
+
+def bkr2ipynb_subparser(subparser):
+    bkr2ipynb_parser = subparser.add_parser('bkr2ipynb', help='converts beaker notebooks to ipynb format')
+    bkr2ipynb_parser.set_defaults(func=main)
+    bkr2ipynb_parser.add_argument('notebooks', nargs='+',
+                                  help="beaker notebooks to be converted. Enter *.bkr in case you want to convert all notebooks at once.")
+    return subparser
+
+def init_parser():
+    version = 'beakerx %s' % (beakerx.__version__)
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--version', action='version', version=version)
+    parser.set_defaults(func=lambda x : subprocess.check_call(["jupyter", "notebook"]))
+
+    subparsers = parser.add_subparsers()
+    install_subparser(subparsers)
+    uninstall_subparser(subparsers)
+    bkr2ipynb_subparser(subparsers)
+    return parser
+
+def parse():
+    parser = init_parser()
+    args = parser.parse_args()
+    args.func(args)

beakerx/beakerx/install.py

@@ -196,17 +196,11 @@ def _install_beakerx(args):
     _set_conf_privileges()
 
 
-def install():
-    try:
-        parser = make_parser()
-        args = parser.parse_args()
-        if args.disable:
-            _disable_beakerx()
-        else:
-            _install_beakerx(args)
-    except KeyboardInterrupt:
-        return 130
-    return 0
+def install(args):
+    _install_beakerx(args)
+
+def uninstall():
+    _disable_beakerx()
 
 
 if __name__ == "__main__":

beakerx/setup.py

@@ -76,8 +76,7 @@
     ],
     entry_points={
         'console_scripts': [
-            'beakerx-install = beakerx.install:install',
-            'bkr2ipynb = beakerx.bkr2ipynb:main',
+            'beakerx = beakerx:run'
         ]
     },
     package_data={

docker/setup.sh

@@ -15,7 +15,7 @@
 
 source activate beakerx
 (cd beakerx; pip install -e . --verbose)
-beakerx-install
+beakerx install
 jupyter labextension install @jupyter-widgets/jupyterlab-manager
 (cd js/lab; jupyter labextension install .)