dcs-sdk 1.7.2__tar.gz → 1.7.4__tar.gz

{dcs_sdk-1.7.2 → dcs_sdk-1.7.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dcs-sdk
-Version: 1.7.2
+Version: 1.7.4
 Summary: SDK for DataChecks
 Author: Waterdip Labs
 Author-email: hello@waterdip.ai
@@ -86,7 +86,7 @@ Requires-Dist: vertica-python (>=1.4.0) ; extra == "vertica" or extra == "all-db
 Description-Content-Type: text/markdown
 
 <h1 align="center">
-  DCS SDK v1.7.2
+  DCS SDK v1.7.4
 </h1>
 
 > SDK for DataChecks

{dcs_sdk-1.7.2 → dcs_sdk-1.7.4}/README.md

@@ -1,5 +1,5 @@
 <h1 align="center">
-  DCS SDK v1.7.2
+  DCS SDK v1.7.4
 </h1>
 
 > SDK for DataChecks

{dcs_sdk-1.7.2 → dcs_sdk-1.7.4}/dcs_sdk/__version__.py

@@ -12,4 +12,4 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 
-__version__ = "1.7.2"
+__version__ = "1.7.4"

{dcs_sdk-1.7.2 → dcs_sdk-1.7.4}/dcs_sdk/sdk/data_diff/data_differ.py

@@ -738,10 +738,10 @@ class DBTableDiffer:
 
             self.response.update({"column_transforms": column_transforms})
             self.response.update({"schema_overrides": schema_overrides})
-            self.config.source.table = self.original_source_table_name
-            self.config.target.table = self.original_target_table_name
-            self.response["source_dataset"]["table_name"] = self.original_source_table_name
-            self.response["target_dataset"]["table_name"] = self.original_target_table_name
+            self.config.source.table = self.original_source_table_name or self.config.source.table
+            self.config.target.table = self.original_target_table_name or self.config.target.table
+            self.response["source_dataset"]["table_name"] = self.original_source_table_name or self.config.source.table
+            self.response["target_dataset"]["table_name"] = self.original_target_table_name or self.config.target.table
             return self.response
         except Exception as e:
             logger.exception(f"Error during diff_tables: {e}")

{dcs_sdk-1.7.2 → dcs_sdk-1.7.4}/dcs_sdk/sdk/utils/utils.py

@@ -439,27 +439,39 @@ def apply_custom_masking(
 
     # common masking columns
     for col in common_masking_cols:
-        src_val = str(source.get(col, ""))
-        tgt_val = str(target.get(col, ""))
+        src_val = source.get(col)
+        tgt_val = target.get(col)
 
-        src_len, tgt_len = len(src_val), len(tgt_val)
-        if src_len == tgt_len and src_val != tgt_val:
-            masked_source[col] = mask_char * (src_len + 1)
-            masked_target[col] = mask_char * tgt_len
+        if src_val is None and tgt_val is None:
+            continue
+
+        src_str = str(src_val) if src_val is not None else ""
+        tgt_str = str(tgt_val) if tgt_val is not None else ""
+
+        src_len, tgt_len = len(src_str), len(tgt_str)
+        if src_len == tgt_len and src_str != tgt_str:
+            if src_val is not None:
+                masked_source[col] = mask_char * (src_len + 1)
+            if tgt_val is not None:
+                masked_target[col] = mask_char * tgt_len
         else:
-            masked_source[col] = mask_char * src_len
-            masked_target[col] = mask_char * tgt_len
+            if src_val is not None:
+                masked_source[col] = mask_char * src_len
+            if tgt_val is not None:
+                masked_target[col] = mask_char * tgt_len
 
     # Non-common columns
     for col in source_masking_cols:
         if col not in common_masking_cols:
-            val = source.get(col, "")
-            masked_source[col] = mask_char * len(val)
+            val = source.get(col)
+            if val is not None:
+                masked_source[col] = mask_char * len(str(val))
 
     for col in target_masking_cols:
         if col not in common_masking_cols:
-            val = target.get(col, "")
-            masked_target[col] = mask_char * len(val)
+            val = target.get(col)
+            if val is not None:
+                masked_target[col] = mask_char * len(str(val))
 
     return masked_source, masked_target
 

{dcs_sdk-1.7.2 → dcs_sdk-1.7.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "dcs-sdk"
-version = "1.7.2"
+version = "1.7.4"
 description = "SDK for DataChecks"
 authors = ["Waterdip Labs <hello@waterdip.ai>"]
 readme = "README.md"