Add theme selector

app/queries/crud.py

@@ -87,7 +87,7 @@ def get_color_palettes():
         return None
 
 
-def get_color_palette():
+def get_color_palette() -> int:
     user_id = int(st.session_state.user_id)
     try:
         return int(connection().query('SELECT color_palette_id FROM users WHERE id = :id''', params={'id': user_id})['color_palette_id'][0])
@@ -123,3 +123,26 @@ def get_colors():
     except Exception as e:
         logger.error(e)
         return None
+
+
+def set_theme(theme:str):
+    user_id = int(st.session_state.user_id)
+    logger.info("Changing theme for user %d to %s", user_id, theme)
+    with connection().session as session:
+        try:
+            query = text('UPDATE users SET theme = :theme WHERE id = :user')
+            session.execute(query, {
+                'theme': theme,
+                'user': user_id
+            })
+        except Exception as e:
+            logger.error(e)
+            session.rollback()
+
+def get_theme() -> str:
+    user_id = int(st.session_state.user_id)
+    try:
+        return connection().query('SELECT theme FROM users u WHERE u.id = :id', params={'id': user_id})['theme'][0]
+    except Exception as e:
+        logger.error(e)
+        return None

app/queries/user.py

@@ -43,25 +43,27 @@ def create_user(email, name, oidc_user_id):
 
 
 def set_user_in_session(user: UserInfoProxy):
-    email = user.email
-    user_id = user.sub
-    if hasattr(user, 'name'):
-        name = user.name
-    else:
-        name = None
 
-    user_entity = find_user_by_oidc_id(user_id)
+    email = user.email if hasattr(user, "email") else None
+    user_id = user.sub if hasattr(user, "sub") else None
+    name = user.name if hasattr(user, "name") else None
+
+    user_entity = find_user_by_oidc_id(user_id) if user_id else st.dataframe()
     if user_entity.empty:
-        user_entity = find_user_by_email(email)
+        user_entity = find_user_by_email(email) if email else st.dataframe()
     if user_entity.empty:
         user_entity = find_default_user()
-        if user_entity.empty:
+        if user_entity.empty and email and name and user_id:
             user_entity = create_user(email, name, user_id)
-        else:
+        elif name:
             update_default_user(email, name, user_id)
             user_entity = find_user_by_oidc_id(user_id)
 
-    st.session_state.user_id = user_entity["id"][0]
-    st.session_state.user_name = user_entity["name"][0]
-    st.session_state.user_email = user_entity["email"][0]
-    st.session_state.user_external_id = user_entity["oidc_user_id"][0]
+    if not user_entity.empty:
+        st.session_state.user_id = user_entity["id"][0]
+        st.session_state.user_name = user_entity["name"][0]
+        st.session_state.user_email = user_entity["email"][0]
+        st.session_state.user_external_id = user_entity["oidc_user_id"][0]
+        st.session_state.current_theme = user_entity["theme"][0]
+    else:
+        logger.warn("No active user found!")