From 913980a02761ff5e6d318ece722228526da8da80 Mon Sep 17 00:00:00 2001
From: Micke Nordin <kano@sunet.se>
Date: Sat, 13 Jan 2024 15:52:01 +0100
Subject: [PATCH] Add KeyError check

---
 jupyter/base/values/values.yaml | 21 ++++++++++++---------
 1 file changed, 12 insertions(+), 9 deletions(-)

diff --git a/jupyter/base/values/values.yaml b/jupyter/base/values/values.yaml
index 3f98e1e..e7352b4 100644
--- a/jupyter/base/values/values.yaml
+++ b/jupyter/base/values/values.yaml
@@ -30,16 +30,19 @@ hub:
           spawner.environment['NEXTCLOUD_ACCESS_TOKEN'] = access_token
 
         async def refresh_user(self, user, handler=None):
-          access_token = self.user_dict[user.name]['access_token']
-          refresh_token = self.user_dict[user.name]['refresh_token']
-          token_response = self.user_dict[user.name]['token_response']
-          now = time.time()
-          expires = self.user_dict[user.name]['token_expires']
-          if now >= expires:
-            print(f'Time is: {now}, token expired: {expires}')
+          try:
+            access_token = self.user_dict[user.name]['access_token']
+            refresh_token = self.user_dict[user.name]['refresh_token']
+            token_response = self.user_dict[user.name]['token_response']
+            now = time.time()
+            expires = self.user_dict[user.name]['token_expires']
+            if now >= expires:
+              print(f'Time is: {now}, token expired: {expires}')
+              return False
+            print(f'Time is: {now}, token expires: {expires}')
+            return True
+          except KeyError:
             return False
-          print(f'Time is: {now}, token expires: {expires}')
-          return True
 
       c.JupyterHub.authenticator_class = NextcloudOAuthenticator
       c.NextcloudOAuthenticator.client_id = os.environ['NEXTCLOUD_CLIENT_ID']