From e93773ac0b90cb42b6eacddd583fc12da1412592 Mon Sep 17 00:00:00 2001 From: Jonathan Sick Date: Wed, 15 May 2024 12:26:23 -0400 Subject: [PATCH] temp: Add more logging statements --- src/noteburst/jupyterclient/jupyterlab.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/noteburst/jupyterclient/jupyterlab.py b/src/noteburst/jupyterclient/jupyterlab.py index 65886cb..4cc932f 100644 --- a/src/noteburst/jupyterclient/jupyterlab.py +++ b/src/noteburst/jupyterclient/jupyterlab.py @@ -524,6 +524,7 @@ async def log_into_lab(self) -> None: method="GET", body=r.text, ) + self.logger.debug("Logged into JupyterLab with XSRF token") async def spawn_lab(self) -> JupyterImage: """Spawn a JupyterLab pod.""" @@ -662,6 +663,7 @@ async def open_lab_session( Send and receive messages from JupyterLab using the ``websocket`` property on `JupyterLabSession`. """ + self.logger.debug("Opening JupyterLab session") session_url = self.url_for(f"user/{self.user.username}/api/sessions") session_type = "notebook" if notebook_name else "console" body = { @@ -673,6 +675,10 @@ async def open_lab_session( headers = {} if self._lab_xsrf: headers["X-XSRFToken"] = self._lab_xsrf + else: + self.logger.warning( + "No XSRF token found for JupyterLab, using hub token." + ) r = await self.http_client.post( session_url, json=body, headers=headers )