from typing import Any, Dict, List from baserow.contrib.database.fields.registries import field_type_registry from baserow.contrib.database.fields.utils.duration import D_H_M def construct_all_possible_field_kwargs( table, link_table, decimal_link_table, file_link_table ) -> Dict[str, List[Dict[str, Any]]]: """ Some baserow field types have multiple different 'modes' which result in different database columns and modes of operation being created. This function creates a dictionary of field type to a list of kwarg dicts, one for each interesting possible 'subtype' of the field. """ all_interesting_field_kwargs = { "text": [{"name": "text", "primary": True}], "long_text": [{"name": "long_text"}], "url": [{"name": "url"}], "email": [{"name": "email"}], "number": [ { "name": "negative_int", "number_negative": True, "number_decimal_places": 0, }, { "name": "positive_int", "number_negative": False, "number_decimal_places": 0, }, { "name": "negative_decimal", "number_negative": "True", "number_decimal_places": 1, }, { "name": "positive_decimal", "number_negative": False, "number_decimal_places": 1, }, ], "rating": [ {"name": "rating", "max_value": 5, "color": "blue", "style": "star"} ], "boolean": [{"name": "boolean"}], "date": [ {"name": "datetime_us", "date_include_time": True, "date_format": "US"}, {"name": "date_us", "date_include_time": False, "date_format": "US"}, { "name": "datetime_eu", "date_include_time": True, "date_format": "EU", }, {"name": "date_eu", "date_include_time": False, "date_format": "EU"}, { "name": "datetime_eu_tzone_visible", "date_include_time": True, "date_format": "EU", "date_force_timezone": "Europe/Amsterdam", "date_show_tzinfo": True, }, { "name": "datetime_eu_tzone_hidden", "date_include_time": True, "date_format": "EU", "date_force_timezone": "Europe/Amsterdam", "date_show_tzinfo": False, }, ], "last_modified": [ { "name": "last_modified_datetime_us", "date_include_time": True, "date_format": "US", }, { "name": "last_modified_date_us", "date_include_time": False, "date_format": "US", }, { "name": "last_modified_datetime_eu", "date_include_time": True, "date_format": "EU", }, { "name": "last_modified_date_eu", "date_include_time": False, "date_format": "EU", }, { "name": "last_modified_datetime_eu_tzone", "date_include_time": True, "date_format": "EU", "date_force_timezone": "Europe/Amsterdam", "date_show_tzinfo": True, }, ], "created_on": [ { "name": "created_on_datetime_us", "date_include_time": True, "date_format": "US", }, { "name": "created_on_date_us", "date_include_time": False, "date_format": "US", }, { "name": "created_on_datetime_eu", "date_include_time": True, "date_format": "EU", }, { "name": "created_on_date_eu", "date_include_time": False, "date_format": "EU", }, { "name": "created_on_datetime_eu_tzone", "date_include_time": True, "date_format": "EU", "date_force_timezone": "Europe/Amsterdam", "date_show_tzinfo": True, }, ], "last_modified_by": [ { "name": "last_modified_by", } ], "created_by": [ { "name": "created_by", } ], "duration": [ {"name": "duration_hm", "duration_format": "h:mm"}, {"name": "duration_hms", "duration_format": "h:mm:ss"}, {"name": "duration_hms_s", "duration_format": "h:mm:ss.s"}, {"name": "duration_hms_ss", "duration_format": "h:mm:ss.ss"}, {"name": "duration_hms_sss", "duration_format": "h:mm:ss.sss"}, {"name": "duration_dh", "duration_format": "d h"}, {"name": "duration_dhm", "duration_format": "d h:mm"}, {"name": "duration_dhms", "duration_format": "d h:mm:ss"}, ], "link_row": [ {"name": "link_row", "link_row_table": link_table}, { "name": "self_link_row", "link_row_table": table, "has_related_field": False, }, { "name": "link_row_without_related", "link_row_table": link_table, "has_related_field": False, }, {"name": "decimal_link_row", "link_row_table": decimal_link_table}, {"name": "file_link_row", "link_row_table": file_link_table}, ], "file": [{"name": "file"}], "single_select": [ { "name": "single_select", "select_options": [ {"id": 0, "value": "A", "color": "red"}, {"id": 1, "value": "B", "color": "blue"}, ], } ], "multiple_select": [ { "name": "multiple_select", "select_options": [ {"id": 2, "value": "C", "color": "orange"}, {"id": 3, "value": "D", "color": "yellow"}, {"id": 4, "value": "E", "color": "green"}, ], } ], "multiple_collaborators": [ { "name": "multiple_collaborators", "notify_user_when_added": False, } ], "phone_number": [{"name": "phone_number"}], "formula": [ # Make one for each Baserow formula type! {"name": "formula_text", "formula": "CONCAT('test ', UPPER('formula'))"}, {"name": "formula_int", "formula": "1"}, {"name": "formula_bool", "formula": "true"}, {"name": "formula_decimal", "formula": "100/3"}, { "name": "formula_dateinterval", "formula": "date_interval('1 day')", "duration_format": D_H_M, }, {"name": "formula_date", "formula": "todate('20200101', 'YYYYMMDD')"}, {"name": "formula_singleselect", "formula": "field('single_select')"}, {"name": "formula_email", "formula": "field('email')"}, { "name": "formula_link_with_label", "formula": "button('https://google.com', 'label')", }, {"name": "formula_link_url_only", "formula": "link('https://google.com')"}, {"name": "formula_multipleselect", "formula": "field('multiple_select')"}, ], "count": [ { "name": "count", "through_field_name": "link_row", } ], "rollup": [ { "name": "rollup", "through_field_name": "decimal_link_row", "target_field_name": "decimal_field", "rollup_function": "sum", "number_decimal_places": 3, } ], "lookup": [ { "name": "lookup", "through_field_name": "link_row", "target_field_name": "text_field", } ], "uuid": [{"name": "uuid"}], "autonumber": [{"name": "autonumber"}], "password": [{"name": "password"}], "ai": [ { "name": "ai", "ai_generative_ai_type": "test_generative_ai", "ai_generative_ai_model": "test_1", "ai_prompt": "Who are you?", } ], } # If you have added a new field please add an entry into the dict above with any # test worthy combinations of kwargs # nosec ignore as this code is test/dev cli tool only, no matter if this assert # does not get run in the cli tools. assert set(field_type_registry.get_types()) == set( # nosec all_interesting_field_kwargs.keys() ), "Please add the new field type to the testing dictionary of interesting kwargs" return all_interesting_field_kwargs