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.