Add debug logs for testing

uc-cdis · Jan 3, 2025 · 8dcd619 · 8dcd619
1 parent 4a4bc15
commit 8dcd619
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/wts/auth_plugins/k8s.py b/wts/auth_plugins/k8s.py
@@ -10,6 +10,7 @@
 
 
 def get_username_from_ip(ip):
+    flask.current_app.logger.info("Getting username from IP {}".format(ip))
     # Fail if we can't load kubernetes config...
     try:
         kubernetes.config.load_incluster_config()
@@ -24,14 +25,25 @@ def get_username_from_ip(ip):
             pod.metadata.annotations
             and POD_USERNAME_ANNOTATION in pod.metadata.annotations
         ):
+            flask.current_app.logger.info(
+                "Found username {} for IP {}".format(
+                    pod.metadata.annotations[POD_USERNAME_ANNOTATION], ip
+                )
+            )
             return pod.metadata.annotations[POD_USERNAME_ANNOTATION]
         elif (
             pod.metadata.annotations
             and JUPYTER_POD_ANNOTATION in pod.metadata.annotations
         ):
+            flask.current_app.logger.info(
+                "Found username {} for IP {}".format(
+                    pod.metadata.annotations[JUPYTER_POD_ANNOTATION], ip
+                )
+            )
             return pod.metadata.annotations[JUPYTER_POD_ANNOTATION]
 
     # No matching pod found
+    flask.current_app.logger.info("No username found for IP {}".format(ip))
     return None