mitosheet: fix bug with conditional format string conditions

Author: naterush

mitosheet/mitosheet/pro/conditional_formatting_utils.py

@@ -49,8 +49,11 @@ def get_conditonal_formatting_result(
 
                 for index in applied_indexes:
                     # We need to make this index valid json, and do so in a way that is consistent with how indexes
-                    # are sent to the frontend
-                    json_index = json.dumps(index, cls=NpEncoder)
+                    # are sent to the frontend. However, if the index is a string, we don't want to add quotes around it
+                    if isinstance(index, str):
+                        json_index = index
+                    else:
+                        json_index = json.dumps(index, cls=NpEncoder)
                     formatted_result[column_id][json_index] = {'backgroundColor': backgroundColor, 'color': color}
 
         except Exception as e:

mitosheet/mitosheet/tests/test_column_definitions.py mitosheet/mitosheet/tests/streamlit/test_column_definitions.py

@@ -1,3 +1,4 @@
+import json
 import pytest
 from mitosheet.mito_backend import MitoBackend, get_mito_backend
 import pandas as pd
@@ -233,7 +234,28 @@ def test_create_backend_with_column_definitions_does_not_error_with_mismatch_con
     assert df_formats[0]['conditional_formats'][0]['invalidFilterColumnIDs'] == []
 
 
+def test_column_definitions_with_string_indexes_conditional_format_works():
+    column_definitions= [
+        [
+            {
+                'columns': ['A'],
+                'conditional_formats': [{
+                    'filters': [{'condition': 'number_exactly', 'value': 2}], 
+                    'font_color': '#c30010', 
+                    'background_color': '#ffcbd1' 
+                }] 
+            }
+        ]
+    ]
+
 
+    mito_backend = get_mito_backend(pd.DataFrame({'A': [1, 2, 3]}, index=['S1', 'S2', 'S3']), column_definitions=column_definitions)
+    sheet_json = mito_backend.get_shared_state_variables()['sheet_data_json']
+    sheet_data = json.loads(sheet_json)[0]
+    conditional_formatting_result = sheet_data['conditionalFormattingResult']
+    assert len(conditional_formatting_result['results']) == 1
+    assert len(conditional_formatting_result['results']['A']) == 1
+    assert 'S2' in conditional_formatting_result['results']['A']