diff --git a/src/sempy_labs/semantic_model/_Fix_IsAvailableInMdxTrue.py b/src/sempy_labs/semantic_model/_Fix_IsAvailableInMdxTrue.py new file mode 100644 index 000000000..179718dbb --- /dev/null +++ b/src/sempy_labs/semantic_model/_Fix_IsAvailableInMdxTrue.py @@ -0,0 +1,62 @@ +# Fix "Set IsAvailableInMdx to true on necessary columns" — standalone BPA fixer. + +from typing import Optional +from uuid import UUID +from sempy._utils._log import log + + +@log +def fix_isavailable_in_mdx_true( + dataset: str | UUID, + workspace: Optional[str | UUID] = None, + scan_only: bool = False, +) -> int: + """ + Sets IsAvailableInMDX to True on key/attribute columns that incorrectly have it False. + + Targets columns that are: (a) used as keys in relationships (To side), or + (b) used in hierarchies, or (c) marked as IsKey. + + Parameters + ---------- + dataset : str | UUID + Name or ID of the semantic model. + workspace : str | uuid.UUID, default=None + The Fabric workspace name or ID. + scan_only : bool, default=False + If True, only reports what would be fixed without making changes. + + Returns + ------- + int + Number of items fixed. + """ + from sempy_labs.tom import connect_semantic_model + + fixed = 0 + with connect_semantic_model(dataset=dataset, readonly=scan_only, workspace=workspace) as tom: + # Collect columns that should have IsAvailableInMDX = True + needed = set() + for rel in tom.model.Relationships: + needed.add((str(rel.ToTable.Name), str(rel.ToColumn.Name))) + for table in tom.model.Tables: + for h in table.Hierarchies: + for lvl in h.Levels: + needed.add((str(table.Name), str(lvl.Column.Name))) + for col in table.Columns: + if col.IsKey: + needed.add((str(table.Name), str(col.Name))) + + for table in tom.model.Tables: + for col in table.Columns: + if (table.Name, col.Name) in needed and not col.IsAvailableInMDX: + if scan_only: + print(f" Would fix: '{table.Name}'[{col.Name}] IsAvailableInMDX → True") + else: + col.IsAvailableInMDX = True + print(f" Fixed: '{table.Name}'[{col.Name}] IsAvailableInMDX → True") + fixed += 1 + + action = "Would fix" if scan_only else "Fixed" + print(f" {action} {fixed} column(s).") + return fixed diff --git a/src/sempy_labs/semantic_model/__init__.py b/src/sempy_labs/semantic_model/__init__.py index 7c117f392..1b2fe5ede 100644 --- a/src/sempy_labs/semantic_model/__init__.py +++ b/src/sempy_labs/semantic_model/__init__.py @@ -6,10 +6,12 @@ from ._caching import ( enable_query_caching, ) +from ._Fix_IsAvailableInMdxTrue import fix_isavailable_in_mdx_true __all__ = [ "approved_for_copilot", "set_endorsement", "make_discoverable", "enable_query_caching", + "fix_isavailable_in_mdx_true", ]