From 1ef515e265b9b749af1fdaa60901a6e8246a0e3d Mon Sep 17 00:00:00 2001
From: Dustin Thomas <stdio@cptlobster.dev>
Date: Sun, 14 Jan 2024 14:14:12 -0600
Subject: [PATCH] setup defaults for environment variables (configured for a
 locally running Postgres docker with default config)

---
 database.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/database.py b/database.py
index 4ee403f..47a351b 100644
--- a/database.py
+++ b/database.py
@@ -1,10 +1,12 @@
 import os
 import psycopg2
 
-DB_ADDRESS = os.environ['DB_ADDRESS']
-DB_USER = os.environ['DB_USER']
-DB_PASSWORD = os.environ['DB_PASSWORD']
-DB_NAME = os.environ['DB_NAME']
+
+DB_ADDRESS = os.getenv('DB_ADDRESS', 'localhost')
+DB_PORT = os.getenv('DB_PORT', 5432)
+DB_USER = os.getenv('DB_USER', 'postgres')
+DB_PASSWORD = os.getenv('DB_PASSWORD', '')
+DB_NAME = os.getenv('DB_NAME', 'postgres')
 
 
 class DBConnector:
@@ -31,7 +33,6 @@ class DBConnector:
             result = []
         return result
 
-
     def read(self, **kwargs):
         """Read rows from a database that match the specified filters.