Fixed unit tests
This commit is contained in:
@@ -78,7 +78,7 @@ def index(session):
|
||||
layout.left_drawer.add(btn_popup, "Test")
|
||||
|
||||
# data grids
|
||||
dgs_manager = DataGridsManager(session_instance)
|
||||
dgs_manager = DataGridsManager(session_instance, save_state=True)
|
||||
layout.left_drawer.add_group("Documents", Div("Documents",
|
||||
dgs_manager.mk_main_icons(),
|
||||
cls="mf-layout-group flex gap-3"))
|
||||
|
||||
@@ -224,6 +224,10 @@ class DataGrid(MultipleInstance):
|
||||
self._columns = None
|
||||
self.commands = Commands(self)
|
||||
|
||||
# reset
|
||||
self._state.selection.selected = None
|
||||
self._state.selection.last_selected = None
|
||||
|
||||
# Obtain DataService from DataServicesManager (no parent hierarchy)
|
||||
data_services_manager = InstancesManager.get_by_type(self._session, DataServicesManager)
|
||||
data_service_id = self.get_data_service_id_from_data_grid_id(self._id)
|
||||
@@ -264,7 +268,7 @@ class DataGrid(MultipleInstance):
|
||||
# self._columns_manager.bind_command("SaveColumnDetails", self.commands.on_column_changed())
|
||||
|
||||
if self._settings.enable_formatting:
|
||||
provider = InstancesManager.get_by_type(self._session, DatagridMetadataProvider, None)
|
||||
provider = DatagridMetadataProvider(self._parent)
|
||||
completion_engine = FormattingCompletionEngine(provider, self.get_table_name())
|
||||
editor_conf = DslEditorConf(engine_id=completion_engine.get_id())
|
||||
dsl = FormattingDSL()
|
||||
@@ -305,6 +309,10 @@ class DataGrid(MultipleInstance):
|
||||
def _fast_access(self):
|
||||
return self._data_service.get_store().ns_fast_access
|
||||
|
||||
@property
|
||||
def _row_data(self):
|
||||
return self._data_service.get_store().ns_row_data
|
||||
|
||||
def _apply_sort(self, df):
|
||||
if df is None:
|
||||
return None
|
||||
@@ -413,9 +421,9 @@ class DataGrid(MultipleInstance):
|
||||
if self._state.table_format:
|
||||
return self._state.table_format
|
||||
|
||||
# Get global tables formatting from DatagridMetadataProvider
|
||||
provider = InstancesManager.get_by_type(self._session, DatagridMetadataProvider, None)
|
||||
return provider.all_tables_formats if provider is not None else []
|
||||
# Get global tables formatting from DataGridsManager
|
||||
dgm = self.get_parent()
|
||||
return dgm.get_state().all_tables_formats if dgm is not None else []
|
||||
|
||||
def _init_columns(self):
|
||||
# Populate UI state from DataService columns when creating a new grid
|
||||
@@ -494,22 +502,6 @@ class DataGrid(MultipleInstance):
|
||||
|
||||
self._state.save()
|
||||
|
||||
def handle_reorder_columns(self, order: list[str]):
|
||||
"""Reorder columns based on a full ordered list of column IDs.
|
||||
|
||||
Args:
|
||||
order: List of col_id strings in the desired display order.
|
||||
Columns not present in order are appended at the end.
|
||||
"""
|
||||
logger.debug(f"handle_reorder_columns: {order=}")
|
||||
columns_by_id = {c.col_id: c for c in self._state.columns}
|
||||
new_order = [columns_by_id[col_id] for col_id in order if col_id in columns_by_id]
|
||||
remaining = [c for c in self._state.columns if c.col_id not in order]
|
||||
self._state.columns = new_order + remaining
|
||||
self._init_columns()
|
||||
self._state.save()
|
||||
return self.render_partial("table")
|
||||
|
||||
def calculate_optimal_column_width(self, col_id: str) -> int:
|
||||
"""
|
||||
Calculate optimal width for a column based on content.
|
||||
@@ -625,6 +617,42 @@ class DataGrid(MultipleInstance):
|
||||
self._state.save()
|
||||
return self.render_partial()
|
||||
|
||||
def handle_columns_reorder(self, order: list[str]):
|
||||
"""Reorder columns based on a full ordered list of column IDs.
|
||||
|
||||
Args:
|
||||
order: List of col_id strings in the desired display order.
|
||||
Columns not present in order are appended at the end.
|
||||
"""
|
||||
logger.debug(f"handle_reorder_columns: {order=}")
|
||||
columns_by_id = {c.col_id: c for c in self._state.columns}
|
||||
new_order = [columns_by_id[col_id] for col_id in order if col_id in columns_by_id]
|
||||
remaining = [c for c in self._state.columns if c.col_id not in order]
|
||||
self._state.columns = new_order + remaining
|
||||
self._init_columns()
|
||||
self._state.save()
|
||||
return self.render_partial("table")
|
||||
|
||||
def handle_columns_updates(self, updates: dict[str, dict]):
|
||||
logger.debug(f"handle_columns_update: {updates=}")
|
||||
|
||||
def _update(col_def):
|
||||
need_saving = False
|
||||
for key, value in update.items():
|
||||
if hasattr(col_def, key):
|
||||
setattr(col_def, key, value)
|
||||
need_saving = True
|
||||
return need_saving
|
||||
|
||||
for col_id, update in updates.items():
|
||||
column_state = [col_def for col_def in self._columns if col_def.col_id == col_id][0]
|
||||
if _update(column_state.get_col_ui_state()):
|
||||
self._state.save()
|
||||
if _update(column_state.get_col_def()):
|
||||
self._data_service.save_state()
|
||||
|
||||
return self.render_partial("table")
|
||||
|
||||
def handle_toggle_columns_manager(self):
|
||||
logger.debug(f"toggle_columns_manager")
|
||||
self._panel.set_title(side="right", title="Columns")
|
||||
@@ -803,7 +831,7 @@ class DataGrid(MultipleInstance):
|
||||
formatted_value = None
|
||||
rules = self._get_format_rules(col_pos, row_index, col_def)
|
||||
if rules:
|
||||
row_data = self._df_store.ns_row_data[row_index] if row_index < len(self._df_store.ns_row_data) else None
|
||||
row_data = self._row_data[row_index] if row_index < len(self._row_data) else None
|
||||
style, formatted_value = self._formatting_engine.apply_format(rules, value, row_data)
|
||||
|
||||
# Use formatted value or convert to string
|
||||
|
||||
@@ -23,6 +23,14 @@ class Commands(BaseCommands):
|
||||
self._owner,
|
||||
self._owner.handle_on_reorder
|
||||
).htmx(target=f"#{self._id}")
|
||||
|
||||
def toggle_column_visibility(self, col_id: str):
|
||||
return Command("ToggleColumnVisibility",
|
||||
"Toggle column visibility",
|
||||
self._owner,
|
||||
self._owner.handle_toggle_column_visibility,
|
||||
kwargs={"col_id": col_id}
|
||||
).htmx(target=f"#{self._id}")
|
||||
|
||||
|
||||
class DataGridColumnsList(MultipleInstance):
|
||||
@@ -43,7 +51,14 @@ class DataGridColumnsList(MultipleInstance):
|
||||
|
||||
def handle_on_reorder(self, order: list):
|
||||
logger.debug(f"on_reorder {order=}")
|
||||
ret = self._parent.handle_reorder_columns(order)
|
||||
ret = self._parent.handle_columns_reorder(order)
|
||||
return self.render(), ret
|
||||
|
||||
def handle_toggle_column_visibility(self, col_id):
|
||||
logger.debug(f"handle_toggle_column_visibility {col_id=}")
|
||||
col_def = [c for c in self.columns if c.col_id == col_id][0]
|
||||
updates = {col_id: {"visible": not col_def.visible}}
|
||||
ret = self._parent.handle_columns_updates(updates)
|
||||
return self.render(), ret
|
||||
|
||||
def mk_column_label(self, col_def: DataGridColumnState):
|
||||
@@ -51,7 +66,7 @@ class DataGridColumnsList(MultipleInstance):
|
||||
mk.icon(grip_horizontal, cls="mf-drag-handle cursor-grab mr-1 opacity-40"),
|
||||
mk.mk(
|
||||
Input(type="checkbox", cls="checkbox checkbox-sm", checked=col_def.visible),
|
||||
# command=self.commands.toggle_column(col_def.col_id)
|
||||
command=self.commands.toggle_column_visibility(col_def.col_id)
|
||||
),
|
||||
mk.mk(
|
||||
Div(
|
||||
|
||||
@@ -127,7 +127,7 @@ class DataGridFormattingEditor(DslEditor):
|
||||
from myfasthtml.controls.DataGridsManager import DataGridsManager
|
||||
manager = InstancesManager.get_by_type(self._session, DataGridsManager)
|
||||
if manager:
|
||||
manager.all_tables_formats = tables_rules
|
||||
manager.get_state().all_tables_formats = tables_rules
|
||||
|
||||
# Step 6: Update state atomically
|
||||
self._parent.get_state().update(state)
|
||||
|
||||
@@ -15,6 +15,7 @@ from myfasthtml.core.DataGridsRegistry import DataGridsRegistry
|
||||
from myfasthtml.core.commands import Command
|
||||
from myfasthtml.core.data.DataServicesManager import DataServicesManager
|
||||
from myfasthtml.core.dbmanager import DbObject
|
||||
from myfasthtml.core.formatting.dataclasses import FormatRule
|
||||
from myfasthtml.core.instances import InstancesManager, SingleInstance
|
||||
from myfasthtml.icons.fluent_p1 import table_add20_regular
|
||||
from myfasthtml.icons.fluent_p3 import folder_open20_regular
|
||||
@@ -31,10 +32,11 @@ class DocumentDefinition:
|
||||
|
||||
|
||||
class DataGridsState(DbObject):
|
||||
def __init__(self, owner, name=None):
|
||||
super().__init__(owner, name=name)
|
||||
def __init__(self, owner, save_state, name=None):
|
||||
super().__init__(owner, save_state=save_state, name=name)
|
||||
with self.initializing():
|
||||
self.elements: list[DocumentDefinition] = []
|
||||
self.all_tables_formats: list[FormatRule] = []
|
||||
|
||||
|
||||
class Commands(BaseCommands):
|
||||
@@ -87,12 +89,12 @@ class DataGridsManager(SingleInstance):
|
||||
by DataServicesManager and DataService.
|
||||
"""
|
||||
|
||||
def __init__(self, parent, _id=None):
|
||||
def __init__(self, parent, _id=None, save_state=None):
|
||||
if not getattr(self, "_is_new_instance", False):
|
||||
return
|
||||
super().__init__(parent, _id=_id)
|
||||
self.commands = Commands(self)
|
||||
self._state = DataGridsState(self)
|
||||
self._state = DataGridsState(self, save_state)
|
||||
self._tree = self._mk_tree()
|
||||
self._tree.bind_command("SelectNode", self.commands.show_document())
|
||||
self._tree.bind_command("DeleteNode", self.commands.delete_grid(), when="before")
|
||||
@@ -102,6 +104,9 @@ class DataGridsManager(SingleInstance):
|
||||
# Data layer — session-scoped singletons
|
||||
self._data_services_manager = DataServicesManager(self._parent)
|
||||
|
||||
def get_state(self):
|
||||
return self._state
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# Grid lifecycle
|
||||
# ------------------------------------------------------------------
|
||||
|
||||
@@ -78,6 +78,12 @@ class DataGridColumnState:
|
||||
self._col_def = col_def
|
||||
self._col_ui_state = col_ui_state
|
||||
|
||||
def get_col_def(self):
|
||||
return self._col_def
|
||||
|
||||
def get_col_ui_state(self):
|
||||
return self._col_ui_state
|
||||
|
||||
@property
|
||||
def col_id(self):
|
||||
return self._col_def.col_id
|
||||
|
||||
@@ -84,6 +84,9 @@ class DataService(MultipleInstance):
|
||||
self._store = DataStore(self, save_state=save_state)
|
||||
self._init_store()
|
||||
|
||||
def save_state(self):
|
||||
self._state.save()
|
||||
|
||||
@property
|
||||
def columns(self) -> list[ColumnDefinition]:
|
||||
"""Return the list of column definitions."""
|
||||
|
||||
@@ -120,3 +120,10 @@ class DataServicesManager(SingleInstance):
|
||||
return service.get_store()
|
||||
logger.warning(f"DataServicesManager: table '{table_name}' not found")
|
||||
return None
|
||||
|
||||
def clear(self):
|
||||
"""
|
||||
For test purposes only.
|
||||
:return:
|
||||
"""
|
||||
self._services.clear()
|
||||
@@ -3,6 +3,7 @@ Completion engine for the formatting DSL.
|
||||
|
||||
Implements the BaseCompletionEngine for DataGrid formatting rules.
|
||||
"""
|
||||
import logging
|
||||
|
||||
from myfasthtml.core.dsl.base_completion import BaseCompletionEngine
|
||||
from myfasthtml.core.dsl.types import Position, Suggestion, CompletionResult
|
||||
@@ -11,6 +12,7 @@ from . import presets
|
||||
from .contexts import Context, DetectedScope, detect_scope, detect_context
|
||||
from .provider import DatagridMetadataProvider
|
||||
|
||||
logger = logging.getLogger("FormattingCompletionEngine")
|
||||
|
||||
class FormattingCompletionEngine(BaseCompletionEngine):
|
||||
"""
|
||||
@@ -222,8 +224,8 @@ class FormattingCompletionEngine(BaseCompletionEngine):
|
||||
if tables:
|
||||
columns = self.provider.list_columns(self.table_name)
|
||||
return [Suggestion(col, "Column", "column") for col in columns]
|
||||
except Exception:
|
||||
pass
|
||||
except Exception as ex:
|
||||
logger.error(f"Error getting column suggestions: {ex}")
|
||||
return []
|
||||
|
||||
def _get_column_suggestions_with_closing_quote(self) -> list[Suggestion]:
|
||||
|
||||
@@ -18,186 +18,185 @@ logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class DatagridMetadataProvider(SingleInstance, BaseMetadataProvider):
|
||||
"""Concrete session-scoped metadata provider for DataGrid DSL engines.
|
||||
"""Concrete session-scoped metadata provider for DataGrid DSL engines.
|
||||
|
||||
Implements BaseMetadataProvider by delegating live data queries to
|
||||
DataServicesManager. Also holds the global formatting presets and the
|
||||
all_tables_formats rule applied to every table.
|
||||
Implements BaseMetadataProvider by delegating live data queries to
|
||||
DataServicesManager. Also holds the global formatting presets and the
|
||||
all_tables_formats rule applied to every table.
|
||||
|
||||
Access pattern (from any component):
|
||||
provider = InstancesManager.get_by_type(session, DatagridMetadataProvider)
|
||||
Access pattern (from any component):
|
||||
provider = InstancesManager.get_by_type(session, DatagridMetadataProvider)
|
||||
|
||||
Attributes:
|
||||
style_presets: Dict of named style presets available in the DSL.
|
||||
formatter_presets: Dict of named formatter presets available in the DSL.
|
||||
all_tables_formats: Global format rules applied to all tables.
|
||||
Attributes:
|
||||
style_presets: Dict of named style presets available in the DSL.
|
||||
formatter_presets: Dict of named formatter presets available in the DSL.
|
||||
all_tables_formats: Global format rules applied to all tables.
|
||||
"""
|
||||
|
||||
def __init__(self, parent=None, session: Optional[dict] = None,
|
||||
_id: Optional[str] = None):
|
||||
super().__init__(parent, session, _id)
|
||||
self.style_presets: dict = DEFAULT_STYLE_PRESETS.copy()
|
||||
self.formatter_presets: dict = DEFAULT_FORMATTER_PRESETS.copy()
|
||||
self.all_tables_formats: list = []
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# Table and column metadata — delegated to DataServicesManager
|
||||
# ------------------------------------------------------------------
|
||||
|
||||
def list_tables(self) -> list[str]:
|
||||
"""Return the list of all registered table names.
|
||||
|
||||
Returns:
|
||||
List of table names in "namespace.name" format.
|
||||
"""
|
||||
manager = self._get_data_services_manager()
|
||||
if manager is None:
|
||||
return []
|
||||
return [s.table_name for s in manager._services.values() if s.table_name]
|
||||
|
||||
def list_columns(self, table_name: str) -> list[str]:
|
||||
"""Return the column identifiers for a table.
|
||||
|
||||
def __init__(self, parent=None, session: Optional[dict] = None,
|
||||
_id: Optional[str] = None):
|
||||
super().__init__(parent, session, _id)
|
||||
with self.initializing():
|
||||
self.style_presets: dict = DEFAULT_STYLE_PRESETS.copy()
|
||||
self.formatter_presets: dict = DEFAULT_FORMATTER_PRESETS.copy()
|
||||
self.all_tables_formats: list = []
|
||||
Args:
|
||||
table_name: Fully qualified table name.
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# Table and column metadata — delegated to DataServicesManager
|
||||
# ------------------------------------------------------------------
|
||||
Returns:
|
||||
List of col_id strings.
|
||||
"""
|
||||
service = self._get_service(table_name)
|
||||
if service is None:
|
||||
return []
|
||||
return [c.col_id for c in service.columns]
|
||||
|
||||
def list_column_values(self, table_name: str, column_name: str) -> list[Any]:
|
||||
"""Return the distinct values present in a column.
|
||||
|
||||
def list_tables(self) -> list[str]:
|
||||
"""Return the list of all registered table names.
|
||||
Args:
|
||||
table_name: Fully qualified table name.
|
||||
column_name: Column identifier.
|
||||
|
||||
Returns:
|
||||
List of table names in "namespace.name" format.
|
||||
"""
|
||||
manager = self._get_data_services_manager()
|
||||
if manager is None:
|
||||
return []
|
||||
return [s.table_name for s in manager._services.values() if s.table_name]
|
||||
Returns:
|
||||
List of distinct values, empty list if not found.
|
||||
"""
|
||||
service = self._get_service(table_name)
|
||||
if service is None:
|
||||
return []
|
||||
store = service.get_store()
|
||||
if store.ne_df is None or column_name not in store.ne_df.columns:
|
||||
return []
|
||||
return store.ne_df[column_name].dropna().unique().tolist()
|
||||
|
||||
def get_row_count(self, table_name: str) -> int:
|
||||
"""Return the number of rows in a table.
|
||||
|
||||
def list_columns(self, table_name: str) -> list[str]:
|
||||
"""Return the column identifiers for a table.
|
||||
Args:
|
||||
table_name: Fully qualified table name.
|
||||
|
||||
Args:
|
||||
table_name: Fully qualified table name.
|
||||
Returns:
|
||||
Row count, or 0 if not found.
|
||||
"""
|
||||
service = self._get_service(table_name)
|
||||
if service is None:
|
||||
return 0
|
||||
store = service.get_store()
|
||||
return store.ns_total_rows or 0
|
||||
|
||||
def get_column_type(self, table_name: str, column_name: str):
|
||||
"""Return the ColumnType for a column.
|
||||
|
||||
Returns:
|
||||
List of col_id strings.
|
||||
"""
|
||||
service = self._get_service(table_name)
|
||||
if service is None:
|
||||
return []
|
||||
return [c.col_id for c in service.columns]
|
||||
Args:
|
||||
table_name: Fully qualified table name.
|
||||
column_name: Column identifier.
|
||||
|
||||
def list_column_values(self, table_name: str, column_name: str) -> list[Any]:
|
||||
"""Return the distinct values present in a column.
|
||||
Returns:
|
||||
ColumnType enum value, or None if not found.
|
||||
"""
|
||||
service = self._get_service(table_name)
|
||||
if service is None:
|
||||
return None
|
||||
for col in service.columns:
|
||||
if col.col_id == column_name:
|
||||
return col.type
|
||||
return None
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# Preset metadata — held locally
|
||||
# ------------------------------------------------------------------
|
||||
|
||||
def list_style_presets(self) -> list[str]:
|
||||
"""Return the names of all registered style presets."""
|
||||
return list(self.style_presets.keys())
|
||||
|
||||
def list_format_presets(self) -> list[str]:
|
||||
"""Return the names of all registered formatter presets."""
|
||||
return list(self.formatter_presets.keys())
|
||||
|
||||
def get_style_presets(self) -> dict:
|
||||
"""Return the full style presets dict."""
|
||||
return self.style_presets
|
||||
|
||||
def get_formatter_presets(self) -> dict:
|
||||
"""Return the full formatter presets dict."""
|
||||
return self.formatter_presets
|
||||
|
||||
def add_style_preset(self, name: str, preset: dict) -> None:
|
||||
"""Add or update a named style preset.
|
||||
|
||||
Args:
|
||||
table_name: Fully qualified table name.
|
||||
column_name: Column identifier.
|
||||
Args:
|
||||
name: Preset name.
|
||||
preset: Style definition dict.
|
||||
"""
|
||||
self.style_presets[name] = preset
|
||||
|
||||
def add_formatter_preset(self, name: str, preset: dict) -> None:
|
||||
"""Add or update a named formatter preset.
|
||||
|
||||
Returns:
|
||||
List of distinct values, empty list if not found.
|
||||
"""
|
||||
service = self._get_service(table_name)
|
||||
if service is None:
|
||||
return []
|
||||
store = service.get_store()
|
||||
if store.ne_df is None or column_name not in store.ne_df.columns:
|
||||
return []
|
||||
return store.ne_df[column_name].dropna().unique().tolist()
|
||||
Args:
|
||||
name: Preset name.
|
||||
preset: Formatter definition dict.
|
||||
"""
|
||||
self.formatter_presets[name] = preset
|
||||
|
||||
def remove_style_preset(self, name: str) -> None:
|
||||
"""Remove a style preset by name.
|
||||
|
||||
def get_row_count(self, table_name: str) -> int:
|
||||
"""Return the number of rows in a table.
|
||||
Args:
|
||||
name: Preset name to remove.
|
||||
"""
|
||||
self.style_presets.pop(name, None)
|
||||
|
||||
def remove_formatter_preset(self, name: str) -> None:
|
||||
"""Remove a formatter preset by name.
|
||||
|
||||
Args:
|
||||
table_name: Fully qualified table name.
|
||||
Args:
|
||||
name: Preset name to remove.
|
||||
"""
|
||||
self.formatter_presets.pop(name, None)
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# Private helpers
|
||||
# ------------------------------------------------------------------
|
||||
|
||||
def _get_data_services_manager(self) -> Optional[DataServicesManager]:
|
||||
"""Return the DataServicesManager for this session."""
|
||||
return InstancesManager.get_by_type(
|
||||
self._session, DataServicesManager, default=None
|
||||
)
|
||||
|
||||
def _get_service(self, table_name: str):
|
||||
"""Return the DataService for a given table name.
|
||||
|
||||
Returns:
|
||||
Row count, or 0 if not found.
|
||||
"""
|
||||
service = self._get_service(table_name)
|
||||
if service is None:
|
||||
return 0
|
||||
store = service.get_store()
|
||||
return store.ns_total_rows or 0
|
||||
Args:
|
||||
table_name: Fully qualified table name.
|
||||
|
||||
def get_column_type(self, table_name: str, column_name: str):
|
||||
"""Return the ColumnType for a column.
|
||||
|
||||
Args:
|
||||
table_name: Fully qualified table name.
|
||||
column_name: Column identifier.
|
||||
|
||||
Returns:
|
||||
ColumnType enum value, or None if not found.
|
||||
"""
|
||||
service = self._get_service(table_name)
|
||||
if service is None:
|
||||
return None
|
||||
for col in service.columns:
|
||||
if col.col_id == column_name:
|
||||
return col.type
|
||||
return None
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# Preset metadata — held locally
|
||||
# ------------------------------------------------------------------
|
||||
|
||||
def list_style_presets(self) -> list[str]:
|
||||
"""Return the names of all registered style presets."""
|
||||
return list(self.style_presets.keys())
|
||||
|
||||
def list_format_presets(self) -> list[str]:
|
||||
"""Return the names of all registered formatter presets."""
|
||||
return list(self.formatter_presets.keys())
|
||||
|
||||
def get_style_presets(self) -> dict:
|
||||
"""Return the full style presets dict."""
|
||||
return self.style_presets
|
||||
|
||||
def get_formatter_presets(self) -> dict:
|
||||
"""Return the full formatter presets dict."""
|
||||
return self.formatter_presets
|
||||
|
||||
def add_style_preset(self, name: str, preset: dict) -> None:
|
||||
"""Add or update a named style preset.
|
||||
|
||||
Args:
|
||||
name: Preset name.
|
||||
preset: Style definition dict.
|
||||
"""
|
||||
self.style_presets[name] = preset
|
||||
|
||||
def add_formatter_preset(self, name: str, preset: dict) -> None:
|
||||
"""Add or update a named formatter preset.
|
||||
|
||||
Args:
|
||||
name: Preset name.
|
||||
preset: Formatter definition dict.
|
||||
"""
|
||||
self.formatter_presets[name] = preset
|
||||
|
||||
def remove_style_preset(self, name: str) -> None:
|
||||
"""Remove a style preset by name.
|
||||
|
||||
Args:
|
||||
name: Preset name to remove.
|
||||
"""
|
||||
self.style_presets.pop(name, None)
|
||||
|
||||
def remove_formatter_preset(self, name: str) -> None:
|
||||
"""Remove a formatter preset by name.
|
||||
|
||||
Args:
|
||||
name: Preset name to remove.
|
||||
"""
|
||||
self.formatter_presets.pop(name, None)
|
||||
|
||||
# ------------------------------------------------------------------
|
||||
# Private helpers
|
||||
# ------------------------------------------------------------------
|
||||
|
||||
def _get_data_services_manager(self) -> Optional[DataServicesManager]:
|
||||
"""Return the DataServicesManager for this session."""
|
||||
return InstancesManager.get_by_type(
|
||||
self._session, DataServicesManager, default=None
|
||||
)
|
||||
|
||||
def _get_service(self, table_name: str):
|
||||
"""Return the DataService for a given table name.
|
||||
|
||||
Args:
|
||||
table_name: Fully qualified table name.
|
||||
|
||||
Returns:
|
||||
DataService instance, or None if not found.
|
||||
"""
|
||||
manager = self._get_data_services_manager()
|
||||
if manager is None:
|
||||
return None
|
||||
for service in manager._services.values():
|
||||
if service.table_name == table_name:
|
||||
return service
|
||||
return None
|
||||
Returns:
|
||||
DataService instance, or None if not found.
|
||||
"""
|
||||
manager = self._get_data_services_manager()
|
||||
if manager is None:
|
||||
return None
|
||||
for service in manager._services.values():
|
||||
if service.table_name == table_name:
|
||||
return service
|
||||
return None
|
||||
|
||||
Reference in New Issue
Block a user