test lab

Author: abhishekkrthakur

colabcode/code.py

@@ -20,13 +20,22 @@
 
 class ColabCode:
     def __init__(
-        self, port=10000, password=None, authtoken=None, mount_drive=False, code=True
+        self,
+        port=10000,
+        password=None,
+        authtoken=None,
+        mount_drive=False,
+        code=True,
+        lab=False,
     ):
         self.port = port
         self.password = password
         self.authtoken = authtoken
         self._mount = mount_drive
         self._code = code
+        self._lab = lab
+        if self._lab:
+            self._run_lab()
         if self._code:
             self._install_code()
             self._install_extensions()
@@ -61,6 +70,24 @@ def _start_server(self):
         else:
             print(f"Public URL: {url}")
 
+    def _run_lab(self):
+        os.system(f"fuser -n tcp -k {self.port}")
+        if self._mount and colab_env:
+            drive.mount("/content/drive")
+        if self.password:
+            lab_cmd = f"jupyter-lab --ip='localhost' --no-browser --NotebookApp.token='' --NotebookApp.password='{self.password}'"
+        else:
+            lab_cmd = "jupyter-lab --ip='localhost' --no-browser --NotebookApp.token='' --NotebookApp.password=''"
+        with subprocess.Popen(
+            [lab_cmd],
+            shell=True,
+            stdout=subprocess.PIPE,
+            bufsize=1,
+            universal_newlines=True,
+        ) as proc:
+            for line in proc.stdout:
+                print(line, end="")
+
     def _run_code(self):
         os.system(f"fuser -n tcp -k {self.port}")
         if self._mount and colab_env:

setup.py

@@ -9,7 +9,7 @@
     setup(
         name="colabcode",
         scripts=["scripts/colabcode"],
-        version="0.1.2",
+        version="0.2.0",
         description="ColabCode - Run codeserver on Colab!",
         long_description=long_description,
         long_description_content_type="text/markdown",
@@ -19,7 +19,12 @@
         license="MIT License",
         packages=find_packages(),
         include_package_data=True,
-        install_requires=["pyngrok>=5.0.0", "nest_asyncio==1.4.3", "uvicorn==0.13.1"],
+        install_requires=[
+            "pyngrok>=5.0.0",
+            "nest_asyncio==1.4.3",
+            "uvicorn==0.13.1",
+            "jupyterlab==3.0.7",
+        ],
         platforms=["linux", "unix"],
         python_requires=">3.5.2",
     )