Preconfig conditional format

mitosheet/mitosheet/mito_backend.py

@@ -29,7 +29,7 @@
 from mitosheet.steps_manager import StepsManager
 from mitosheet.telemetry.telemetry_utils import (log, log_event_processed,
                                                  telemetry_turned_on)
-from mitosheet.types import CodeOptions, MitoTheme, ParamMetadata
+from mitosheet.types import CodeOptions, ConditionalFormat, MitoTheme, ParamMetadata
 from mitosheet.updates.replay_analysis import REPLAY_ANALYSIS_UPDATE
 from mitosheet.user.create import try_create_user_json_file
 from mitosheet.user.db import USER_JSON_PATH, get_user_field
@@ -57,6 +57,7 @@ def __init__(
             user_defined_importers: Optional[List[Callable]]=None,
             user_defined_editors: Optional[List[Callable]]=None,
             code_options: Optional[CodeOptions]=None,
+            partial_conditional_formats: Optional[List[Dict[str, Any]]]=None,
             theme: Optional[MitoTheme]=None,
         ):
         """
@@ -92,6 +93,16 @@ def __init__(
 
             if not os.path.exists(import_folder):
                 raise ValueError(f"Import folder {import_folder} does not exist. Please change the file path or create the folder.")
+
+
+        conditional_formats = [{
+            'format_uuid': 'preset_conditional_format',
+            'columnIDs': partial_conditional_formats['columnHeaders'],
+            'filters': [partial_conditional_formats['filters']],
+            'invalidFilterColumnIDs': [],
+            'color': partial_conditional_formats['color'],
+            'backgroundColor': partial_conditional_formats['backgroundColor']
+        }]
 
         # Set up the state container to hold private widget state
         self.steps_manager = StepsManager(
@@ -104,6 +115,7 @@ def __init__(
             user_defined_importers=all_custom_importers,
             user_defined_editors=user_defined_editors,
             code_options=code_options,
+            conditional_formats=conditional_formats,
             theme=theme
         )
 

mitosheet/mitosheet/pro/conditional_formatting_utils.py

@@ -18,6 +18,7 @@ def get_conditonal_formatting_result(
     formatted_result: ConditionalFormattingCellResults = dict()
 
     for conditional_format in conditional_formatting_rules:
+        print(conditional_format)
         try:
 
             format_uuid = conditional_format["format_uuid"]
@@ -52,7 +53,9 @@ def get_conditonal_formatting_result(
                     # are sent to the frontend
                     json_index = json.dumps(index, cls=NpEncoder)
                     formatted_result[column_id][json_index] = {'backgroundColor': backgroundColor, 'color': color}
+
         except Exception as e:
+            print("IN THIS EXCEPTION")
             if format_uuid not in invalid_conditional_formats:
                 invalid_conditional_formats[format_uuid] = []
             invalid_conditional_formats[format_uuid].append(column_id)

mitosheet/mitosheet/step_performers/filter.py

@@ -260,5 +260,5 @@ def check_filters_contain_condition_that_needs_full_df(filters: List[Union[Filte
             filter_group: FilterGroup = filter_or_group #type: ignore
             if check_filters_contain_condition_that_needs_full_df(filter_group["filters"]):
                 return True
-        
+
     return False

mitosheet/mitosheet/steps_manager.py

@@ -17,6 +17,7 @@
 from mitosheet.enterprise.mito_config import MitoConfig
 from mitosheet.enterprise.telemetry.mito_log_uploader import MitoLogUploader
 from mitosheet.experiments.experiment_utils import get_current_experiment
+from mitosheet.pro.conditional_formatting_utils import get_conditonal_formatting_result
 from mitosheet.step_performers.column_steps.set_column_formula import get_user_defined_sheet_function_objects
 from mitosheet.step_performers.import_steps.dataframe_import import DataframeImportStepPerformer
 from mitosheet.step_performers.import_steps.excel_range_import import ExcelRangeImportStepPerformer
@@ -35,7 +36,7 @@
     SnowflakeImportStepPerformer
 from mitosheet.transpiler.transpile import transpile
 from mitosheet.transpiler.transpile_utils import get_default_code_options
-from mitosheet.types import CodeOptions, MitoTheme, ParamMetadata
+from mitosheet.types import CodeOptions, ConditionalFormat, MitoTheme, ParamMetadata
 from mitosheet.updates import UPDATES
 from mitosheet.user.utils import is_enterprise, is_running_test
 from mitosheet.utils import NpEncoder, dfs_to_array_for_json, get_new_id, is_default_df_names
@@ -187,6 +188,7 @@ def __init__(
             user_defined_importers: Optional[List[Callable]]=None,
             user_defined_editors: Optional[List[Callable]]=None,
             code_options: Optional[CodeOptions]=None,
+            conditional_formats: Optional[List[ConditionalFormat]]=None,
             theme: Optional[MitoTheme]=None,
         ):
         """
@@ -265,12 +267,23 @@ def __init__(
                     df_names=df_names,
                     user_defined_functions=self.user_defined_functions, 
                     user_defined_importers=self.user_defined_importers,
-                    user_defined_editors=self.user_defined_editors
+                    user_defined_editors=self.user_defined_editors,
                 ), 
                 {}
             )
         ]
 
+        df_formats = {
+            'columns': {},
+            'headers': {},
+            'rows': {'even': {}, 'odd': {}},
+            'border': {},
+            'conditional_formats': conditional_formats
+        }
+
+        self.steps_including_skipped[0].df_formats[0] = df_formats
+        self.steps_including_skipped[0].post_state.df_formats[0] = df_formats
+
         """
         To help with redo, we store a list of a list of the steps that 
         existed in the step manager before the user clicked undo or reset,

mitosheet/mitosheet/streamlit/v1/spreadsheet.py

@@ -13,7 +13,7 @@
 
 from mitosheet.mito_backend import MitoBackend
 from mitosheet.selection_utils import get_selected_element
-from mitosheet.types import CodeOptions, ParamMetadata, ParamType
+from mitosheet.types import CodeOptions, ConditionalFormat, ParamMetadata, ParamType
 from mitosheet.utils import get_new_id
 
 CURRENT_MITO_ANALYSIS_VERSION = 1
@@ -118,12 +118,14 @@ def __init__(
             self,
             code: str,
             code_options: Optional[CodeOptions],
+            conditional_formats: Optional[List[Dict[str, Any]]],
             fully_parameterized_function: str,
             param_metadata: List[ParamMetadata],
             mito_analysis_version: int=CURRENT_MITO_ANALYSIS_VERSION
         ):
         self.__code = code
         self.__code_options = code_options
+        self.__conditional_formats = conditional_formats
         self.__fully_parameterized_function = fully_parameterized_function
         self.__param_metadata = param_metadata
         self.mito_analysis_version = mito_analysis_version
@@ -143,6 +145,7 @@ def to_json(self) -> str:
         return json.dumps({
             'code': self.__code,
             'code_options': self.__code_options,
+            'conditional_formats': self.__conditional_formats,
             'fully_parameterized_function': self.__fully_parameterized_function,
             'param_metadata': self.__param_metadata,
             'mito_analysis_version': self.mito_analysis_version
@@ -158,6 +161,7 @@ def from_json(json_str: str) -> 'RunnableAnalysis':
         return RunnableAnalysis(
             json_dict['code'],
             json_dict['code_options'],
+            json_dict['conditional_formats'],
             json_dict['fully_parameterized_function'],
             json_dict['param_metadata'],
             mito_analysis_version=json_dict['mito_analysis_version']
@@ -269,6 +273,7 @@ def _get_mito_backend(
             _editors: Optional[List[Callable]]=None, 
             _sheet_functions: Optional[List[Callable]]=None, 
             _code_options: Optional[CodeOptions]=None,
+            _conditional_formats: Optional[Dict[str, Any]]=None,
             import_folder: Optional[str]=None,
             df_names: Optional[List[str]]=None,
             session_id: Optional[str]=None,
@@ -280,6 +285,7 @@ def _get_mito_backend(
             import_folder=import_folder,
             user_defined_importers=_importers, user_defined_functions=_sheet_functions, user_defined_editors=_editors,
             code_options=_code_options,
+            partial_conditional_formats=_conditional_formats,
         )
 
         # Make a send function that stores the responses in a list
@@ -320,6 +326,7 @@ def spreadsheet( # type: ignore
             df_names: Optional[List[str]]=None,
             import_folder: Optional[str]=None,
             code_options: Optional[CodeOptions]=None,
+            conditional_formats: Optional[Dict[str, Any]]=None,
             return_type: str='default',
             height: Optional[str]=None,
             key=None
@@ -366,6 +373,7 @@ def spreadsheet( # type: ignore
             _importers=importers, 
             _editors=editors,
             _code_options=code_options,
+            _conditional_formats=conditional_formats,
             import_folder=import_folder,
             session_id=session_id,
             df_names=df_names, 

mitosheet/target-streamlit-app.py

@@ -0,0 +1,22 @@
+import streamlit as st
+from mitosheet.streamlit.v1 import spreadsheet
+import pandas as pd
+
+st.set_page_config(layout="wide")
+
+st.title('SKU analysis app')
+
+df = pd.read_csv('./SKU.csv')
+new_dfs, code = spreadsheet(
+    df, 
+    conditional_formats={
+        'columnHeaders': ['quantity'],
+        'filters': {
+            'condition': 'greater_than_or_equal',
+            'value': 5
+        },
+        'backgroundColor': '#ffc9bb',
+        'color': '#c61a09'
+    },
+    height='900px'
+)