ingestr 0.13.89__tar.gz → 0.13.90__tar.gz

{ingestr-0.13.89 → ingestr-0.13.90}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ingestr
-Version: 0.13.89
+Version: 0.13.90
 Summary: ingestr is a command-line application that ingests data from various sources and stores them in any database.
 Project-URL: Homepage, https://github.com/bruin-data/ingestr
 Project-URL: Issues, https://github.com/bruin-data/ingestr/issues

ingestr-0.13.90/ingestr/src/buildinfo.py

@@ -0,0 +1 @@
+version = "v0.13.90"

{ingestr-0.13.89 → ingestr-0.13.90}/ingestr/src/smartsheets/__init__.py

@@ -48,7 +48,7 @@ def _get_sheet_data(smartsheet_client: smartsheet.Smartsheet, sheet_id: int):
     # Transform rows to a list of dictionaries
     column_titles = [col.title for col in sheet.columns]
     for row in sheet.rows:
-        row_data = {}
+        row_data = {"_row_id": row.id}
         for i, cell in enumerate(row.cells):
             row_data[column_titles[i]] = cell.value
         yield row_data

{ingestr-0.13.89 → ingestr-0.13.90}/ingestr/tests/unit/test_smartsheets.py

@@ -60,8 +60,8 @@ class TestSmartsheetSource(unittest.TestCase):
         resource = smartsheet_source(access_token="test_token", sheet_id="123")
         data = list(resource)
         self.assertEqual(len(data), 2)
-        self.assertEqual(data[0], {"Col A": "r1c1", "Col B": "r1c2"})
-        self.assertEqual(data[1], {"Col A": "r2c1", "Col B": "r2c2"})
+        self.assertEqual(data[0], {"_row_id": 101, "Col A": "r1c1", "Col B": "r1c2"})
+        self.assertEqual(data[1], {"_row_id": 102, "Col A": "r2c1", "Col B": "r2c2"})
 
         mock_smartsheet_client.assert_called_once_with("test_token")
         mock_client_instance.Sheets.get_sheet.assert_any_call(
@@ -127,8 +127,8 @@ class TestSmartsheetSource(unittest.TestCase):
         data = list(data_generator)
 
         self.assertEqual(len(data), 2)
-        self.assertEqual(data[0], {"ID": 1, "Value": "Alpha"})
-        self.assertEqual(data[1], {"ID": 2, "Value": "Beta"})
+        self.assertEqual(data[0], {"_row_id": 201, "ID": 1, "Value": "Alpha"})
+        self.assertEqual(data[1], {"_row_id": 202, "ID": 2, "Value": "Beta"})
         mock_smartsheet_client_instance.Sheets.get_sheet.assert_called_once_with(456)
 
 

ingestr-0.13.89/ingestr/src/buildinfo.py

@@ -1 +0,0 @@
-version = "v0.13.89"