small changes to email_service

email_service/email_lambda.py

@@ -11,7 +11,7 @@
 import pandas as pd
 from psycopg2.extensions import connection
 from psycopg2 import connect
-from psycopg2.extras import RealDictCursor, execute
+from psycopg2.extras import RealDictCursor
 import datetime
 
 from pipeline.common import DataProcessor
@@ -36,7 +36,7 @@ def __init__(self) -> None:
         """
         self.logger = logger
 
-    def get_user_data(self, conn, data):
+    def get_user_data(self, conn):
         """
         Loads the data into an RDS database
         """
@@ -68,3 +68,10 @@ def get_connection(self) -> connection:
         )
         connecting.autocommit = True
         return connecting
+
+
+if __name__ == "__main__":
+    db = DatabaseConnection()
+    conn = db.get_connection()
+    load = Load()
+    print(load.get_user_data(conn))