#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,65 @@
+# Copyright 2018 TWO SIGMA OPEN SOURCE, LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#        http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import argparse
+import sys
+import beakerx
+from notebook import notebookapp as app
+from .install import install, uninstall
+from .bkr2ipynb import main
+
+def install_subparser(subparser):
+    install_parser = subparser.add_parser('install', help='installs BeakerX extensions')
+    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 extensions')
+    uninstall_parser.set_defaults(func=lambda args : 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 run_jupyter(jupyter_commands):
+    app.launch_new_instance(jupyter_commands)
+
+def init_parser():
+
+    parser = argparse.ArgumentParser()
+    parser.add_argument('--version', action='version', version=beakerx.__version__)
+    parser.set_defaults(func=run_jupyter)
+
+    subparsers = parser.add_subparsers()
+    install_subparser(subparsers)
+    uninstall_subparser(subparsers)
+    bkr2ipynb_subparser(subparsers)
+    return parser
+
+def parse():
+    parser = init_parser()
+    args, jupyter_commands = parser.parse_known_args()
+    if args.func == run_jupyter:
+        args.func(jupyter_commands)
+    elif not jupyter_commands:
+        args.func(args)
+    else:
+        parser.parse_args(jupyter_commands)
+

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 .)