Add a backup directory prefix variable

postgres-dev · postgres-dev · commit 48195d1393ca · 2024-01-29T20:35:57.000+07:00
diff --git a/testgres/plugins/pg_probackup2/pg_probackup2/storage/fs_backup.py b/testgres/plugins/pg_probackup2/pg_probackup2/storage/fs_backup.py
@@ -42,11 +42,12 @@ class FSTestBackupDir(TestBackupDir):
 
     """ Backup directory. Usually created by running pg_probackup init -B <path>"""
 
-    def __init__(self, *, path, backup):
-        if os.path.isabs(path):
-            self.path = os.path.join(path, backup)
+    def __init__(self, *, rel_path, backup):
+        backup_prefix = os.environ.get('PG_PROBACKUP_TEST_BACKUP_DIR_PREFIX', '')
+        if os.path.isabs(backup_prefix):
+            self.path = os.path.join(backup_prefix, rel_path, backup)
         else:
-            self.path = os.path.join(init_params.tmp_path, path, backup)
+            self.path = os.path.join(init_params.tmp_path, backup_prefix, rel_path, backup)
         self.pb_args = ('-B', self.path)
 
     def list_instance_backups(self, instance):
