From 09163ac4548008e02dcdcbb04365b0f9dace467e Mon Sep 17 00:00:00 2001
From: Alexander Kukushkin <alexander.kukushkin@zalando.de>
Date: Fri, 22 May 2015 12:03:43 +0200
Subject: [PATCH] If there is no connect_address defined in yml file use
 address from listen as a fallback

---
 helpers/postgresql.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/helpers/postgresql.py b/helpers/postgresql.py
index f8d91fd6..8d32d65d 100644
--- a/helpers/postgresql.py
+++ b/helpers/postgresql.py
@@ -37,8 +37,9 @@ class Postgresql:
 
         self.config = config
 
+        connect_address = config.get('connect_address', config['listen']) or config['listen']
         self.connection_string = 'postgres://{username}:{password}@{connect_address}/postgres'.format(
-            connect_address=self.config['connect_address'], **self.replication)
+            connect_address=connect_address, **self.replication)
 
         self.conn = None
         self.cursor_holder = None