style: format code with Autopep8, Black and Ruff Formatter

This commit fixes the style issues introduced in 4f629e5 according to the output
from Autopep8, Black and Ruff Formatter.

Details: https://github.com/ajschofield/de-project-bentley/pull/65

2 files changed, 23 insertions, 19 deletions

diff --git a/src/extract_lambda.py b/src/extract_lambda.py
index 5a5a631..9b17ef2 100644
--- a/src/extract_lambda.py
+++ b/src/extract_lambda.py
@@ -151,9 +151,9 @@ def process_and_upload_tables(db, existing_files, client=boto3.client("s3")):
         table_name = table[0]
         rows = db.run(
             f"SELECT * FROM {identifier(table_name)} WHERE last_updated >= :latest;",
-            latest={datetime.strftime(latest_timestamp, "%Y-%m-%d %H:%M:%S")},  
+            latest={datetime.strftime(latest_timestamp, "%Y-%m-%d %H:%M:%S")},
         )
-        print('rows', rows)
+        print("rows", rows)
         # Creating a temporary file path and writing the column name to it followed by each row of data
         if rows:
             csv_file_path = f"/tmp/{table_name}.csv"
@@ -183,7 +183,5 @@ def process_and_upload_tables(db, existing_files, client=boto3.client("s3")):
                 logger.error(f"Error uploading to S3: {e}")
         else:
             load_status["no change"].append(table_name)
-            logger.info(
-                f"No new data"
-            )
+            logger.info(f"No new data")
     return load_status
diff --git a/tests/test_extract_lambda.py b/tests/test_extract_lambda.py
index 3405743..5a1c5b2 100644
--- a/tests/test_extract_lambda.py
+++ b/tests/test_extract_lambda.py
@@ -12,7 +12,7 @@ from src.extract_lambda import (
     DBConnectionException,
     lambda_handler,
     process_and_upload_tables,
-    retrieve_secrets
+    retrieve_secrets,
 )
 
 
@@ -25,7 +25,9 @@ def mock_config():
         "password": "password",
         "database": "db",
     }
-    with patch("src.extract_lambda.retrieve_secrets", return_value=env_vars) as mock_config:
+    with patch(
+        "src.extract_lambda.retrieve_secrets", return_value=env_vars
+    ) as mock_config:
         yield mock_config
 
 
@@ -185,31 +187,35 @@ class TestProcessAndUploadTables:
         queries = [
             "SELECT table_name FROM information_schema.tables WHERE table_schema='public' AND table_type='BASE TABLE';",
             "SELECT * FROM Fruits WHERE last_updated > :latest;",
-            "SELECT column_name FROM INFORMATION_SCHEMA.COLUMNS where table_name = 'Fruits';"
+            "SELECT column_name FROM INFORMATION_SCHEMA.COLUMNS where table_name = 'Fruits';",
         ]
         return_values = [
-            [['Fruits']],
+            [["Fruits"]],
             [],  # No new rows with a more recent last_updated timestamp
-            [['Food_type'], ['Flavour'], ['Colour'], ['last_updated']]
+            [["Food_type"], ["Flavour"], ["Colour"], ["last_updated"]],
         ]
         vals = dict(zip(queries, return_values))
 
         # Patch the database connection and set return values for queries
-        with patch('src.extract_lambda.Connection') as mock_db:
+        with patch("src.extract_lambda.Connection") as mock_db:
             mock_db().run.side_effect = return_values
-            s3_key = 'Fruits/2024/08/15/Fruits_16:46:30.csv'
+            s3_key = "Fruits/2024/08/15/Fruits_16:46:30.csv"
             existing_files = {
-                s3_key: 'Food_type,Flavour,Colour,last_updated\nVegetable,Sour,Green,2022-11-03 14:20:49.962\nBerry,Sweet,Red,2022-11-03 14:20:49.962'
+                s3_key: "Food_type,Flavour,Colour,last_updated\nVegetable,Sour,Green,2022-11-03 14:20:49.962\nBerry,Sweet,Red,2022-11-03 14:20:49.962"
             }
 
             # Simulate S3 bucket and file setup
-            s3_client.create_bucket(Bucket='test_extract_bucket', 
-                                    CreateBucketConfiguration={'LocationConstraint': 'eu-west-2'})
-            s3_client.upload_file('tests/dummy_identical.csv', 'test_extract_bucket', s3_key)
-            
+            s3_client.create_bucket(
+                Bucket="test_extract_bucket",
+                CreateBucketConfiguration={"LocationConstraint": "eu-west-2"},
+            )
+            s3_client.upload_file(
+                "tests/dummy_identical.csv", "test_extract_bucket", s3_key
+            )
+
             # Run the process_and_upload_tables function
             process_and_upload_tables(mock_db(), existing_files, client=s3_client)
             # Assert that the log contains "No new data"
-            assert 'No new data' in caplog.text
+            assert "No new data" in caplog.text
 
-    # process and upload tables needs more tests
\ No newline at end of file
+    # process and upload tables needs more tests