include pg_extras.conf in config

miketonks · Sep 19, 2019 · 909f2c9 · 909f2c9
1 parent cfc48cb
commit 909f2c9
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/helpers/postgresql.py b/helpers/postgresql.py
@@ -204,6 +204,24 @@ def copy_pg_hba(self):
         else:
             logger.info("No pg_hba.conf file found - skipping")
 
+        if os.path.exists('pg_extras.conf'):
+            logger.info("Copying pg_extras.conf file")
+            shutil.copy2('pg_extras.conf', self.data_dir)
+        else:
+            logger.info("No pg_extras.conf file found - writing empty file")
+            with open("%s/pg_extras.conf" % self.data_dir, "w"):
+                pass
+
+        pgconf = "%s/postgresql.conf" % self.data_dir
+        line = "include = 'pg_extras.conf'"
+        with open(pgconf, "r") as f:
+            if any(line == l.rstrip("\r\n") for l in f.readlines()):
+                logger.info("pg_extras.conf include found")
+                return
+
+        logger.info("Adding pg_extras.conf include")
+        with open(pgconf, "a") as f:
+            f.write(line + "\n")
 
     def write_recovery_conf(self, leader_hash):
         f = open("%s/recovery.conf" % self.data_dir, "w")