semantic-link-labs 0.4.2__py3-none-any.whl → 0.6.0__py3-none-any.whl

sempy_labs/tom/_model.py

@@ -21,7 +21,8 @@ class TOMWrapper:
     """
     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.
+    `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
@@ -93,7 +94,9 @@ 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):
@@ -500,7 +503,8 @@ class TOMWrapper:
         description: Optional[str] = None,
     ):
         """
-        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.
+        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
         ----------
@@ -582,11 +586,15 @@ class TOMWrapper:
         tp.Table = self.model.Tables[table_name]
         tp.FilterExpression = filter_expression
 
-        try:
+        if any(
+            t.Name == table_name and r.Name == role_name
+            for r in self.model.Roles
+            for t in r.TablePermissions
+        ):
             self.model.Roles[role_name].TablePermissions[
                 table_name
             ].FilterExpression = filter_expression
-        except:
+        else:
             self.model.Roles[role_name].TablePermissions.Add(tp)
 
     def set_ols(
@@ -613,17 +621,22 @@ class TOMWrapper:
         permission = permission.capitalize()
 
         if permission not in ["Read", "None", "Default"]:
-            print(f"ERROR! Invalid 'permission' value.")
-            return
+            raise ValueError(f"{icons.red_dot} Invalid 'permission' value.")
 
         cp = TOM.ColumnPermission()
         cp.Column = self.model.Tables[table_name].Columns[column_name]
         cp.MetadataPermission = System.Enum.Parse(TOM.MetadataPermission, permission)
-        try:
+
+        if any(
+            c.Name == column_name and t.Name == table_name and r.Name == role_name
+            for r in self.model.Roles
+            for t in r.TablePermissions
+            for c in t.ColumnPermissions
+        ):
             self.model.Roles[role_name].TablePermissions[table_name].ColumnPermissions[
                 column_name
             ].MetadataPermission = System.Enum.Parse(TOM.MetadataPermission, permission)
-        except:
+        else:
             self.model.Roles[role_name].TablePermissions[
                 table_name
             ].ColumnPermissions.Add(cp)
@@ -658,22 +671,22 @@ class TOMWrapper:
         import Microsoft.AnalysisServices.Tabular as TOM
 
         if isinstance(columns, str):
-            print(
+            raise ValueError(
                 f"{icons.red_dot} The 'levels' parameter must be a list. For example: ['Continent', 'Country', 'City']"
             )
-            return
+
         if len(columns) == 1:
-            print(f"{icons.red_dot} 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(
+            raise ValueError(
                 f"{icons.red_dot} If specifying level names, you must specify a level for each column."
             )
-            return
 
         obj = TOM.Hierarchy()
         obj.Name = hierarchy_name
@@ -864,10 +877,8 @@ class TOMWrapper:
         cul = TOM.Culture()
         cul.Name = language
 
-        try:
+        if not any(c.Name == language for c in self.model.Cultures):
             self.model.Cultures.Add(cul)
-        except:
-            pass
 
     def add_perspective(self, perspective_name: str):
         """
@@ -1003,7 +1014,7 @@ class TOMWrapper:
         import System
 
         if base_column is not None and base_table is None:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} If you specify the base table you must also specify the base column"
             )
 
@@ -1015,10 +1026,9 @@ class TOMWrapper:
 
         summarizationTypes = ["Sum", "GroupBy", "Count", "Min", "Max"]
         if summarization_type not in summarizationTypes:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} The 'summarization_type' parameter must be one of the following valuse: {summarizationTypes}."
             )
-            return
 
         ao = TOM.AlternateOf()
         ao.Summarization = System.Enum.Parse(TOM.SummarizationType, summarization_type)
@@ -1096,9 +1106,9 @@ class TOMWrapper:
         ann.Name = name
         ann.Value = value
 
-        try:
+        if any(a.Name == name for a in object.Annotations):
             object.Annotations[name].Value = value
-        except:
+        else:
             object.Annotations.Add(ann)
 
     def get_annotation_value(self, object, name: str):
@@ -1117,8 +1127,12 @@ class TOMWrapper:
         str
             The annotation value.
         """
+        if any(a.Name == name for a in object.Annotations):
+            value = object.Annotations[name].Value
+        else:
+            value = None
 
-        return object.Annotations[name].Value
+        return value
 
     def remove_annotation(self, object, name: str):
         """
@@ -1196,9 +1210,9 @@ class TOMWrapper:
         ep.Name = name
         ep.Value = value
 
-        try:
+        if any(a.Name == name for a in object.Annotations):
             object.ExtendedProperties[name].Value = value
-        except:
+        else:
             object.ExtendedProperties.Add(ep)
 
     def get_extended_property_value(self, object, name: str):
@@ -1217,8 +1231,12 @@ class TOMWrapper:
         str
             The extended property value.
         """
+        if any(a.Name == name for a in object.ExtendedProperties):
+            value = object.ExtendedProperties[name].Value
+        else:
+            value = None
 
-        return object.ExtendedProperties[name].Value
+        return value
 
     def remove_extended_property(self, object, name: str):
         """
@@ -1277,10 +1295,9 @@ class TOMWrapper:
         objectType = object.ObjectType
 
         if objectType not in validObjects:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Only the following object types are valid for perspectives: {validObjects}."
             )
-            return
 
         object.Model.Perspectives[perspective_name]
 
@@ -1302,7 +1319,7 @@ class TOMWrapper:
                     object.Parent.Name
                 ].PerspectiveHierarchies[object.Name]
             return True
-        except:
+        except Exception:
             return False
 
     def add_to_perspective(
@@ -1331,17 +1348,17 @@ class TOMWrapper:
         objectType = object.ObjectType
 
         if objectType not in validObjects:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Only the following object types are valid for perspectives: {validObjects}."
             )
-            return
-        try:
+
+        if any(p.Name == perspective_name for p in self.model.Perspectives):
             object.Model.Perspectives[perspective_name]
-        except:
-            print(f"{icons.red_dot} The '{perspective_name}' perspective does not exist.")
-            return
+        else:
+            raise ValueError(
+                f"{icons.red_dot} The '{perspective_name}' perspective does not exist."
+            )
 
-        # try:
         if objectType == TOM.ObjectType.Table:
             pt = TOM.PerspectiveTable()
             pt.Table = object
@@ -1364,8 +1381,6 @@ class TOMWrapper:
             object.Model.Perspectives[perspective_name].PerspectiveTables[
                 object.Parent.Name
             ].PerspectiveHierarchies.Add(ph)
-        # except:
-        #    pass
 
     def remove_from_perspective(
         self,
@@ -1393,17 +1408,15 @@ class TOMWrapper:
         objectType = object.ObjectType
 
         if objectType not in validObjects:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Only the following object types are valid for perspectives: {validObjects}."
             )
-            return
-        try:
-            object.Model.Perspectives[perspective_name]
-        except:
-            print(f"{icons.red_dot} The '{perspective_name}' perspective does not exist.")
-            return
 
-        # try:
+        if not any(p.Name == perspective_name for p in self.model.Perspectives):
+            raise ValueError(
+                f"{icons.red_dot} The '{perspective_name}' perspective does not exist."
+            )
+
         if objectType == TOM.ObjectType.Table:
             pt = object.Model.Perspectives[perspective_name].PerspectiveTables[
                 object.Name
@@ -1436,12 +1449,12 @@ class TOMWrapper:
             object.Model.Perspectives[perspective_name].PerspectiveTables[
                 object.Parent.Name
             ].PerspectiveHierarchies.Remove(ph)
-        # except:
-        #    pass
 
     def set_translation(
         self,
-        object: Union["TOM.Table", "TOM.Column", "TOM.Measure", "TOM.Hierarchy"],
+        object: Union[
+            "TOM.Table", "TOM.Column", "TOM.Measure", "TOM.Hierarchy", "TOM.Level"
+        ],
         language: str,
         property: str,
         value: str,
@@ -1471,11 +1484,13 @@ class TOMWrapper:
             TOM.ObjectType.Column,
             TOM.ObjectType.Measure,
             TOM.ObjectType.Hierarchy,
-        ]  # , 'Level'
+            TOM.ObjectType.Level,
+        ]
 
         if object.ObjectType not in validObjects:
-            print(f"{icons.red_dot} 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,
@@ -1484,22 +1499,38 @@ class TOMWrapper:
         }
 
         prop = mapping.get(property)
+        if prop is 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(
+        if not any(c.Name == language for c in self.model.Cultures):
+            raise ValueError(
                 f"{icons.red_dot} The '{language}' translation language does not exist in the semantic model."
             )
-            return
 
         object.Model.Cultures[language].ObjectTranslations.SetTranslation(
             object, prop, value
         )
 
+        if object.ObjectType in [TOM.ObjectType.Table, TOM.ObjectType.Measure]:
+            print(
+                f"{icons.green_dot} The {property} property for the '{object.Name}' {str(object.ObjectType).lower()} has been translated into '{language}' as '{value}'."
+            )
+        elif object.ObjectType in [
+            TOM.ObjectType.Column,
+            TOM.ObjectType.Hierarchy,
+            TOM.ObjectType.Level,
+        ]:
+            print(
+                f"{icons.green_dot} The {property} property for the '{object.Parent.Name}'[{object.Name}] {str(object.ObjectType).lower()} has been translated into '{language}' as '{value}'."
+            )
+
     def remove_translation(
         self,
-        object: Union["TOM.Table", "TOM.Column", "TOM.Measure", "TOM.Hierarchy"],
+        object: Union[
+            "TOM.Table", "TOM.Column", "TOM.Measure", "TOM.Hierarchy", "TOM.Level"
+        ],
         language: str,
     ):
         """
@@ -1537,7 +1568,7 @@ class TOMWrapper:
             for lang in object.Model.Cultures:
                 try:
                     self.remove_translation(object=object, language=lang.Name)
-                except:
+                except Exception:
                     pass
         if objType in ["Table", "Column", "Measure", "Hierarchy"]:
             for persp in object.Model.Perspectives:
@@ -1545,7 +1576,7 @@ class TOMWrapper:
                     self.remove_from_perspective(
                         object=object, perspective_name=persp.Name
                     )
-                except:
+                except Exception:
                     pass
 
         if objType == TOM.ObjectType.Column:
@@ -1653,12 +1684,12 @@ class TOMWrapper:
         objType = column.ObjectType
 
         if objType == TOM.ObjectType.Column:
-            for l in self.all_levels():
+            for lev in self.all_levels():
                 if (
-                    l.Parent.Table.Name == column.Parent.Name
-                    and l.Column.Name == column.Name
+                    lev.Parent.Table.Name == column.Parent.Name
+                    and lev.Column.Name == column.Name
                 ):
-                    yield l.Parent
+                    yield lev.Parent
 
     def used_in_sort_by(self, column: "TOM.Column"):
         """
@@ -1851,7 +1882,7 @@ class TOMWrapper:
                 if m.Name in meas:
                     yield m
 
-    def hybrid_tables(self):
+    def all_hybrid_tables(self):
         """
         Outputs the `hybrid tables <https://learn.microsoft.com/power-bi/connect-data/service-dataset-modes-understand#hybrid-tables>`_ within a semantic model.
 
@@ -1870,7 +1901,7 @@ class TOMWrapper:
                 if any(p.Mode == TOM.ModeType.DirectQuery for p in t.Partitions):
                     yield t
 
-    def date_tables(self):
+    def all_date_tables(self):
         """
         Outputs the tables which are marked as `date tables <https://learn.microsoft.com/power-bi/transform-model/desktop-date-tables>`_ within a semantic model.
 
@@ -1937,7 +1968,11 @@ class TOMWrapper:
         """
         import Microsoft.AnalysisServices.Tabular as TOM
 
-        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')
+        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):
         """
@@ -1955,10 +1990,9 @@ class TOMWrapper:
         t = self.model.Tables[table_name]
         c = t.Columns[column_name]
         if c.DataType != TOM.DataType.DateTime:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} The column specified in the 'column_name' parameter in this function must be of DateTime data type."
             )
-            return
 
         daxQuery = f"""
         define measure '{table_name}'[test] = 
@@ -1978,10 +2012,9 @@ class TOMWrapper:
         )
         value = df["1"].iloc[0]
         if value != "1":
-            print(
+            raise ValueError(
                 f"{icons.red_dot} The '{column_name}' within the '{table_name}' table does not contain contiguous date values."
             )
-            return
 
         # Mark as a date table
         t.DataCategory = "Time"
@@ -2035,7 +2068,7 @@ class TOMWrapper:
         -------
         bool
             Indicates if the semantic model has a hybrid table.
-        """    
+        """
 
         return any(self.is_hybrid_table(table_name=t.Name) for t in self.model.Tables)
 
@@ -2176,15 +2209,19 @@ class TOMWrapper:
         # https://github.com/m-kovalsky/Tabular/blob/master/KPI%20Graphics.md
 
         if measure_name == target:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} The 'target' parameter cannot be the same measure as the 'measure_name' parameter."
             )
-            return
 
         if status_graphic is None:
             status_graphic = "Three Circles Colored"
 
-        valid_status_types = ["Linear", "LinearReversed", "Centered", "CenteredReversed"]
+        valid_status_types = [
+            "Linear",
+            "LinearReversed",
+            "Centered",
+            "CenteredReversed",
+        ]
         status_type = status_type
         if status_type is None:
             status_type = "Linear"
@@ -2192,43 +2229,47 @@ class TOMWrapper:
             status_type = status_type.title().replace(" ", "")
 
         if status_type not in valid_status_types:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} '{status_type}' is an invalid status_type. Please choose from these options: {valid_status_types}."
             )
-            return
 
         if status_type in ["Linear", "LinearReversed"]:
             if upper_bound is not None or lower_mid_bound is not None:
-                print(
+                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."
                 )
-                return
+
             elif upper_bound <= lower_bound:
-                print(f"{icons.red_dot} 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(
+                raise ValueError(
                     f"{icons.red_dot} The 'upper_mid_bound' and 'lower_mid_bound' parameters are necessary in the 'Centered' and 'CenteredReversed' status types."
                 )
-                return
             elif upper_bound <= upper_mid_bound:
-                print(f"{icons.red_dot} 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"{icons.red_dot} 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"{icons.red_dot} 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(
+        except Exception:
+            raise ValueError(
                 f"{icons.red_dot} The '{measure_name}' measure does not exist in the '{self._dataset}' semantic model within the '{self._workspace}'."
             )
-            return
 
         graphics = [
             "Cylinder",
@@ -2251,10 +2292,9 @@ class TOMWrapper:
         ]
 
         if status_graphic not in graphics:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} The '{status_graphic}' status graphic is not valid. Please choose from these options: {graphics}."
             )
-            return
 
         measure_target = True
 
@@ -2262,15 +2302,15 @@ class TOMWrapper:
             float(target)
             tgt = str(target)
             measure_target = False
-        except:
+        except Exception:
             try:
                 tgt = next(
                     format_dax_object_name(m.Parent.Name, m.Name)
                     for m in self.all_measures()
                     if m.Name == target
                 )
-            except:
-                print(
+            except Exception:
+                raise ValueError(
                     f"{icons.red_dot} The '{target}' measure does not exist in the '{self._dataset}' semantic model within the '{self._workspace}'."
                 )
 
@@ -2294,11 +2334,11 @@ class TOMWrapper:
         kpi.StatusExpression = expr
 
         ms = self.model.Tables[table_name].Measures[measure_name]
-        try:
+        if ms.KPI is not None:
             ms.KPI.TargetExpression = tgt
             ms.KPI.StatusGraphic = status_graphic
             ms.KPI.StatusExpression = expr
-        except:
+        else:
             ms.KPI = kpi
 
     def set_aggregations(self, table_name: str, agg_table_name: str):
@@ -2317,6 +2357,8 @@ class TOMWrapper:
 
         """
 
+        import Microsoft.AnalysisServices.Tabular as TOM
+
         for c in self.model.Tables[agg_table_name].Columns:
 
             dataType = c.DataType
@@ -2354,7 +2396,7 @@ class TOMWrapper:
             The IsAvailableInMdx property value.
         """
 
-        self.model.Tables[table_name].Columns[column_name].IsAvailableInMdx = value
+        self.model.Tables[table_name].Columns[column_name].IsAvailableInMDX = value
 
     def set_summarize_by(
         self, table_name: str, column_name: str, value: Optional[str] = None
@@ -2373,6 +2415,7 @@ class TOMWrapper:
             Defaults to none which resolves to 'Default'.
             `Aggregate valid values <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.aggregatefunction?view=analysisservices-dotnet>`_
         """
+        import Microsoft.AnalysisServices.Tabular as TOM
         import System
 
         values = [
@@ -2396,10 +2439,9 @@ class TOMWrapper:
         )
 
         if value not in values:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} '{value}' is not a valid value for the SummarizeBy property. These are the valid values: {values}."
             )
-            return
 
         self.model.Tables[table_name].Columns[column_name].SummarizeBy = (
             System.Enum.Parse(TOM.AggregateFunction, value)
@@ -2415,6 +2457,7 @@ class TOMWrapper:
             The DirectLakeBehavior property value.
             `DirectLakeBehavior valid values <https://learn.microsoft.com/dotnet/api/microsoft.analysisservices.tabular.directlakebehavior?view=analysisservices-dotnet>`_
         """
+        import Microsoft.AnalysisServices.Tabular as TOM
         import System
 
         direct_lake_behavior = direct_lake_behavior.capitalize()
@@ -2428,10 +2471,9 @@ class TOMWrapper:
         dlValues = ["Automatic", "DirectLakeOnly", "DirectQueryOnly"]
 
         if direct_lake_behavior not in dlValues:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} The 'direct_lake_behavior' parameter must be one of these values: {dlValues}."
             )
-            return
 
         self.model.DirectLakeBehavior = System.Enum.Parse(
             TOM.DirectLakeBehavior, direct_lake_behavior
@@ -2529,13 +2571,14 @@ class TOMWrapper:
         import Microsoft.AnalysisServices.Tabular as TOM
 
         if isinstance(objects, str):
-            print(f"{icons.red_dot} 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(
+            raise ValueError(
                 f"{icons.red_dot} There must be more than one object (column/measure) within the objects parameter."
             )
-            return
 
         expr = ""
         i = 0
@@ -2571,10 +2614,9 @@ class TOMWrapper:
                     )
                     success = True
             if not success:
-                print(
+                raise ValueError(
                     f"{icons.red_dot} The '{obj}' object was not found in the '{self._dataset}' semantic model."
                 )
-                return
             else:
                 i += 1
 
@@ -2742,7 +2784,7 @@ class TOMWrapper:
         try:
             runId = self.get_annotation_value(object=self.model, name="Vertipaq_Run")
             runId = str(int(runId) + 1)
-        except:
+        except Exception:
             runId = "1"
         self.set_annotation(object=self.model, name="Vertipaq_Run", value=runId)
 
@@ -2771,7 +2813,7 @@ class TOMWrapper:
                 object=object, name="Vertipaq_RecordCount"
             )
 
-        return int(result)
+        return int(result) if result is not None else 0
 
     def records_per_segment(self, object: "TOM.Partition"):
         """
@@ -2796,7 +2838,7 @@ class TOMWrapper:
                 object=object, name="Vertipaq_RecordsPerSegment"
             )
 
-        return float(result)
+        return float(result) if result is not None else 0
 
     def used_size(self, object: Union["TOM.Hierarchy", "TOM.Relationship"]):
         """
@@ -2821,7 +2863,7 @@ class TOMWrapper:
         elif objType == TOM.ObjectType.Relationship:
             result = self.get_annotation_value(object=object, name="Vertipaq_UsedSize")
 
-        return int(result)
+        return int(result) if result is not None else 0
 
     def data_size(self, column: "TOM.Column"):
         """
@@ -2844,7 +2886,7 @@ class TOMWrapper:
         if objType == TOM.ObjectType.Column:
             result = self.get_annotation_value(object=column, name="Vertipaq_DataSize")
 
-        return int(result)
+        return int(result) if result is not None else 0
 
     def dictionary_size(self, column: "TOM.Column"):
         """
@@ -2869,7 +2911,7 @@ class TOMWrapper:
                 object=column, name="Vertipaq_DictionarySize"
             )
 
-        return int(result)
+        return int(result) if result is not None else 0
 
     def total_size(self, object: Union["TOM.Table", "TOM.Column"]):
         """
@@ -2894,7 +2936,7 @@ class TOMWrapper:
         elif objType == TOM.ObjectType.Table:
             result = self.get_annotation_value(object=object, name="Vertipaq_TotalSize")
 
-        return int(result)
+        return int(result) if result is not None else 0
 
     def cardinality(self, column: "TOM.Column"):
         """
@@ -2919,7 +2961,7 @@ class TOMWrapper:
                 object=column, name="Vertipaq_Cardinality"
             )
 
-        return int(result)
+        return int(result) if result is not None else 0
 
     def depends_on(self, object, dependencies: pd.DataFrame):
         """
@@ -2947,7 +2989,7 @@ class TOMWrapper:
             objParentName = objName
 
         fil = dependencies[
-            (dependencies["Object Type"] == objType)
+            (dependencies["Object Type"] == str(objType))
             & (dependencies["Table Name"] == objParentName)
             & (dependencies["Object Name"] == objName)
         ]
@@ -3004,7 +3046,7 @@ class TOMWrapper:
             objParentName = objName
 
         fil = dependencies[
-            (dependencies["Referenced Object Type"] == objType)
+            (dependencies["Referenced Object Type"] == str(objType))
             & (dependencies["Referenced Table"] == objParentName)
             & (dependencies["Referenced Object"] == objName)
         ]
@@ -3051,7 +3093,7 @@ class TOMWrapper:
 
         for obj in self.depends_on(object=object, dependencies=dependencies):
             if obj.ObjectType == TOM.ObjectType.Measure:
-                if (obj.Parent.Name + obj.Name in object.Expression) or (
+                if (f"{obj.Parent.Name}[{obj.Name}]" in object.Expression) or (
                     format_dax_object_name(obj.Parent.Name, obj.Name)
                     in object.Expression
                 ):
@@ -3075,15 +3117,22 @@ class TOMWrapper:
         """
         import Microsoft.AnalysisServices.Tabular as TOM
 
-        def create_pattern(a, b):
-            return r"(?<!" + re.escape(a) + r"\[)(?<!" + re.escape(a) + r"'\[)" + re.escape(b)
+        def create_pattern(tableList, b):
+            patterns = [
+                r"(?<!" + re.escape(table) + r"\[)(?<!" + re.escape(table) + r"'\[)"
+                for table in tableList
+            ]
+            combined_pattern = "".join(patterns) + re.escape(b)
+            return combined_pattern
 
         for obj in self.depends_on(object=object, dependencies=dependencies):
             if obj.ObjectType == TOM.ObjectType.Column:
+                tableList = []
+                for c in self.all_columns():
+                    if c.Name == obj.Name:
+                        tableList.append(c.Parent.Name)
                 if (
-                    re.search(
-                        create_pattern(obj.Parent.Name, obj.Name), object.Expression
-                    )
+                    re.search(create_pattern(tableList, obj.Name), object.Expression)
                     is not None
                 ):
                     yield obj
@@ -3244,26 +3293,24 @@ class TOMWrapper:
         rolling_window_granularity = rolling_window_granularity.capitalize()
 
         if incremental_granularity not in incGran:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Invalid 'incremental_granularity' value. Please choose from the following options: {incGran}."
             )
-            return
+
         if rolling_window_granularity not in incGran:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Invalid 'rolling_window_granularity' value. Please choose from the following options: {incGran}."
             )
-            return
 
         if rolling_window_periods < 1:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Invalid 'rolling_window_periods' value. Must be a value greater than 0."
             )
-            return
+
         if incremental_periods < 1:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Invalid 'incremental_periods' value. Must be a value greater than 0."
             )
-            return
 
         t = self.model.Tables[table_name]
 
@@ -3271,10 +3318,9 @@ class TOMWrapper:
             dc = t.Columns[detect_data_changes_column]
 
             if dc.DataType != TOM.DataType.DateTime:
-                print(
+                raise ValueError(
                     f"{icons.red_dot} Invalid 'detect_data_changes_column' parameter. This column must be of DateTime data type."
                 )
-                return
 
         rp = TOM.BasicRefreshPolicy()
         rp.IncrementalPeriods = incremental_periods
@@ -3354,26 +3400,24 @@ class TOMWrapper:
         rolling_window_granularity = rolling_window_granularity.capitalize()
 
         if incremental_granularity not in incGran:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Invalid 'incremental_granularity' value. Please choose from the following options: {incGran}."
             )
-            return
+
         if rolling_window_granularity not in incGran:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Invalid 'rolling_window_granularity' value. Please choose from the following options: {incGran}."
             )
-            return
 
         if rolling_window_periods < 1:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Invalid 'rolling_window_periods' value. Must be a value greater than 0."
             )
-            return
+
         if incremental_periods < 1:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Invalid 'incremental_periods' value. Must be a value greater than 0."
             )
-            return
 
         date_format = "%m/%d/%Y"
 
@@ -3388,10 +3432,9 @@ class TOMWrapper:
         end_day = date_obj_end.day
 
         if date_obj_end <= date_obj_start:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Invalid 'start_date' or 'end_date'. The 'end_date' must be after the 'start_date'."
             )
-            return
 
         t = self.model.Tables[table_name]
 
@@ -3400,20 +3443,18 @@ class TOMWrapper:
         dType = c.DataType
 
         if dType != TOM.DataType.DateTime:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} The {fcName} column is of '{dType}' data type. The column chosen must be of DateTime data type."
             )
-            return
 
         if detect_data_changes_column is not None:
             dc = t.Columns[detect_data_changes_column]
             dcType = dc.DataType
 
             if dcType != TOM.DataType.DateTime:
-                print(
+                raise ValueError(
                     f"{icons.red_dot} Invalid 'detect_data_changes_column' parameter. This column must be of DateTime data type."
                 )
-                return
 
         # Start changes:
 
@@ -3421,10 +3462,10 @@ class TOMWrapper:
         i = 0
         for p in t.Partitions:
             if p.SourceType != TOM.PartitionSourceType.M:
-                print(
+                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."
                 )
-                return
+
             elif i == 0:
                 text = p.Expression
                 text = text.rstrip()
@@ -3440,8 +3481,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
@@ -3537,15 +3577,13 @@ class TOMWrapper:
         ht = self.is_hybrid_table(table_name=table_name)
 
         if not ht:
-            print(
+            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}."
             )
-            return
         if p.Mode != TOM.ModeType.DirectQuery:
-            print(
+            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}."
             )
-            return
 
         dcd = TOM.DataCoverageDefinition()
         dcd.Expression = expression
@@ -3572,10 +3610,9 @@ class TOMWrapper:
         value = value.capitalize()
 
         if value not in values:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Invalid encoding hint value. Please choose from these options: {values}."
             )
-            return
 
         self.model.Tables[table_name].Columns[column_name].EncodingHint = (
             System.Enum.Parse(TOM.EncodingHintType, value)
@@ -3617,10 +3654,9 @@ class TOMWrapper:
             value = "Boolean"
 
         if value not in values:
-            print(
+            raise ValueError(
                 f"{icons.red_dot} Invalid data type. Please choose from these options: {values}."
             )
-            return
 
         self.model.Tables[table_name].Columns[column_name].DataType = System.Enum.Parse(
             TOM.DataType, value
@@ -3652,44 +3688,369 @@ class TOMWrapper:
         for t in time_intel:
             t = t.capitalize()
             if t not in [time_intel_options]:
-                print(
+                raise ValueError(
                     f"{icons.red_dot} The '{t}' time intelligence variation is not supported. Valid options: {time_intel_options}."
                 )
-                return
 
         # Validate measure and extract table name
-        for m in self.all_measures():
-            if m.Name == measure_name:
-                table_name = m.Parent.Name
+        matching_measures = [
+            m.Parent.Name for m in self.all_measures() if m.Name == measure_name
+        ]
 
         if table_name is None:
-            print(
+            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."
             )
-            return
 
+        table_name = matching_measures[0]
         # Validate date table
         if not self.is_date_table(date_table):
-            print(
+            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."
             )
-            return
 
         # Extract date key from date table
-        for c in self.all_columns():
-            if c.Parent.Name == date_table and c.IsKey:
-                date_key = c.Name
+        matching_columns = [
+            c.Name
+            for c in self.all_columns()
+            if c.Parent.Name == date_table and c.IsKey
+        ]
+
+        if not matching_columns:
+            raise ValueError(
+                f"{icons.red_dot} The '{date_table}' table does not have a date key column in the '{self._dataset}' semantic model within the '{self._workspace}' workspace."
+            )
+
+        date_key = matching_columns[0]
 
         # Create the new time intelligence measures
         for t in time_intel:
-            if t == "MTD":
-                expr = f"CALCULATE([{measure_name}],DATES{time_intel}('{date_table}'[{date_key}]))"
-                new_meas_name = f"{measure_name} {t}"
-                self.add_measure(
-                    table_name=table_name,
-                    measure_name=new_meas_name,
-                    expression=expr,
-                )
+            expr = f"CALCULATE([{measure_name}],DATES{t}('{date_table}'[{date_key}]))"
+            new_meas_name = f"{measure_name} {t}"
+            self.add_measure(
+                table_name=table_name,
+                measure_name=new_meas_name,
+                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"{icons.red_dot} 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 update_measure(
+        self,
+        measure_name: str,
+        expression: Optional[str | None] = None,
+        format_string: Optional[str | None] = None,
+        hidden: Optional[bool | None] = None,
+        description: Optional[str | None] = None,
+        display_folder: Optional[str | None] = None,
+    ):
+        """
+        Updates a measure within a semantic model.
+
+        Parameters
+        ----------
+        measure_name : str
+            Name of the measure.
+        expression : str, default=None
+            DAX expression of the measure.
+            Defaults to None which keeps the existing setting.
+        format_string : str, default=None
+            Format string of the measure.
+            Defaults to None which keeps the existing setting.
+        hidden : bool, default=None
+            Whether the measure will be hidden or visible.
+            Defaults to None which keeps the existing setting.
+        description : str, default=None
+            A description of the measure.
+            Defaults to None which keeps the existing setting.
+        display_folder : str, default=None
+            The display folder in which the measure will reside.
+            Defaults to None which keeps the existing setting.
+        """
+
+        table_name = next(
+            m.Parent.Name for m in self.all_measures() if m.Name == measure_name
+        )
+        m = self.model.Tables[table_name].Measures[measure_name]
+        if expression is not None:
+            m.Expression = expression
+        if format_string is not None:
+            m.FormatString = format_string
+        if hidden is not None:
+            m.IsHidden = hidden
+        if description is not None:
+            m.Description = description
+        if display_folder is not None:
+            m.DisplayFolder = display_folder
+
+    def update_column(
+        self,
+        table_name: str,
+        column_name: str,
+        source_column: Optional[str | None] = None,
+        data_type: Optional[str | None] = None,
+        expression: Optional[str | None] = None,
+        format_string: Optional[str | None] = None,
+        hidden: Optional[bool | None] = None,
+        description: Optional[str | None] = None,
+        display_folder: Optional[str | None] = None,
+        data_category: Optional[str | None] = None,
+        key: Optional[bool | None] = None,
+        summarize_by: Optional[str | None] = None,
+    ):
+        """
+        Updates a column within a semantic model.
+
+        Parameters
+        ----------
+        table_name : str
+            Name of the table in which the column exists.
+        column_name : str
+            Name of the column.
+        source_column : str, default=None
+            The source column for the column (for data columns only).
+            Defaults to None which keeps the existing setting.
+        data_type : str, default=None
+            The data type of the column.
+            Defaults to None which keeps the existing setting.
+        expression : str, default=None
+            The DAX expression of the column (for calculated columns only).
+            Defaults to None which keeps the existing setting.
+        format_string : str, default=None
+            Format string of the column.
+            Defaults to None which keeps the existing setting.
+        hidden : bool, default=None
+            Whether the column will be hidden or visible.
+            Defaults to None which keeps the existing setting.
+        description : str, default=None
+            A description of the column.
+            Defaults to None which keeps the existing setting.
+        display_folder : str, default=None
+            The display folder in which the column will reside.
+            Defaults to None which keeps the existing setting.
+        data_category : str, default=None
+            The data category of the column.
+            Defaults to None which keeps the existing setting.
+        key : bool, default=False
+            Marks the column as the primary key of the table.
+            Defaults to None which keeps the existing setting.
+        summarize_by : str, default=None
+            Sets the value for the Summarize By property of the column.
+            Defaults to None which keeps the existing setting.
+        """
+
+        import Microsoft.AnalysisServices.Tabular as TOM
+        import System
+
+        c = self.model.Tables[table_name].Measures[column_name]
+        if c.Type == TOM.ColumnType.Data:
+            if source_column is not None:
+                c.SourceColumn = source_column
+        if c.Type == TOM.ColumnType.Calculated:
+            if expression is not None:
+                c.Expression = expression
+        if data_type is not None:
+            c.DataType = System.Enum.Parse(TOM.DataType, data_type)
+        if format_string is not None:
+            c.FormatString = format_string
+        if hidden is not None:
+            c.IsHidden = hidden
+        if description is not None:
+            c.Description = description
+        if display_folder is not None:
+            c.DisplayFolder = display_folder
+        if key is not None:
+            c.IsKey = key
+        if data_category is not None:
+            c.DataCategory = data_category
+        if summarize_by is not None:
+            c.SummarizeBy = System.Enum.Parse(TOM.AggregateFunction, summarize_by)
+
+    def update_role(
+        self,
+        role_name: str,
+        model_permission: Optional[str | None] = None,
+        description: Optional[str | None] = None,
+    ):
+        """
+        Updates a role within a semantic model.
+
+        Parameters
+        ----------
+        role_name : str
+            Name of the role.
+        model_permission : str, default=None
+            The model permission for the role.
+            Defaults to None which keeps the existing setting.
+        description : str, default=None
+            The description of the role.
+            Defaults to None which keeps the existing setting.
+        """
+
+        import Microsoft.AnalysisServices.Tabular as TOM
+        import System
+
+        obj = self.model.Roles[role_name]
+
+        if model_permission is not None:
+            obj.ModelPermission = System.Enum.Parse(
+                TOM.ModelPermission, model_permission
+            )
+        if description is not None:
+            obj.Description = description
+
+    def update_calculation_item(
+        self,
+        table_name: str,
+        calculation_item_name: str,
+        expression: Optional[str | None] = None,
+        ordinal: Optional[int | None] = None,
+        format_string_expression: Optional[str | None] = None,
+        description: Optional[str | None] = None,
+    ):
+        """
+        Updates a calculation item within a semantic model.
+
+        Parameters
+        ----------
+        table_name : str
+            Name of the calculation group (table).
+        calculation_item_name : str
+            Name of the calculation item.
+        expression : str, default=None
+            The DAX expression of the calculation item.
+            Defaults to None which keeps the existing setting.
+        ordinal : int, default=None
+            The ordinal of the calculation item.
+            Defaults to None which keeps the existing setting.
+        format_string_expression : str, default=None
+            The format string expression for the calculation item.
+            Defaults to None which keeps the existing setting.
+        description : str, default=None
+            The description of the role.
+            Defaults to None which keeps the existing setting.
+        """
+
+        obj = self.Tables[table_name].CalculationGroup.CalculationItems[
+            calculation_item_name
+        ]
+
+        if expression is not None:
+            obj.Expression = expression
+        if format_string_expression is not None:
+            obj.FormatStringDefinition.Expression = format_string_expression
+        if ordinal is not None:
+            obj.Ordinal = ordinal
+        if description is not None:
+            obj.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"{icons.red_dot} 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 is_calculated_table(self, table_name: str):
+        """
+        Identifies if a table is a calculated table.
+
+        Parameters
+        ----------
+        table_name : str
+            Name of the table.
+
+        Returns
+        -------
+        bool
+            A boolean value indicating whether the table is a calculated table.
+        """
+
+        import Microsoft.AnalysisServices.Tabular as TOM
+
+        isCalcTable = False
+        t = self.model.Tables[table_name]
+        if t.ObjectType == TOM.ObjectType.Table:
+            if any(
+                p.SourceType == TOM.PartitionSourceType.Calculated for p in t.Partitions
+            ):
+                isCalcTable = True
+        return isCalcTable
 
     def close(self):
         if not self._readonly and self.model is not None: