diff --git a/postgresql_info.py b/postgresql_info.py index 5da5122..f416921 100644 --- a/postgresql_info.py +++ b/postgresql_info.py @@ -5,11 +5,16 @@ load_dotenv() pg_name = os.getenv("DB_NAME") or "TJWater" pg_host = os.getenv("DB_HOST") or "127.0.0.1" +pg_port = os.getenv("DB_PORT") or "5432" pg_user = os.getenv("DB_USER") or "tjwater" pg_password = os.getenv("DB_PASSWORD") or "Tjwater@123456" def get_pgconn_string( - db_name=pg_name, db_host=pg_host, db_user=pg_user, db_password=pg_password + db_name=pg_name, + db_host=pg_host, + db_port=pg_port, + db_user=pg_user, + db_password=pg_password, ): - return f"dbname={db_name} host={db_host} user={db_user} password={db_password}" + return f"dbname={db_name} host={db_host} port={db_port} user={db_user} password={db_password}" \ No newline at end of file