semantic-link-labs 0.4.1__py3-none-any.whl → 0.5.0__py3-none-any.whl

sempy_labs/{_tom.py → tom/_model.py}

@@ -8,7 +8,7 @@ from sempy_labs._list_functions import list_relationships
 from sempy_labs._refresh_semantic_model import refresh_semantic_model
 from sempy_labs.directlake._fallback import check_fallback_reason
 from contextlib import contextmanager
-from typing import List, Optional, Union, TYPE_CHECKING
+from typing import List, Iterator, Optional, Union, TYPE_CHECKING
 from sempy._utils._log import log
 import sempy_labs._icons as icons
 
@@ -17,32 +17,28 @@ if TYPE_CHECKING:
     import Microsoft.AnalysisServices.Tabular as TOM
 
 
-checked = "\u2611"
-unchecked = "\u2610"
-start_bold = "\033[1m"
-end_bold = "\033[0m"
-
-
 class TOMWrapper:
     """
-    Convenience wrapper around the TOM object model for a semantic model. Always use connect_semantic_model function to make sure the TOM object is initialized correctly.
+    Convenience wrapper around the TOM object model for a semantic model. Always use the connect_semantic_model function to make sure the TOM object is initialized correctly.
+
+    `XMLA read/write endpoints <https://learn.microsoft.com/power-bi/enterprise/service-premium-connect-tools#to-enable-read-write-for-a-premium-capacity>`_ must be enabled if setting the readonly parameter to False.
     """
 
-    dataset: str
-    workspace: str
-    readonly: bool
-    tables_added: List[str]
+    _dataset: str
+    _workspace: str
+    _readonly: bool
+    _tables_added: List[str]
 
     def __init__(self, dataset, workspace, readonly):
-        self.dataset = dataset
-        self.workspace = workspace
-        self.readonly = readonly
-        self.tables_added = []
+        self._dataset = dataset
+        self._workspace = workspace
+        self._readonly = readonly
+        self._tables_added = []
 
-        self.tom_server = fabric.create_tom_server(
+        self._tom_server = fabric.create_tom_server(
             readonly=readonly, workspace=workspace
         )
-        self.model = self.tom_server.Databases.GetByName(dataset).Model
+        self.model = self._tom_server.Databases.GetByName(dataset).Model
 
     def all_columns(self):
         """
@@ -97,7 +93,7 @@ class TOMWrapper:
         import Microsoft.AnalysisServices.Tabular as TOM
 
         for t in self.model.Tables:
-            if any(p.SourceType == TOM.ColumnType.Calculated for p in t.Partitions):
+            if any(p.SourceType == TOM.PartitionSourceType.Calculated for p in t.Partitions):
                 yield t
 
     def all_calculation_groups(self):
@@ -181,10 +177,10 @@ class TOMWrapper:
             All levels within the semantic model.
         """
 
-        for t in self.model.Tables:
-            for h in t.Hierarchies:
-                for l in h.Levels:
-                    yield l
+        for table in self.model.Tables:
+            for hierarchy in table.Hierarchies:
+                for level in hierarchy.Levels:
+                    yield level
 
     def all_calculation_items(self):
         """
@@ -459,7 +455,7 @@ class TOMWrapper:
             Marks the column as the primary key of the table.
         summarize_by : str, default=None
             Sets the value for the Summarize By property of the column.
-            Defaults to None resolves to 'Default'.
+            Defaults to None which resolves to 'Default'.
         """
         import Microsoft.AnalysisServices.Tabular as TOM
         import System
@@ -504,7 +500,7 @@ class TOMWrapper:
         description: Optional[str] = None,
     ):
         """
-        Adds a calculation item to a calculation group within a semantic model.
+        Adds a `calculation item <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.calculationitem?view=analysisservices-dotnet>`_ to a `calculation group <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.calculationgroup?view=analysisservices-dotnet>`_ within a semantic model.
 
         Parameters
         ----------
@@ -642,7 +638,7 @@ class TOMWrapper:
         hierarchy_hidden: Optional[bool] = False,
     ):
         """
-        Adds a hierarchy to a table within a semantic model.
+        Adds a `hierarchy <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.hierarchy?view=analysisservices-dotnet>`_ to a table within a semantic model.
 
         Parameters
         ----------
@@ -662,22 +658,16 @@ class TOMWrapper:
         import Microsoft.AnalysisServices.Tabular as TOM
 
         if isinstance(columns, str):
-            print(
-                f"The 'levels' parameter must be a list. For example: ['Continent', 'Country', 'City']"
-            )
-            return
+            raise ValueError(f"{icons.red_dot} The 'levels' parameter must be a list. For example: ['Continent', 'Country', 'City']")
+            
         if len(columns) == 1:
-            print(f"There must be at least 2 levels in order to create a hierarchy.")
-            return
+            raise ValueError(f"{icons.red_dot} There must be at least 2 levels in order to create a hierarchy.")            
 
         if levels is None:
             levels = columns
 
         if len(columns) != len(levels):
-            print(
-                f"If specifying level names, you must specify a level for each column."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} If specifying level names, you must specify a level for each column.")            
 
         obj = TOM.Hierarchy()
         obj.Name = hierarchy_name
@@ -707,7 +697,7 @@ class TOMWrapper:
         rely_on_referential_integrity: Optional[bool] = False,
     ):
         """
-        Adds a relationship to a semantic model.
+        Adds a `relationship <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.singlecolumnrelationship?view=analysisservices-dotnet>`_ to a semantic model.
 
         Parameters
         ----------
@@ -779,7 +769,7 @@ class TOMWrapper:
         hidden: Optional[bool] = False,
     ):
         """
-        Adds a calculation group to a semantic model.
+        Adds a `calculation group <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.calculationgroup?view=analysisservices-dotnet>`_ to a semantic model.
 
         Parameters
         ----------
@@ -832,7 +822,7 @@ class TOMWrapper:
         self, name: str, expression: str, description: Optional[str] = None
     ):
         """
-        Adds an expression to a semantic model.
+        Adds an `expression <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.namedexpression?view=analysisservices-dotnet>`_ to a semantic model.
 
         Parameters
         ----------
@@ -856,7 +846,7 @@ class TOMWrapper:
 
     def add_translation(self, language: str):
         """
-        Adds a translation language (culture) to a semantic model.
+        Adds a `translation language <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.culture?view=analysisservices-dotnet>`_ (culture) to a semantic model.
 
         Parameters
         ----------
@@ -875,7 +865,7 @@ class TOMWrapper:
 
     def add_perspective(self, perspective_name: str):
         """
-        Adds a perspective to a semantic model.
+        Adds a `perspective <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.perspective?view=analysisservices-dotnet>`_ to a semantic model.
 
         Parameters
         ----------
@@ -910,18 +900,22 @@ class TOMWrapper:
         mode : str, default=None
             The query mode for the partition.
             Defaults to None which resolves to 'Import'.
+            `Valid mode values <https://learn.microsoft.com/en-us/dotnet/api/microsoft.analysisservices.tabular.modetype?view=analysisservices-dotnet>`_
         description : str, default=None
             A description for the partition.
         """
         import Microsoft.AnalysisServices.Tabular as TOM
         import System
 
-        mode = (
-            mode.title()
-            .replace("query", "Query")
-            .replace(" ", "")
-            .replace("lake", "Lake")
-        )
+        if mode is None:
+            mode = "Default"
+        else:
+            mode = (
+                mode.title()
+                .replace("query", "Query")
+                .replace(" ", "")
+                .replace("lake", "Lake")
+            )
 
         mp = TOM.MPartitionSource()
         mp.Expression = expression
@@ -930,8 +924,6 @@ class TOMWrapper:
         p.Source = mp
         if description is not None:
             p.Description = description
-        if mode is None:
-            mode = "Default"
         p.Mode = System.Enum.Parse(TOM.ModeType, mode)
 
         self.model.Tables[table_name].Partitions.Add(p)
@@ -966,7 +958,7 @@ class TOMWrapper:
         if expression is None:
             ep.ExpressionSource = self.model.Expressions["DatabaseQuery"]
         else:
-            ep.ExpressionSource = expression
+            ep.ExpressionSource = self.model.Expressions[expression]
         p = TOM.Partition()
         p.Name = table_name
         p.Source = ep
@@ -985,7 +977,7 @@ class TOMWrapper:
         base_column: Optional[str] = None,
     ):
         """
-        Sets the 'alternate of' property on a column.
+        Sets the `alternate of <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.alternateof?view=analysisservices-dotnet>`_ property on a column.
 
         Parameters
         ----------
@@ -1005,9 +997,7 @@ class TOMWrapper:
         import System
 
         if base_column is not None and base_table is None:
-            print(
-                f"ERROR: If you specify the base table you must also specify the base column"
-            )
+            raise ValueError(f"{icons.red_dot} If you specify the base table you must also specify the base column")
 
         summarization_type = (
             summarization_type.replace(" ", "")
@@ -1017,10 +1007,7 @@ class TOMWrapper:
 
         summarizationTypes = ["Sum", "GroupBy", "Count", "Min", "Max"]
         if summarization_type not in summarizationTypes:
-            print(
-                f"The 'summarization_type' parameter must be one of the following valuse: {summarizationTypes}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} The 'summarization_type' parameter must be one of the following valuse: {summarizationTypes}.")
 
         ao = TOM.AlternateOf()
         ao.Summarization = System.Enum.Parse(TOM.SummarizationType, summarization_type)
@@ -1039,7 +1026,7 @@ class TOMWrapper:
 
     def remove_alternate_of(self, table_name: str, column_name: str):
         """
-        Removes the 'alternate of' property on a column.
+        Removes the `alternate of <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.alternateof?view=analysisservices-dotnet>`_ property on a column.
 
         Parameters
         ----------
@@ -1059,7 +1046,7 @@ class TOMWrapper:
         self, object
     ) -> "Microsoft.AnalysisServices.Tabular.Annotation":
         """
-        Shows all annotations for a given object within a semantic model.
+        Shows all `annotations <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.annotation?view=analysisservices-dotnet>`_ for a given object within a semantic model.
 
         Parameters
         ----------
@@ -1081,7 +1068,7 @@ class TOMWrapper:
 
     def set_annotation(self, object, name: str, value: str):
         """
-        Sets an annotation on an object within the semantic model.
+        Sets an `annotation <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.annotation?view=analysisservices-dotnet>`_ on an object within the semantic model.
 
         Parameters
         ----------
@@ -1105,7 +1092,7 @@ class TOMWrapper:
 
     def get_annotation_value(self, object, name: str):
         """
-        Obtains the annotation value for a given annotation on an object within the semantic model.
+        Obtains the `annotation <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.annotation?view=analysisservices-dotnet>`_ value for a given annotation on an object within the semantic model.
 
         Parameters
         ----------
@@ -1124,7 +1111,7 @@ class TOMWrapper:
 
     def remove_annotation(self, object, name: str):
         """
-        Removes an annotation on an object within the semantic model.
+        Removes an `annotation <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.annotation?view=analysisservices-dotnet>`_ on an object within the semantic model.
 
         Parameters
         ----------
@@ -1138,7 +1125,7 @@ class TOMWrapper:
 
     def clear_annotations(self, object):
         """
-        Removes all annotations on an object within the semantic model.
+        Removes all `annotations <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.annotation?view=analysisservices-dotnet>`_ on an object within the semantic model.
 
         Parameters
         ----------
@@ -1152,7 +1139,7 @@ class TOMWrapper:
         self, object
     ) -> "Microsoft.AnalysisServices.Tabular.ExtendedProperty":
         """
-        Retrieves all extended properties on an object within the semantic model.
+        Retrieves all `extended properties <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.extendedproperty?view=analysisservices-dotnet>`_ on an object within the semantic model.
 
         Parameters
         ----------
@@ -1172,7 +1159,7 @@ class TOMWrapper:
         self, object, extended_property_type: str, name: str, value: str
     ):
         """
-        Sets an extended property on an object within the semantic model.
+        Sets an `extended property <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.extendedproperty?view=analysisservices-dotnet>`_ on an object within the semantic model.
 
         Parameters
         ----------
@@ -1205,7 +1192,7 @@ class TOMWrapper:
 
     def get_extended_property_value(self, object, name: str):
         """
-        Retrieves the value of an extended property for an object within the semantic model.
+        Retrieves the value of an `extended property <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.extendedproperty?view=analysisservices-dotnet>`_ for an object within the semantic model.
 
         Parameters
         ----------
@@ -1224,7 +1211,7 @@ class TOMWrapper:
 
     def remove_extended_property(self, object, name: str):
         """
-        Removes an extended property on an object within the semantic model.
+        Removes an `extended property <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.extendedproperty?view=analysisservices-dotnet>`_ on an object within the semantic model.
 
         Parameters
         ----------
@@ -1238,7 +1225,7 @@ class TOMWrapper:
 
     def clear_extended_properties(self, object):
         """
-        Removes all extended properties on an object within the semantic model.
+        Removes all `extended properties <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.extendedproperty?view=analysisservices-dotnet>`_ on an object within the semantic model.
 
         Parameters
         ----------
@@ -1254,7 +1241,7 @@ class TOMWrapper:
         perspective_name: str,
     ):
         """
-        Indicates whether an object is contained within a given perspective.
+        Indicates whether an object is contained within a given `perspective <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.perspective?view=analysisservices-dotnet>`_.
 
         Parameters
         ----------
@@ -1279,10 +1266,7 @@ class TOMWrapper:
         objectType = object.ObjectType
 
         if objectType not in validObjects:
-            print(
-                f"Only the following object types are valid for perspectives: {validObjects}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} Only the following object types are valid for perspectives: {validObjects}.")
 
         object.Model.Perspectives[perspective_name]
 
@@ -1313,7 +1297,7 @@ class TOMWrapper:
         perspective_name: str,
     ):
         """
-        Adds an object to a perspective.
+        Adds an object to a `perspective <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.perspective?view=analysisservices-dotnet>`_.
 
         Parameters
         ----------
@@ -1333,15 +1317,12 @@ class TOMWrapper:
         objectType = object.ObjectType
 
         if objectType not in validObjects:
-            print(
-                f"Only the following object types are valid for perspectives: {validObjects}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} Only the following object types are valid for perspectives: {validObjects}.")
+
         try:
             object.Model.Perspectives[perspective_name]
         except:
-            print(f"The '{perspective_name}' perspective does not exist.")
-            return
+            raise ValueError(f"{icons.red_dot} The '{perspective_name}' perspective does not exist.")
 
         # try:
         if objectType == TOM.ObjectType.Table:
@@ -1375,7 +1356,7 @@ class TOMWrapper:
         perspective_name: str,
     ):
         """
-        Removes an object from a perspective.
+        Removes an object from a `perspective <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.perspective?view=analysisservices-dotnet>`_.
 
         Parameters
         ----------
@@ -1395,15 +1376,12 @@ class TOMWrapper:
         objectType = object.ObjectType
 
         if objectType not in validObjects:
-            print(
-                f"Only the following object types are valid for perspectives: {validObjects}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} Only the following object types are valid for perspectives: {validObjects}.")
+
         try:
             object.Model.Perspectives[perspective_name]
         except:
-            print(f"The '{perspective_name}' perspective does not exist.")
-            return
+            raise ValueError(f"{icons.red_dot} The '{perspective_name}' perspective does not exist.")
 
         # try:
         if objectType == TOM.ObjectType.Table:
@@ -1449,7 +1427,7 @@ class TOMWrapper:
         value: str,
     ):
         """
-        Sets a translation value for an object's property.
+        Sets a `translation <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.culture?view=analysisservices-dotnet>`_ value for an object's property.
 
         Parameters
         ----------
@@ -1476,8 +1454,7 @@ class TOMWrapper:
         ]  # , 'Level'
 
         if object.ObjectType not in validObjects:
-            print(f"Translations can only be set to {validObjects}.")
-            return
+            raise ValueError(f"{icons.red_dot} Translations can only be set to {validObjects}.")
 
         mapping = {
             "Name": TOM.TranslatedProperty.Caption,
@@ -1486,14 +1463,13 @@ class TOMWrapper:
         }
 
         prop = mapping.get(property)
+        if prop == None:
+            raise ValueError(f"{icons.red_dot} Invalid property value. Please choose from the following: ['Name', 'Description', Display Folder].")
 
         try:
             object.Model.Cultures[language]
         except:
-            print(
-                f"The '{language}' translation language does not exist in the semantic model."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} The '{language}' translation language does not exist in the semantic model.")
 
         object.Model.Cultures[language].ObjectTranslations.SetTranslation(
             object, prop, value
@@ -1505,7 +1481,7 @@ class TOMWrapper:
         language: str,
     ):
         """
-        Removes an object's translation value.
+        Removes an object's `translation <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.culture?view=analysisservices-dotnet>`_ value.
 
         Parameters
         ----------
@@ -1579,7 +1555,7 @@ class TOMWrapper:
 
     def used_in_relationships(self, object: Union["TOM.Table", "TOM.Column"]):
         """
-        Shows all relationships in which a table/column is used.
+        Shows all `relationships <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.singlecolumnrelationship?view=analysisservices-dotnet>`_ in which a table/column is used.
 
         Parameters
         ----------
@@ -1612,7 +1588,7 @@ class TOMWrapper:
 
     def used_in_levels(self, column: "TOM.Column"):
         """
-        Shows all levels in which a column is used.
+        Shows all `levels <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.level?view=analysisservices-dotnet>`_ in which a column is used.
 
         Parameters
         ----------
@@ -1629,16 +1605,16 @@ class TOMWrapper:
         objType = column.ObjectType
 
         if objType == TOM.ObjectType.Column:
-            for l in self.all_levels():
+            for level in self.all_levels():
                 if (
-                    l.Parent.Table.Name == column.Parent.Name
-                    and l.Column.Name == column.Name
+                    level.Parent.Table.Name == column.Parent.Name
+                    and level.Column.Name == column.Name
                 ):
-                    yield l
+                    yield level
 
     def used_in_hierarchies(self, column: "TOM.Column"):
         """
-        Shows all hierarchies in which a column is used.
+        Shows all `hierarchies <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.hierarchy?view=analysisservices-dotnet>`_ in which a column is used.
 
         Parameters
         ----------
@@ -1691,7 +1667,7 @@ class TOMWrapper:
         dependencies: pd.DataFrame,
     ):
         """
-        Identifies the filter expressions which reference a given object.
+        Identifies the row level security `filter expressions <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.tablepermission.filterexpression?view=analysisservices-dotnet#microsoft-analysisservices-tabular-tablepermission-filterexpression>`_ which reference a given object.
 
         Parameters
         ----------
@@ -1855,7 +1831,7 @@ class TOMWrapper:
 
     def hybrid_tables(self):
         """
-        Outputs the hybrid tables within a semantic model.
+        Outputs the `hybrid tables <https://learn.microsoft.com/power-bi/connect-data/service-dataset-modes-understand#hybrid-tables>`_ within a semantic model.
 
         Parameters
         ----------
@@ -1874,7 +1850,7 @@ class TOMWrapper:
 
     def date_tables(self):
         """
-        Outputs the tables which are marked as date tables within a semantic model.
+        Outputs the tables which are marked as `date tables <https://learn.microsoft.com/power-bi/transform-model/desktop-date-tables>`_ within a semantic model.
 
         Parameters
         ----------
@@ -1895,7 +1871,7 @@ class TOMWrapper:
 
     def is_hybrid_table(self, table_name: str):
         """
-        Identifies if a table is a hybrid table.
+        Identifies if a table is a `hybrid table <https://learn.microsoft.com/power-bi/connect-data/service-dataset-modes-understand#hybrid-tables>`_.
 
         Parameters
         ----------
@@ -1925,7 +1901,7 @@ class TOMWrapper:
 
     def is_date_table(self, table_name: str):
         """
-        Identifies if a table is marked as a date table.
+        Identifies if a table is marked as a `date tables <https://learn.microsoft.com/power-bi/transform-model/desktop-date-tables>`_.
 
         Parameters
         ----------
@@ -1939,18 +1915,11 @@ class TOMWrapper:
         """
         import Microsoft.AnalysisServices.Tabular as TOM
 
-        isDateTable = False
-        t = self.model.Tables[table_name]
-
-        if t.DataCategory == "Time":
-            if any(c.IsKey and c.DataType == TOM.DataType.DateTime for c in t.Columns):
-                isDateTable = True
-
-        return isDateTable
+        return any(c.IsKey and c.DataType == TOM.DataType.DateTime for c in self.all_columns() if c.Parent.Name == table_name and c.Parent.DataCategory == 'Time')
 
     def mark_as_date_table(self, table_name: str, column_name: str):
         """
-        Marks a table as a date table.
+        Marks a table as a `date table <https://learn.microsoft.com/power-bi/transform-model/desktop-date-tables>`_.
 
         Parameters
         ----------
@@ -1964,11 +1933,8 @@ class TOMWrapper:
         t = self.model.Tables[table_name]
         c = t.Columns[column_name]
         if c.DataType != TOM.DataType.DateTime:
-            print(
-                f"{icons.red_dot} The column specified in the 'column_name' parameter in this function must be of DateTime data type."
-            )
-            return
-
+            raise ValueError(f"{icons.red_dot} The column specified in the 'column_name' parameter in this function must be of DateTime data type.")
+            
         daxQuery = f"""
         define measure '{table_name}'[test] = 
         var mn = MIN('{table_name}'[{column_name}])
@@ -1983,14 +1949,11 @@ class TOMWrapper:
         )
         """
         df = fabric.evaluate_dax(
-            dataset=self.dataset, workspace=self.workspace, dax_string=daxQuery
+            dataset=self._dataset, workspace=self._workspace, dax_string=daxQuery
         )
         value = df["1"].iloc[0]
         if value != "1":
-            print(
-                f"{icons.red_dot} The '{column_name}' within the '{table_name}' table does not contain contiguous date values."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} The '{column_name}' within the '{table_name}' table does not contain contiguous date values.")
 
         # Mark as a date table
         t.DataCategory = "Time"
@@ -2001,7 +1964,7 @@ class TOMWrapper:
 
     def has_aggs(self):
         """
-        Identifies if a semantic model has any aggregations.
+        Identifies if a semantic model has any `aggregations <https://learn.microsoft.com/power-bi/transform-model/aggregations-advanced>`_.
 
         Parameters
         ----------
@@ -2012,17 +1975,11 @@ class TOMWrapper:
             Indicates if the semantic model has any aggregations.
         """
 
-        hasAggs = False
-
-        for c in self.all_columns():
-            if c.AlterateOf is not None:
-                hasAggs = True
-
-        return hasAggs
+        return any(c.AlternateOf is not None for c in self.all_columns())
 
     def is_agg_table(self, table_name: str):
         """
-        Identifies if a table has aggregations.
+        Identifies if a table has `aggregations <https://learn.microsoft.com/power-bi/transform-model/aggregations-advanced>`_.
 
         Parameters
         ----------
@@ -2041,7 +1998,7 @@ class TOMWrapper:
 
     def has_hybrid_table(self):
         """
-        Identifies if a semantic model has a hybrid table.
+        Identifies if a semantic model has a `hybrid table <https://learn.microsoft.com/power-bi/connect-data/service-dataset-modes-understand#hybrid-tables>`_.
 
         Parameters
         ----------
@@ -2050,19 +2007,13 @@ class TOMWrapper:
         -------
         bool
             Indicates if the semantic model has a hybrid table.
-        """
+        """    
 
-        hasHybridTable = False
-
-        for t in self.model.Tables:
-            if self.is_hybrid_table(table_name=t.Name):
-                hasHybridTable = True
-
-        return hasHybridTable
+        return any(self.is_hybrid_table(table_name=t.Name) for t in self.model.Tables)
 
     def has_date_table(self):
         """
-        Identifies if a semantic model has a table marked as a date table.
+        Identifies if a semantic model has a table marked as a `date table <https://learn.microsoft.com/power-bi/transform-model/desktop-date-tables>`_.
 
         Parameters
         ----------
@@ -2073,17 +2024,11 @@ class TOMWrapper:
             Indicates if the semantic model has a table marked as a date table.
         """
 
-        hasDateTable = False
-
-        for t in self.model.Tables:
-            if self.is_date_table(table_name=t.Name):
-                hasDateTable = True
-
-        return hasDateTable
+        return any(self.is_date_table(table_name=t.Name) for t in self.model.Tables)
 
     def is_direct_lake(self):
         """
-        Identifies if a semantic model is in Direct Lake mode.
+        Identifies if a semantic model is in `Direct Lake <https://learn.microsoft.com/fabric/get-started/direct-lake-overview>`_ mode.
 
         Parameters
         ----------
@@ -2093,6 +2038,7 @@ class TOMWrapper:
         bool
             Indicates if the semantic model is in Direct Lake mode.
         """
+        import Microsoft.AnalysisServices.Tabular as TOM
 
         return any(
             p.Mode == TOM.ModeType.DirectLake
@@ -2102,7 +2048,7 @@ class TOMWrapper:
 
     def is_field_parameter(self, table_name: str):
         """
-        Identifies if a table is a field parameter.
+        Identifies if a table is a `field parameter <https://learn.microsoft.com/power-bi/create-reports/power-bi-field-parameters>`_.
 
         Parameters
         ----------
@@ -2134,7 +2080,7 @@ class TOMWrapper:
 
     def is_auto_date_table(self, table_name: str):
         """
-        Identifies if a table is an auto-date table.
+        Identifies if a table is an `auto date/time table <https://learn.microsoft.com/power-bi/transform-model/desktop-auto-date-time>`_.
 
         Parameters
         ----------
@@ -2174,7 +2120,7 @@ class TOMWrapper:
         status_graphic: Optional[str] = None,
     ):
         """
-        Sets the properties to add/update a KPI for a measure.
+        Sets the properties to add/update a `KPI <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.kpi?view=analysisservices-dotnet>`_ for a measure.
 
         Parameters
         ----------
@@ -2202,58 +2148,44 @@ class TOMWrapper:
         # https://github.com/m-kovalsky/Tabular/blob/master/KPI%20Graphics.md
 
         if measure_name == target:
-            print(
-                f"The 'target' parameter cannot be the same measure as the 'measure_name' parameter."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} The 'target' parameter cannot be the same measure as the 'measure_name' parameter.")
 
         if status_graphic is None:
             status_graphic = "Three Circles Colored"
 
-        statusType = ["Linear", "LinearReversed", "Centered", "CenteredReversed"]
-        status_type = status_type.title().replace(" ", "")
-
+        valid_status_types = ["Linear", "LinearReversed", "Centered", "CenteredReversed"]
+        status_type = status_type
         if status_type is None:
             status_type = "Linear"
+        else:
+            status_type = status_type.title().replace(" ", "")
 
-        if status_type not in statusType:
-            print(
-                f"'{status_type}' is an invalid status_type. Please choose from these options: {statusType}."
-            )
-            return
+        if status_type not in valid_status_types:
+            raise ValueError(f"{icons.red_dot} '{status_type}' is an invalid status_type. Please choose from these options: {valid_status_types}.")
 
         if status_type in ["Linear", "LinearReversed"]:
             if upper_bound is not None or lower_mid_bound is not None:
-                print(
-                    f"The 'upper_mid_bound' and 'lower_mid_bound' parameters are not used in the 'Linear' and 'LinearReversed' status types. Make sure these parameters are set to None."
-                )
-                return
+                raise ValueError(f"{icons.red_dot} The 'upper_mid_bound' and 'lower_mid_bound' parameters are not used in the 'Linear' and 'LinearReversed' status types. Make sure these parameters are set to None.")
+
             elif upper_bound <= lower_bound:
-                print(f"The upper_bound must be greater than the lower_bound.")
-                return
+                raise ValueError(f"{icons.red_dot} The upper_bound must be greater than the lower_bound.")
 
         if status_type in ["Centered", "CenteredReversed"]:
             if upper_mid_bound is None or lower_mid_bound is None:
-                print(
-                    f"The 'upper_mid_bound' and 'lower_mid_bound' parameters are necessary in the 'Centered' and 'CenteredReversed' status types."
-                )
-                return
+                raise ValueError(f"{icons.red_dot} The 'upper_mid_bound' and 'lower_mid_bound' parameters are necessary in the 'Centered' and 'CenteredReversed' status types.")
             elif upper_bound <= upper_mid_bound:
-                print(f"The upper_bound must be greater than the upper_mid_bound.")
+                raise ValueError(f"{icons.red_dot} The upper_bound must be greater than the upper_mid_bound.")
             elif upper_mid_bound <= lower_mid_bound:
-                print(f"The upper_mid_bound must be greater than the lower_mid_bound.")
+                raise ValueError(f"{icons.red_dot} The upper_mid_bound must be greater than the lower_mid_bound.")
             elif lower_mid_bound <= lower_bound:
-                print(f"The lower_mid_bound must be greater than the lower_bound.")
+                raise ValueError(f"{icons.red_dot} The lower_mid_bound must be greater than the lower_bound.")
 
         try:
             table_name = next(
                 m.Parent.Name for m in self.all_measures() if m.Name == measure_name
             )
         except:
-            print(
-                f"The '{measure_name}' measure does not exist in the '{self.dataset}' semantic model within the '{self.workspace}'."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} The '{measure_name}' measure does not exist in the '{self._dataset}' semantic model within the '{self._workspace}'.")            
 
         graphics = [
             "Cylinder",
@@ -2276,10 +2208,7 @@ class TOMWrapper:
         ]
 
         if status_graphic not in graphics:
-            print(
-                f"The '{status_graphic}' status graphic is not valid. Please choose from these options: {graphics}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} The '{status_graphic}' status graphic is not valid. Please choose from these options: {graphics}.")
 
         measure_target = True
 
@@ -2295,10 +2224,8 @@ class TOMWrapper:
                     if m.Name == target
                 )
             except:
-                print(
-                    f"The '{target}' measure does not exist in the '{self.dataset}' semantic model within the '{self.workspace}'."
-                )
-
+                raise ValueError(f"{icons.red_dot} The '{target}' measure does not exist in the '{self._dataset}' semantic model within the '{self._workspace}'.")
+            
         if measure_target:
             expr = f"var x = [{measure_name}]/[{target}]\nreturn"
         else:
@@ -2328,7 +2255,7 @@ class TOMWrapper:
 
     def set_aggregations(self, table_name: str, agg_table_name: str):
         """
-        Sets the aggregations (alternate of) for all the columns in an aggregation table based on a base table.
+        Sets the `aggregations <https://learn.microsoft.com/power-bi/transform-model/aggregations-advanced>`_ (alternate of) for all the columns in an aggregation table based on a base table.
 
         Parameters
         ----------
@@ -2367,7 +2294,7 @@ class TOMWrapper:
         self, table_name: str, column_name: str, value: Optional[bool] = False
     ):
         """
-        Sets the IsAvailableInMdx property on a column.
+        Sets the `IsAvailableInMDX <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.column.isavailableinmdx?view=analysisservices-dotnet#microsoft-analysisservices-tabular-column-isavailableinmdx>`_ property on a column.
 
         Parameters
         ----------
@@ -2385,7 +2312,7 @@ class TOMWrapper:
         self, table_name: str, column_name: str, value: Optional[str] = None
     ):
         """
-        Sets the SummarizeBy property on a column.
+        Sets the `SummarizeBy <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.column.summarizeby?view=analysisservices-dotnet#microsoft-analysisservices-tabular-column-summarizeby>`_ property on a column.
 
         Parameters
         ----------
@@ -2421,10 +2348,7 @@ class TOMWrapper:
         )
 
         if value not in values:
-            print(
-                f"'{value}' is not a valid value for the SummarizeBy property. These are the valid values: {values}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} '{value}' is not a valid value for the SummarizeBy property. These are the valid values: {values}.")
 
         self.model.Tables[table_name].Columns[column_name].SummarizeBy = (
             System.Enum.Parse(TOM.AggregateFunction, value)
@@ -2432,7 +2356,7 @@ class TOMWrapper:
 
     def set_direct_lake_behavior(self, direct_lake_behavior: str):
         """
-        Sets the Direct Lake Behavior property for a semantic model.
+        Sets the `Direct Lake Behavior <https://learn.microsoft.com/fabric/get-started/direct-lake-overview#fallback-behavior>`_ property for a semantic model.
 
         Parameters
         ----------
@@ -2453,10 +2377,7 @@ class TOMWrapper:
         dlValues = ["Automatic", "DirectLakeOnly", "DirectQueryOnly"]
 
         if direct_lake_behavior not in dlValues:
-            print(
-                f"The 'direct_lake_behavior' parameter must be one of these values: {dlValues}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} The 'direct_lake_behavior' parameter must be one of these values: {dlValues}.")
 
         self.model.DirectLakeBehavior = System.Enum.Parse(
             TOM.DirectLakeBehavior, direct_lake_behavior
@@ -2539,7 +2460,7 @@ class TOMWrapper:
 
     def add_field_parameter(self, table_name: str, objects: List[str]):
         """
-        Adds a table to the semantic model.
+        Adds a `field parameter <https://learn.microsoft.com/power-bi/create-reports/power-bi-field-parameters>`_ to the semantic model.
 
         Parameters
         ----------
@@ -2551,14 +2472,13 @@ class TOMWrapper:
             Measures may be formatted as '[Measure Name]' or 'Measure Name'.
         """
 
+        import Microsoft.AnalysisServices.Tabular as TOM
+
         if isinstance(objects, str):
-            print(f"The 'objects' parameter must be a list of columns/measures.")
-            return
+            raise ValueError(f"{icons.red_dot} The 'objects' parameter must be a list of columns/measures.")
+
         if len(objects) == 1:
-            print(
-                f"There must be more than one object (column/measure) within the objects parameter."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} There must be more than one object (column/measure) within the objects parameter.")
 
         expr = ""
         i = 0
@@ -2594,10 +2514,7 @@ class TOMWrapper:
                     )
                     success = True
             if not success:
-                print(
-                    f"The '{obj}' object was not found in the '{self.dataset}' semantic model."
-                )
-                return
+                raise ValueError(f"{icons.red_dot} The '{obj}' object was not found in the '{self._dataset}' semantic model.")
             else:
                 i += 1
 
@@ -2631,7 +2548,6 @@ class TOMWrapper:
         )
 
         self.set_extended_property(
-            self=self,
             object=self.model.Tables[table_name].Columns[col2],
             extended_property_type="Json",
             name="ParameterMetadata",
@@ -2649,11 +2565,11 @@ class TOMWrapper:
         ].Columns[col3]
         self.model.Tables[table_name].Columns[table_name].RelatedColumnDetails = rcd
 
-        self.tables_added.append(table_name)
+        self._tables_added.append(table_name)
 
     def remove_vertipaq_annotations(self):
         """
-        Removes the annotations set using the [set_vertipaq_annotations] function.
+        Removes the annotations set using the set_vertipaq_annotations function.
         """
 
         for t in self.model.Tables:
@@ -2683,10 +2599,10 @@ class TOMWrapper:
         """
 
         dfT = fabric.list_tables(
-            dataset=self.dataset, workspace=self.workspace, extended=True
+            dataset=self._dataset, workspace=self._workspace, extended=True
         )
         dfC = fabric.list_columns(
-            dataset=self.dataset, workspace=self.workspace, extended=True
+            dataset=self._dataset, workspace=self._workspace, extended=True
         )
         # intList = ['Total Size']#, 'Data Size', 'Dictionary Size', 'Hierarchy Size']
         dfCSum = dfC.groupby(["Table Name"])["Total Size"].sum().reset_index()
@@ -2698,16 +2614,16 @@ class TOMWrapper:
             how="inner",
         )
         dfP = fabric.list_partitions(
-            dataset=self.dataset, workspace=self.workspace, extended=True
+            dataset=self._dataset, workspace=self._workspace, extended=True
         )
         dfP["Records per Segment"] = round(
             dfP["Record Count"] / dfP["Segment Count"], 2
         )
         dfH = fabric.list_hierarchies(
-            dataset=self.dataset, workspace=self.workspace, extended=True
+            dataset=self._dataset, workspace=self._workspace, extended=True
         )
         dfR = list_relationships(
-            dataset=self.dataset, workspace=self.workspace, extended=True
+            dataset=self._dataset, workspace=self._workspace, extended=True
         )
 
         for t in self.model.Tables:
@@ -3100,7 +3016,7 @@ class TOMWrapper:
         import Microsoft.AnalysisServices.Tabular as TOM
 
         def create_pattern(a, b):
-            return r"(?<!" + a + "\[)(?<!" + a + "'\[)" + b
+            return r"(?<!" + re.escape(a) + r"\[)(?<!" + re.escape(a) + r"'\[)" + re.escape(b)
 
         for obj in self.depends_on(object=object, dependencies=dependencies):
             if obj.ObjectType == TOM.ObjectType.Column:
@@ -3128,7 +3044,7 @@ class TOMWrapper:
         usingView = False
 
         if self.is_direct_lake():
-            df = check_fallback_reason(dataset=self.dataset, workspace=self.workspace)
+            df = check_fallback_reason(dataset=self._dataset, workspace=self._workspace)
             df_filt = df[df["FallbackReasonID"] == 2]
 
             if len(df_filt) > 0:
@@ -3138,7 +3054,7 @@ class TOMWrapper:
 
     def has_incremental_refresh_policy(self, table_name: str):
         """
-        Identifies whether a table has an incremental refresh policy.
+        Identifies whether a table has an `incremental refresh <https://learn.microsoft.com/power-bi/connect-data/incremental-refresh-overview>`_ policy.
 
         Parameters
         ----------
@@ -3161,7 +3077,7 @@ class TOMWrapper:
 
     def show_incremental_refresh_policy(self, table_name: str):
         """
-        Prints the incremental refresh policy for a table.
+        Prints the `incremental refresh <https://learn.microsoft.com/power-bi/connect-data/incremental-refresh-overview>`_ policy for a table.
 
         Parameters
         ----------
@@ -3169,11 +3085,13 @@ class TOMWrapper:
             Name of the table.
         """
 
+        import Microsoft.AnalysisServices.Tabular as TOM
+
         rp = self.model.Tables[table_name].RefreshPolicy
 
         if rp is None:
             print(
-                f"The '{table_name}' table in the '{self.dataset}' semantic model within the '{self.workspace}' workspace does not have an incremental refresh policy."
+                f"{icons.yellow_dot} The '{table_name}' table in the '{self._dataset}' semantic model within the '{self._workspace}' workspace does not have an incremental refresh policy."
             )
         else:
             print(f"Table Name: {table_name}")
@@ -3181,33 +3099,33 @@ class TOMWrapper:
             icGran = str(rp.IncrementalGranularity).lower()
             if rp.RollingWindowPeriods > 1:
                 print(
-                    f"Archive data starting {start_bold}{rp.RollingWindowPeriods} {rwGran}s{end_bold} before refresh date."
+                    f"Archive data starting {icons.start_bold}{rp.RollingWindowPeriods} {rwGran}s{icons.end_bold} before refresh date."
                 )
             else:
                 print(
-                    f"Archive data starting {start_bold}{rp.RollingWindowPeriods} {rwGran}{end_bold} before refresh date."
+                    f"Archive data starting {icons.start_bold}{rp.RollingWindowPeriods} {rwGran}{icons.end_bold} before refresh date."
                 )
             if rp.IncrementalPeriods > 1:
                 print(
-                    f"Incrementally refresh data {start_bold}{rp.IncrementalPeriods} {icGran}s{end_bold} before refresh date."
+                    f"Incrementally refresh data {icons.start_bold}{rp.IncrementalPeriods} {icGran}s{icons.end_bold} before refresh date."
                 )
             else:
                 print(
-                    f"Incrementally refresh data {start_bold}{rp.IncrementalPeriods} {icGran}{end_bold} before refresh date."
+                    f"Incrementally refresh data {icons.start_bold}{rp.IncrementalPeriods} {icGran}{icons.end_bold} before refresh date."
                 )
 
             if rp.Mode == TOM.RefreshPolicyMode.Hybrid:
                 print(
-                    f"{checked} Get the latest data in real time with DirectQuery (Premium only)"
+                    f"{icons.checked} Get the latest data in real time with DirectQuery (Premium only)"
                 )
             else:
                 print(
-                    f"{unchecked} Get the latest data in real time with DirectQuery (Premium only)"
+                    f"{icons.unchecked} Get the latest data in real time with DirectQuery (Premium only)"
                 )
             if rp.IncrementalPeriodsOffset == -1:
-                print(f"{checked} Only refresh complete days")
+                print(f"{icons.checked} Only refresh complete days")
             else:
-                print(f"{unchecked} Only refresh complete days")
+                print(f"{icons.unchecked} Only refresh complete days")
             if len(rp.PollingExpression) > 0:
                 pattern = r"\[([^\]]+)\]"
                 match = re.search(pattern, rp.PollingExpression)
@@ -3215,10 +3133,10 @@ class TOMWrapper:
                     col = match[0][1:-1]
                     fullCol = format_dax_object_name(table_name, col)
                     print(
-                        f"{checked} Detect data changes: {start_bold}{fullCol}{end_bold}"
+                        f"{icons.checked} Detect data changes: {icons.start_bold}{fullCol}{icons.end_bold}"
                     )
             else:
-                print(f"{unchecked} Detect data changes")
+                print(f"{icons.unchecked} Detect data changes")
 
     def update_incremental_refresh_policy(
         self,
@@ -3231,7 +3149,7 @@ class TOMWrapper:
         detect_data_changes_column: Optional[str] = None,
     ):
         """
-        Updates the incremental refresh policy for a table within a semantic model.
+        Updates the `incremental refresh <https://learn.microsoft.com/power-bi/connect-data/incremental-refresh-overview>`_ policy for a table within a semantic model.
 
         Parameters
         ----------
@@ -3266,26 +3184,16 @@ class TOMWrapper:
         rolling_window_granularity = rolling_window_granularity.capitalize()
 
         if incremental_granularity not in incGran:
-            print(
-                f"{icons.red_dot} Invalid 'incremental_granularity' value. Please choose from the following options: {incGran}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} Invalid 'incremental_granularity' value. Please choose from the following options: {incGran}.")
+ 
         if rolling_window_granularity not in incGran:
-            print(
-                f"{icons.red_dot} Invalid 'rolling_window_granularity' value. Please choose from the following options: {incGran}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} Invalid 'rolling_window_granularity' value. Please choose from the following options: {incGran}.")
 
         if rolling_window_periods < 1:
-            print(
-                f"{icons.red_dot} Invalid 'rolling_window_periods' value. Must be a value greater than 0."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} Invalid 'rolling_window_periods' value. Must be a value greater than 0.")
+            
         if incremental_periods < 1:
-            print(
-                f"{icons.red_dot} Invalid 'incremental_periods' value. Must be a value greater than 0."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} Invalid 'incremental_periods' value. Must be a value greater than 0.")
 
         t = self.model.Tables[table_name]
 
@@ -3293,10 +3201,7 @@ class TOMWrapper:
             dc = t.Columns[detect_data_changes_column]
 
             if dc.DataType != TOM.DataType.DateTime:
-                print(
-                    f"{icons.red_dot} Invalid 'detect_data_changes_column' parameter. This column must be of DateTime data type."
-                )
-                return
+                raise ValueError(f"{icons.red_dot} Invalid 'detect_data_changes_column' parameter. This column must be of DateTime data type.")
 
         rp = TOM.BasicRefreshPolicy()
         rp.IncrementalPeriods = incremental_periods
@@ -3339,7 +3244,7 @@ class TOMWrapper:
         detect_data_changes_column: Optional[str] = None,
     ):
         """
-        Adds anincremental refresh policy for a table within a semantic model.
+        Adds an `incremental refresh <https://learn.microsoft.com/power-bi/connect-data/incremental-refresh-overview>`_ policy for a table within a semantic model.
 
         Parameters
         ----------
@@ -3376,26 +3281,16 @@ class TOMWrapper:
         rolling_window_granularity = rolling_window_granularity.capitalize()
 
         if incremental_granularity not in incGran:
-            print(
-                f"{icons.red_dot} Invalid 'incremental_granularity' value. Please choose from the following options: {incGran}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} Invalid 'incremental_granularity' value. Please choose from the following options: {incGran}.")
+            
         if rolling_window_granularity not in incGran:
-            print(
-                f"{icons.red_dot} Invalid 'rolling_window_granularity' value. Please choose from the following options: {incGran}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} Invalid 'rolling_window_granularity' value. Please choose from the following options: {incGran}.")
 
         if rolling_window_periods < 1:
-            print(
-                f"{icons.red_dot} Invalid 'rolling_window_periods' value. Must be a value greater than 0."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} Invalid 'rolling_window_periods' value. Must be a value greater than 0.")
+
         if incremental_periods < 1:
-            print(
-                f"{icons.red_dot} Invalid 'incremental_periods' value. Must be a value greater than 0."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} Invalid 'incremental_periods' value. Must be a value greater than 0.")
 
         date_format = "%m/%d/%Y"
 
@@ -3410,10 +3305,7 @@ class TOMWrapper:
         end_day = date_obj_end.day
 
         if date_obj_end <= date_obj_start:
-            print(
-                f"{icons.red_dot} Invalid 'start_date' or 'end_date'. The 'end_date' must be after the 'start_date'."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} Invalid 'start_date' or 'end_date'. The 'end_date' must be after the 'start_date'.")
 
         t = self.model.Tables[table_name]
 
@@ -3422,20 +3314,14 @@ class TOMWrapper:
         dType = c.DataType
 
         if dType != TOM.DataType.DateTime:
-            print(
-                f"{icons.red_dot} The {fcName} column is of '{dType}' data type. The column chosen must be of DateTime data type."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} The {fcName} column is of '{dType}' data type. The column chosen must be of DateTime data type.")
 
         if detect_data_changes_column is not None:
             dc = t.Columns[detect_data_changes_column]
             dcType = dc.DataType
 
             if dcType != TOM.DataType.DateTime:
-                print(
-                    f"{icons.red_dot} Invalid 'detect_data_changes_column' parameter. This column must be of DateTime data type."
-                )
-                return
+                raise ValueError(f"{icons.red_dot} Invalid 'detect_data_changes_column' parameter. This column must be of DateTime data type.")
 
         # Start changes:
 
@@ -3443,10 +3329,8 @@ class TOMWrapper:
         i = 0
         for p in t.Partitions:
             if p.SourceType != TOM.PartitionSourceType.M:
-                print(
-                    f"{icons.red_dot} Invalid partition source type. Incremental refresh can only be set up if the table's partition is an M-partition."
-                )
-                return
+                raise ValueError(f"{icons.red_dot} Invalid partition source type. Incremental refresh can only be set up if the table's partition is an M-partition.")
+
             elif i == 0:
                 text = p.Expression
                 text = text.rstrip()
@@ -3462,8 +3346,7 @@ class TOMWrapper:
 
                     print(text_before_last_match)
                 else:
-                    print(f"{icons.red_dot} Invalid M-partition expression.")
-                    return
+                    raise ValueError(f"{icons.red_dot} Invalid M-partition expression.")
 
                 endExpr = f'#"Filtered Rows IR" = Table.SelectRows({obj}, each [{column_name}] >= RangeStart and [{column_name}] <= RangeEnd)\n#"Filtered Rows IR"'
                 finalExpr = text_before_last_match + endExpr
@@ -3514,7 +3397,7 @@ class TOMWrapper:
         max_parallelism: Optional[int] = 0,
     ):
         """
-        Applies the incremental refresh policy for a table within a semantic model.
+        `Applies the incremental refresh <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.table.applyrefreshpolicy?view=analysisservices-dotnet#microsoft-analysisservices-tabular-table-applyrefreshpolicy(system-boolean-system-int32)>`_ policy for a table within a semantic model.
 
         Parameters
         ----------
@@ -3538,7 +3421,7 @@ class TOMWrapper:
         self, table_name: str, partition_name: str, expression: str
     ):
         """
-        Sets the data coverage definition for a partition.
+        Sets the `data coverage definition <https://learn.microsoft.com/analysis-services/tom/table-partitions?view=asallproducts-allversions>`_ for a partition.
 
         Parameters
         ----------
@@ -3559,15 +3442,9 @@ class TOMWrapper:
         ht = self.is_hybrid_table(table_name=table_name)
 
         if not ht:
-            print(
-                f"The data coverage definition property is only applicable to hybrid tables. See the documentation: {doc}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} The `data coverage definition <https://learn.microsoft.com/analysis-services/tom/table-partitions?view=asallproducts-allversions>`_ property is only applicable to `hybrid tables <https://learn.microsoft.com/power-bi/connect-data/service-dataset-modes-understand#hybrid-tables>`_. See the documentation: {doc}.")
         if p.Mode != TOM.ModeType.DirectQuery:
-            print(
-                f"The data coverage definition property is only applicable to the DirectQuery partition of a hybrid table. See the documentation: {doc}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} The `data coverage definition <https://learn.microsoft.com/analysis-services/tom/table-partitions?view=asallproducts-allversions>`_ property is only applicable to the DirectQuery partition of a `hybrid table <https://learn.microsoft.com/power-bi/connect-data/service-dataset-modes-understand#hybrid-tables>`_. See the documentation: {doc}.")
 
         dcd = TOM.DataCoverageDefinition()
         dcd.Expression = expression
@@ -3575,7 +3452,7 @@ class TOMWrapper:
 
     def set_encoding_hint(self, table_name: str, column_name: str, value: str):
         """
-        Sets the encoding hint for a column.
+        Sets the `encoding hint <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.encodinghinttype?view=analysisservices-dotnet>`_ for a column.
 
         Parameters
         ----------
@@ -3594,10 +3471,7 @@ class TOMWrapper:
         value = value.capitalize()
 
         if value not in values:
-            print(
-                f"{icons.red_dot} Invalid encoding hint value. Please choose from these options: {values}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} Invalid encoding hint value. Please choose from these options: {values}.")
 
         self.model.Tables[table_name].Columns[column_name].EncodingHint = (
             System.Enum.Parse(TOM.EncodingHintType, value)
@@ -3605,7 +3479,7 @@ class TOMWrapper:
 
     def set_data_type(self, table_name: str, column_name: str, value: str):
         """
-        Sets the data type for a column.
+        Sets the `data type <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.datatype?view=analysisservices-dotnet>`_ for a column.
 
         Parameters
         ----------
@@ -3617,6 +3491,7 @@ class TOMWrapper:
             The data type.
             `Data type valid values <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.datatype?view=analysisservices-dotnet>`_
         """
+        import Microsoft.AnalysisServices.Tabular as TOM
         import System
 
         values = [
@@ -3638,10 +3513,7 @@ class TOMWrapper:
             value = "Boolean"
 
         if value not in values:
-            print(
-                f"{icons.red_dot} Invalid data type. Please choose from these options: {values}."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} Invalid data type. Please choose from these options: {values}.")
 
         self.model.Tables[table_name].Columns[column_name].DataType = System.Enum.Parse(
             TOM.DataType, value
@@ -3673,10 +3545,7 @@ class TOMWrapper:
         for t in time_intel:
             t = t.capitalize()
             if t not in [time_intel_options]:
-                print(
-                    f"The '{t}' time intelligence variation is not supported. Valid options: {time_intel_options}."
-                )
-                return
+                raise ValueError(f"{icons.red_dot} The '{t}' time intelligence variation is not supported. Valid options: {time_intel_options}.")
 
         # Validate measure and extract table name
         for m in self.all_measures():
@@ -3684,17 +3553,11 @@ class TOMWrapper:
                 table_name = m.Parent.Name
 
         if table_name is None:
-            print(
-                f"The '{measure_name}' is not a valid measure in the '{self.dataset}' semantic model within the '{self.workspace}' workspace."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} The '{measure_name}' is not a valid measure in the '{self._dataset}' semantic model within the '{self._workspace}' workspace.")
 
         # Validate date table
         if not self.is_date_table(date_table):
-            print(
-                f"{icons.red_dot} The '{date_table}' table is not a valid date table in the '{self.dataset}' wemantic model within the '{self.workspace}' workspace."
-            )
-            return
+            raise ValueError(f"{icons.red_dot} The '{date_table}' table is not a valid date table in the '{self._dataset}' wemantic model within the '{self._workspace}' workspace.")
 
         # Extract date key from date table
         for c in self.all_columns():
@@ -3712,26 +3575,100 @@ class TOMWrapper:
                     expression=expr,
                 )
 
+    def update_m_partition(self, table_name: str, partition_name: str, expression: Optional[str | None] = None, mode: Optional[str | None] = None, description: Optional[str | None] = None):
+
+        """
+        Updates an M partition for a table within a semantic model.
+
+        Parameters
+        ----------
+        table_name : str
+            Name of the table.       
+        partition_name : str
+            Name of the partition.
+        expression : str, default=None
+            The `M expression <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.mpartitionsource.expression?view=analysisservices-dotnet>`_ containing the logic for the partition.
+            Defaults to None which keeps the existing setting.
+        mode : str, default=None
+            The query `mode <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.modetype?view=analysisservices-dotnet>`_ of the partition.
+            Defaults to None which keeps the existing setting.
+        description : str, default=None
+            The description of the partition.
+            Defaults to None which keeps the existing setting.
+        """
+
+        import Microsoft.AnalysisServices.Tabular as TOM
+        import System
+
+        p = self.model.Tables[table_name].Partitions[partition_name]
+        if p.SourceType != TOM.PartitionSourceType.M:
+            raise ValueError(f"Invalid partition source type. This function is only for M partitions.")
+        if expression is not None:
+            p.Source.Expression = expression
+        if mode is not None:
+            p.Mode = System.Enum.Parse(TOM.ModeType, mode)
+        if description is not None:
+            p.Description = description
+
+    def set_sort_by_column(self, table_name: str, column_name: str, sort_by_column: str):
+
+        """
+        Sets the sort by column for a column in a semantic model.
+
+        Parameters
+        ----------
+        table_name : str
+            Name of the table.       
+        column_name : str
+            Name of the column.
+        sort_by_column : str
+            Name of the column to use for sorting. Must be of integer (Int64) data type.
+        """
+
+        import Microsoft.AnalysisServices.Tabular as TOM
+
+        sbc = self.model.Tables[table_name].Columns[sort_by_column]
+
+        if sbc.DataType != TOM.DataType.Int64:
+            raise ValueError(f"Invalid sort by column data type. The sort by column must be of 'Int64' data type.")
+        
+        self.model.Tables[table_name].Columns[column_name].SortByColumn = sbc
+
+    def remove_sort_by_column(self, table_name: str, column_name: str):
+
+        """
+        Removes the sort by column for a column in a semantic model.
+
+        Parameters
+        ----------
+        table_name : str
+            Name of the table.       
+        column_name : str
+            Name of the column.
+        """
+
+        self.model.Tables[table_name].Columns[column_name].SortByColumn = None
+
     def close(self):
-        if not self.readonly and self.model is not None:
+        if not self._readonly and self.model is not None:
             self.model.SaveChanges()
 
-            if len(self.tables_added) > 0:
+            if len(self._tables_added) > 0:
                 refresh_semantic_model(
-                    dataset=self.dataset,
-                    tables=self.tables_added,
-                    workspace=self.workspace,
+                    dataset=self._dataset,
+                    tables=self._tables_added,
+                    workspace=self._workspace,
                 )
             self.model = None
 
-        self.tom_server.Dispose()
+        self._tom_server.Dispose()
 
 
 @log
 @contextmanager
 def connect_semantic_model(
     dataset: str, readonly: bool = True, workspace: Optional[str] = None
-):
+) -> Iterator[TOMWrapper]:
     """
     Connects to the Tabular Object Model (TOM) within a semantic model.
 
@@ -3748,7 +3685,7 @@ def connect_semantic_model(
 
     Returns
     -------
-    str
+    typing.Iterator[TOMWrapper]
         A connection to the semantic model's Tabular Object Model.
     """
 
@@ -3763,4 +3700,4 @@ def connect_semantic_model(
     try:
         yield tw
     finally:
-        tw.close()
+        tw.close()