Rework debug logging.

lsst-sqre · Aug 2, 2024 · ec9b870 · ec9b870
1 parent 35099a8
commit ec9b870
Showing 1 changed file with 38 additions and 30 deletions.
diff --git a/src/mobu/storage/nublado.py b/src/mobu/storage/nublado.py
@@ -597,27 +597,31 @@ async def auth_to_hub(self) -> None:
         # redirection, because httpx doesn't do that if following redirects
         # automatically.
         while r.is_redirect:
+            self._logger.debug(
+                "Following hub redirect looking for _xsrf cookies",
+                method=r.request.method,
+                url=r.url.copy_with(query=None, fragment=None),
+                status_code=r.status_code,
+            )
             xsrf = self._extract_xsrf(r)
-            if xsrf and xsrf != self._lab_xsrf:
-                if xsrf != self._hub_xsrf:
-                    self._hub_xsrf = xsrf
-                    self._logger.debug(
-                        "Set _hub_xsrf",
-                        url=r.url.copy_with(query=None, fragment=None),
-                        status_code=r.status_code,
-                    )
-            next_url = urljoin(url, r.headers["Location"])
-            r = await self._client.get(next_url, follow_redirects=False)
-        r.raise_for_status()
-        xsrf = self._extract_xsrf(r)
-        if xsrf and xsrf != self._lab_xsrf:
-            if xsrf != self._hub_xsrf:
+            if xsrf and xsrf != self._hub_xsrf:
                 self._hub_xsrf = xsrf
                 self._logger.debug(
                     "Set _hub_xsrf",
                     url=r.url.copy_with(query=None, fragment=None),
                     status_code=r.status_code,
                 )
+            next_url = urljoin(url, r.headers["Location"])
+            r = await self._client.get(next_url, follow_redirects=False)
+        r.raise_for_status()
+        xsrf = self._extract_xsrf(r)
+        if xsrf and xsrf != self._hub_xsrf:
+            self._hub_xsrf = xsrf
+            self._logger.debug(
+                "Set _hub_xsrf",
+                url=r.url.copy_with(query=None, fragment=None),
+                status_code=r.status_code,
+            )
         elif not self._hub_xsrf:
             msg = "No _xsrf cookie set in login reply from JupyterHub"
             raise JupyterProtocolError(msg)
@@ -647,29 +651,33 @@ async def auth_to_lab(self) -> None:
             url, headers=headers, follow_redirects=False
         )
         while r.is_redirect:
-            xsrf = self._extract_xsrf(r)
-            if xsrf and xsrf != self._hub_xsrf:
-                if xsrf != self._lab_xsrf:
-                    self._lab_xsrf = xsrf
-                    self._logger.debug(
-                        "Set _lab_xsrf",
-                        url=r.url.copy_with(query=None, fragment=None),
-                        status_code=r.status_code,
-                    )
-            next_url = urljoin(url, r.headers["Location"])
-            r = await self._client.get(
-                next_url, headers=headers, follow_redirects=False
+            self._logger.debug(
+                "Following lab redirect looking for _xsrf cookies",
+                method=r.request.method,
+                url=r.url.copy_with(query=None, fragment=None),
+                status_code=r.status_code,
             )
-        r.raise_for_status()
-        xsrf = self._extract_xsrf(r)
-        if xsrf and xsrf != self._hub_xsrf:
-            if xsrf != self._lab_xsrf:
+            xsrf = self._extract_xsrf(r)
+            if xsrf and xsrf != self._lab_xsrf:
                 self._lab_xsrf = xsrf
                 self._logger.debug(
                     "Set _lab_xsrf",
                     url=r.url.copy_with(query=None, fragment=None),
                     status_code=r.status_code,
                 )
+            next_url = urljoin(url, r.headers["Location"])
+            r = await self._client.get(
+                next_url, headers=headers, follow_redirects=False
+            )
+        r.raise_for_status()
+        xsrf = self._extract_xsrf(r)
+        if xsrf and xsrf != self._lab_xsrf:
+            self._lab_xsrf = xsrf
+            self._logger.debug(
+                "Set _lab_xsrf",
+                url=r.url.copy_with(query=None, fragment=None),
+                status_code=r.status_code,
+            )
         if not self._lab_xsrf:
             msg = "No _xsrf cookie set in login reply from lab"
             raise JupyterProtocolError(msg)