lint

Author: nhsdevws

e2e_batch/Makefile

@@ -1,4 +1,4 @@
 -include .env
 
 test:
-	ENVIRONMENT=$(ENVIRONMENT) poetry run python -m unittest -v -c
+    ENVIRONMENT=$(ENVIRONMENT) poetry run python -m unittest -v -c

e2e_batch/constants.py

@@ -22,6 +22,7 @@
 ack_metadata_queue_name = f"imms-{environment}-ack-metadata-queue.fifo"
 audit_table_name = f"immunisation-batch-{environment}-audit-table"
 
+
 class EventName():
     CREATE = "INSERT"
     UPDATE = "MODIFY"

e2e_batch/utils.py

@@ -21,11 +21,10 @@
     ACK_PREFIX,
     FILE_NAME_VAL_ERROR,
     HEADER_RESPONSE_CODE_COLUMN,
-    ActionFlag,
-    Operation,
     RAVS_URI,
 )
 
+
 def upload_file_to_s3(file_name, bucket, prefix):
     """Upload the given file to the specified bucket under the provided prefix.
     Returns the S3 key if successful, or raises an exception."""
@@ -273,7 +272,7 @@ def fetch_pk_and_operation_from_dynamodb(identifier_pk):
         Logs any exceptions encountered during the DynamoDB query.
     """
     try:
-        response = table.query(
+        response = events_table.query(
             IndexName="IdentifierGSI",
             KeyConditionExpression="IdentifierPK = :identifier_pk",
             ExpressionAttributeValues={":identifier_pk": identifier_pk},
@@ -391,6 +390,7 @@ def verify_final_ack_file(file_key):
         )
     return True
 
+
 def delete_filename_from_audit_table(filename) -> bool:
 
     # 1. Query the GSI to get all items with the given filename
@@ -409,6 +409,7 @@ def delete_filename_from_audit_table(filename) -> bool:
         logger.error(f"Error deleting from audit table: {e}")
         return False
 
+
 def delete_filename_from_events_table(identifier) -> bool:
 
     # 1. Query the GSI to get all items with the given filename
@@ -428,6 +429,7 @@ def delete_filename_from_events_table(identifier) -> bool:
         logger.warning(f"Error deleting from events table: {e}")
         return False
 
+
 def poll_s3_file_pattern(prefix, search_pattern):
     """Poll the ACK_BUCKET for an ack file that contains the input_file_name as a substring."""