add pre poll function

jsc_custom/spawner/spawner.py

@@ -4,6 +4,7 @@ import json
 import os
 import random
 import re
+import time
 from datetime import datetime
 
 from datefinder import find_dates
@@ -459,6 +460,34 @@ def custom_port(spawner, user_options):
     return port
 
 
+async def pre_poll_hook(spawner):
+    custom_config = get_custom_config()
+    system = spawner.user_options.get("system", "")
+    backend_service = (
+        custom_config.get("systems", {}).get(system, {}).get("backendService", "None")
+    )
+    if (
+        custom_config.get("backendServices", {})
+        .get(backend_service, {})
+        .get("pollAccessTokenRequired", False)
+    ):
+        # If we know we need need the access token to perform a valid poll request,
+        # let's avoid the call to the Outpost if it's not helpful anyway.
+        user = spawner.user
+        auth_state = await user.get_auth_state()
+        if not auth_state:
+            return False
+        threshold = 2 * user.authenticator.true_auth_refresh_age
+        now = time.time()
+        rest_time = int(auth_state.get("exp", now)) - now
+        if not auth_state.get("access_token", None) or threshold > rest_time:
+            spawner.log.info(
+                f"{spawner._log_name} - Do not call Outpost for polling (access_token available: {bool(auth_state.get('acccess_token'))}, rest_time: {rest_time}"
+            )
+            return False
+    return True
+
+
 async def pre_spawn_hook(spawner):
     custom_config = get_custom_config()
     service, version = spawner.user_options.get("profile", "").split("/")

pyproject.toml

@@ -31,6 +31,7 @@ dependencies = [
     "urllib3",
     "pyyaml",
     "jupyterhub-unicorespawner",
+    "jupyterhub-forwardbasespawner>=1.0.8",
     "jupyterhub-outpostspawner",
     "oauthenticator"
 ]