testgres-pg-probackup2 0.0.5__tar.gz → 0.1.0__tar.gz

{testgres_pg_probackup2-0.0.5 → testgres_pg_probackup2-0.1.0}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: testgres_pg_probackup2
-Version: 0.0.5
+Version: 0.1.0
 Summary: Plugin for testgres that manages pg_probackup2
 Home-page: https://github.com/postgrespro/testgres
 Author: Postgres Professional

{testgres_pg_probackup2-0.0.5 → testgres_pg_probackup2-0.1.0}/pg_probackup2/app.py

@@ -45,6 +45,7 @@ class ProbackupApp:
 
     def __init__(self, test_class: unittest.TestCase,
                  pg_node, pb_log_path, test_env, auto_compress_alg, backup_dir, probackup_path=None):
+        self.process = None
         self.test_class = test_class
         self.pg_node = pg_node
         self.pb_log_path = pb_log_path
@@ -60,8 +61,35 @@ class ProbackupApp:
         self.test_class.output = None
         self.execution_time = None
 
+    def form_daemon_process(self, cmdline, env):
+        def stream_output(stream: subprocess.PIPE) -> None:
+            try:
+                for line in iter(stream.readline, ''):
+                    print(line)
+                    self.test_class.output += line
+            finally:
+                stream.close()
+
+        self.process = subprocess.Popen(
+            cmdline,
+            stdout=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            text=True,
+            env=env
+        )
+        logging.info(f"Process started in background with PID: {self.process.pid}")
+
+        if self.process.stdout and self.process.stderr:
+            stdout_thread = threading.Thread(target=stream_output, args=(self.process.stdout,), daemon=True)
+            stderr_thread = threading.Thread(target=stream_output, args=(self.process.stderr,), daemon=True)
+
+            stdout_thread.start()
+            stderr_thread.start()
+
+        return self.process.pid
+
     def run(self, command, gdb=False, old_binary=False, return_id=True, env=None,
-            skip_log_directory=False, expect_error=False, use_backup_dir=True):
+            skip_log_directory=False, expect_error=False, use_backup_dir=True, daemonize=False):
         """
         Run pg_probackup
         backup_dir: target directory for making backup
@@ -118,11 +146,14 @@ class ProbackupApp:
                 logging.warning("pg_probackup gdb suspended, waiting gdb connection on localhost:{0}".format(gdb_port))
 
             start_time = time.time()
-            self.test_class.output = subprocess.check_output(
-                cmdline,
-                stderr=subprocess.STDOUT,
-                env=env
-            ).decode('utf-8', errors='replace')
+            if daemonize:
+                return self.form_daemon_process(cmdline, env)
+            else:
+                self.test_class.output = subprocess.check_output(
+                    cmdline,
+                    stderr=subprocess.STDOUT,
+                    env=env
+                ).decode('utf-8', errors='replace')
             end_time = time.time()
             self.execution_time = end_time - start_time
 
@@ -811,5 +842,22 @@ class ProbackupApp:
         ]
         return self.run(cmd + options, expect_error=expect_error)
 
+    def maintain(
+            self, instance=None, backup_id=None,
+            options=None, old_binary=False, gdb=False, expect_error=False
+    ):
+        if options is None:
+            options = []
+        cmd_list = [
+            'maintain',
+        ]
+        if instance:
+            cmd_list += ['--instance={0}'.format(instance)]
+        if backup_id:
+            cmd_list += ['-i', backup_id]
+
+        return self.run(cmd_list + options, old_binary=old_binary, gdb=gdb,
+                        expect_error=expect_error)
+
     def build_backup_dir(self, backup='backup'):
         return fs_backup_class(rel_path=self.rel_path, backup=backup)

{testgres_pg_probackup2-0.0.5 → testgres_pg_probackup2-0.1.0}/pg_probackup2/init_helpers.py

@@ -121,8 +121,7 @@ class Init(object):
                     self.probackup_path = probackup_path_tmp
 
         if not self.probackup_path:
-            logging.error('pg_probackup binary is not found')
-            exit(1)
+            raise Exception('pg_probackup binary is not found')
 
         if os.name == 'posix':
             self.EXTERNAL_DIRECTORY_DELIMITER = ':'
@@ -213,11 +212,15 @@ class Init(object):
         if self.probackup_version.split('.')[0].isdigit():
             self.major_version = int(self.probackup_version.split('.')[0])
         else:
-            logging.error('Can\'t process pg_probackup version \"{}\": the major version is expected to be a number'.format(self.probackup_version))
-            sys.exit(1)
+            raise Exception('Can\'t process pg_probackup version \"{}\": the major version is expected to be a number'.format(self.probackup_version))
 
     def test_env(self):
         return self._test_env.copy()
 
 
-init_params = Init()
+try:
+    init_params = Init()
+except Exception as e:
+    logging.error(str(e))
+    logging.warning("testgres.plugins.probackup2.init_params is set to None.")
+    init_params = None

{testgres_pg_probackup2-0.0.5 → testgres_pg_probackup2-0.1.0}/setup.py

@@ -4,7 +4,7 @@ except ImportError:
     from distutils.core import setup
 
 setup(
-    version='0.0.5',
+    version='0.1.0',
     name='testgres_pg_probackup2',
     packages=['pg_probackup2', 'pg_probackup2.storage'],
     description='Plugin for testgres that manages pg_probackup2',

{testgres_pg_probackup2-0.0.5 → testgres_pg_probackup2-0.1.0}/testgres_pg_probackup2.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: testgres_pg_probackup2
-Version: 0.0.5
+Version: 0.1.0
 Summary: Plugin for testgres that manages pg_probackup2
 Home-page: https://github.com/postgrespro/testgres
 Author: Postgres Professional