Merge pull request #763 from dyvenia/add_viadot_source_to_bigquery_so…

…urce Add `_viadot_source` column to `BigQuery`
dyvenia · Oct 12, 2023 · d437df0 · d437df0
2 parents 39cfc03 + da7cce0
commit d437df0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -8,6 +8,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 - Added `Office365-REST-Python-Client` library to `requirements`
 - Added `GetSalesQuotationData` view in `BusinessCore` source.
 - Added new ViewType `queue_interaction_detail_view` to Genesys.
+- Added new column `_viadot_source` to BigQuery extraction.
 
 ### Fixed
 

diff --git a/viadot/sources/bigquery.py b/viadot/sources/bigquery.py
@@ -7,6 +7,7 @@
 from ..config import local_config
 from ..exceptions import CredentialError, DBDataAccessError
 from .base import Source
+from ..utils import add_viadot_metadata_columns
 
 
 class BigQuery(Source):
@@ -100,9 +101,11 @@ def get_project_id(self) -> str:
         """
         return self.credentials["project_id"]
 
+    @add_viadot_metadata_columns()
     def query_to_df(self, query: str) -> pd.DataFrame:
         """
         Query throught Bigquery table.
+        `add_viadot_metadata_columns` adds one column to the DF - `_viadot_source`.
 
         Args:
             query (str): SQL-Like Query to return data values.