Add user color settings

app/queries/crud.py

@@ -79,9 +79,47 @@ def get_counter(counter_id:int):
         return None
 
 
-def get_colors(palette_id:int):
+def get_color_palettes():
     try:
-        return connection().query('''SELECT color1,color2,color3,color4,color5 FROM color_palettes WHERE id = :id''', params={'id': palette_id})
+        return connection().query('SELECT * FROM color_palettes''')
+    except Exception as e:
+        logger.error(e)
+        return None
+
+
+def get_color_palette():
+    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])
+    except Exception as e:
+        logger.error(e)
+        return None
+
+
+def set_color_palette(palette_id:int):
+    user_id = int(st.session_state.user_id)
+    logger.info("Changing palette for user %d to %d", user_id, palette_id)
+    with connection().session as session:
+        try:
+            query = text('UPDATE users SET color_palette_id = :palette WHERE id = :user')
+            session.execute(query, {
+                'palette': palette_id,
+                'user': user_id
+            })
+        except Exception as e:
+            logger.error(e)
+            session.rollback()
+
+
+def get_colors():
+    user_id = int(st.session_state.user_id)
+    try:
+        return connection().query('''
+              SELECT color1,color2,color3,color4,color5 
+              FROM users u
+              LEFT JOIN color_palettes p ON p.id = u.color_palette_id
+              WHERE u.id = :id
+        ''', params={'id': user_id})
     except Exception as e:
         logger.error(e)
         return None