database connection added to func

1 files changed, 32 insertions, 0 deletions

diff --git a/src/extract_lambda.py b/src/extract_lambda.py
index e69de29..7d56c66 100644
--- a/src/extract_lambda.py
+++ b/src/extract_lambda.py
@@ -0,0 +1,32 @@
+from pg8000.native import Connection, Error, DatabaseError, InterfaceError
+from dotenv import load_dotenv
+import os
+
+load_dotenv()
+
+def extract():
+
+# temporary credentials for dev- will not have access when uploaded
+
+    database = os.getenv('database')
+    user = os.getenv('user')
+    password = os.getenv('password')
+    host = os.getenv('host')
+    port = os.getenv('port')
+
+
+    try:
+        db = Connection.run(
+        database=database,
+        user=user,
+        password=password,
+        host=host,
+        port=port
+        )
+    except DatabaseError as e:
+        print(e)
+    except InterfaceError as i:
+        print(i)
+
+
+    
\ No newline at end of file