add config for deployment (jupyter-server#98)

* add config for deployment

* Bump to 0.6.1

Author: Zsailer

data_studio_jupyter_extensions/_version.py

@@ -1 +1 @@
-__version__ = "0.6.0"  # pragma: no cover
+__version__ = "0.6.1"  # pragma: no cover

data_studio_jupyter_extensions/app.py

@@ -21,6 +21,7 @@
 from .telemetry.utils import DEFAULT_STATIC_FILES_PATH
 from .telemetry.utils import DEFAULT_TEMPLATE_FILES_PATH
 from .telemetry.utils import get_schema_files
+from .traits import IntFromEnv
 from .traits import UnicodeFromEnv
 from .utils import apply_defaults
 from .utils import get_available_mode_names
@@ -136,6 +137,14 @@ def _default_ssl_cert_file(self):  # pragma: no cover
         allow_none=True,
     ).tag(config=True)
 
+    ds_ws_max_buffer_size = IntFromEnv(
+        name="DS_TORNADO_MAX_BUFFER", default_value=400 * 1024 * 1024  # 400MiB
+    ).tag(config=True)
+
+    ds_ws_max_message_size = IntFromEnv(
+        name="DS_TORNADO_MAX_MSG", default_value=400 * 1024 * 1024  # 400MiB
+    ).tag(config=True)
+
     telemetry_bus_class = Type(default_value=TelemetryBus).tag(config=True)
 
     def _load_mode(self):
@@ -161,6 +170,7 @@ def _jupyter_server_config(self):
             "KernelProvisionerFactory": {
                 "default_provisioner_name": "notebook-service-provisioner"
             },
+            "KernelManager": {"autorestart": False},
             "Session": {"key": b"notebooks"},
             "ServerApp": {
                 "log_level": "DEBUG",
@@ -170,6 +180,10 @@ def _jupyter_server_config(self):
                 "logout_handler_class": "data_studio_jupyter_extensions.auth.logout.DSLogoutHandler",
                 "cookie_options": {"expires_days": 1},
                 "base_url": self.server_base_url,
+                "tornado_settings": {
+                    "ds_ws_max_buffer_size": self.ds_ws_max_buffer_size,
+                    "ds_ws_max_message_size": self.ds_ws_max_message_size,
+                },
             },
         }
         apply_defaults(config, base_config)

pyproject.toml

@@ -23,7 +23,7 @@ testpaths = [
 ]
 
 [tool.tbump.version]
-current = "0.6.0"
+current = "0.6.1"
 regex = '''
   (?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)
   ((?P<channel>a|b|rc|.dev)(?P<release>\d+))?

version

@@ -1 +1 @@
-0.6.0
+0.6.1