Revert "Removed whitespaces removal (#1065)" (#1085)

This reverts commit 105f461.
dyvenia · Oct 9, 2024 · c8837fd · c8837fd
1 parent d406f2f
commit c8837fd
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/viadot/sources/sap_rfc.py b/src/viadot/sources/sap_rfc.py
@@ -1136,6 +1136,12 @@ def to_df(self, tests: dict | None = None) -> pd.DataFrame:  # noqa: C901, PLR09
                     ):
                         df_tmp = pd.DataFrame(columns=fields)
                         df_tmp[fields] = records
+                        # SAP adds whitespaces to the first extracted column value.
+                        # If whitespace is in unique column, it must be removed to make
+                        # a proper merge.
+                        for col in self.rfc_unique_id:
+                            df_tmp[col] = df_tmp[col].str.strip()
+                            df[col] = df[col].str.strip()
                         df = pd.merge(df, df_tmp, on=self.rfc_unique_id, how="outer")
                     elif not start:
                         df[fields] = records