🦺 Event validation complete

inews/infra/io.py

@@ -13,8 +13,6 @@
 load_dotenv()
 
 session_token = os.environ["AWS_SESSION_TOKEN"] if "AWS_SESSION_TOKEN" in os.environ else None
-DATA_PATH = Path("/tmp") if "AWS_LAMBDA_FUNCTION_NAME" in os.environ else Path("data")
-
 s3 = boto3.resource(
     "s3",
     region_name=os.environ["AWS_REGION"],
@@ -23,6 +21,7 @@
     aws_session_token=session_token,
 )
 
+
 # S3
 BUCKET = "intergalactic-news"
 CHANNELS_S3_FILE = "channels_state.json"
@@ -37,6 +36,7 @@
 MAILING_CONFIG_FILE = Path("config/mailing.yaml")
 
 # Data
+DATA_PATH = Path("/tmp") if "AWS_LAMBDA_FUNCTION_NAME" in os.environ else Path("data")
 CHANNELS_LOCAL_FILE = DATA_PATH / Path("channels_state.json")
 TRANSCRIPTS_LOCAL_PATH = DATA_PATH / Path("transcripts/")
 SUMMARIES_LOCAL_PATH = DATA_PATH / Path("summaries/")

inews/infra/utils.py

@@ -2,4 +2,10 @@
 
 
 def validate_event(event: dict) -> RunStatus:
-    return RunStatus.TEST if event["Status"] == "test" else RunStatus.PROD
+    print(event)
+    if event["Status"] == "test":
+        return RunStatus.TEST
+    elif event["Status"] == "prod":
+        return RunStatus.PROD
+    else:
+        raise ValueError

lambda_function.py

@@ -4,5 +4,6 @@
 
 def handler(event, context):
     status = utils.validate_event(event)
+    print(status)
     pipeline.run_data(status)
     pipeline.run_mailing(status)