Rewrite the conditional processing of backup_prefix

asavchkov · asavchkov · commit 52fc53757a3f · 2024-02-04T20:02:20.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
@@ -44,12 +44,9 @@ class FSTestBackupDir(TestBackupDir):
 
     def __init__(self, *, rel_path, backup):
         backup_prefix = os.environ.get('PG_PROBACKUP_TEST_BACKUP_DIR_PREFIX')
-        if backup_prefix:
-            if not os.path.isabs(backup_prefix):
-                raise Exception(f"PG_PROBACKUP_TEST_BACKUP_DIR_PREFIX must be an absolute path, current value: {backup_prefix}")
-            self.path = os.path.join(backup_prefix, rel_path, backup)
-        else:
-            self.path = os.path.join(init_params.tmp_path, rel_path, backup)
+        if not os.path.isabs(backup_prefix):
+            raise Exception(f"PG_PROBACKUP_TEST_BACKUP_DIR_PREFIX must be an absolute path, current value: {backup_prefix}")
+        self.path = os.path.join(backup_prefix or init_params.tmp_path, rel_path, backup)
         self.pb_args = ('-B', self.path)
 
     def list_instance_backups(self, instance):
