Usage of external testgres.os_ops and testgres.common

docs/Makefile

@@ -17,4 +17,5 @@ help:
 # Catch-all target: route all unknown targets to Sphinx using the new
 # "make mode" option.  $(O) is meant as a shortcut for $(SPHINXOPTS).
 %: Makefile
+	@pip install --force-reinstall ..
 	@$(SPHINXBUILD) -M $@ "$(SOURCEDIR)" "$(BUILDDIR)" $(SPHINXOPTS) $(O)

docs/source/conf.py

@@ -14,11 +14,19 @@
 #
 import os
 import sys
-sys.path.insert(0, os.path.abspath('../..'))
+import testgres
+
+assert testgres.__path__ is not None
+assert len(testgres.__path__) == 1
+assert type(testgres.__path__[0] == str)  # noqa: E721
+p = os.path.dirname(testgres.__path__[0])
+assert type(p) == str  # noqa: E721
+sys.path.insert(0, os.path.abspath(p))
 
 # -- Project information -----------------------------------------------------
 
 project = u'testgres'
+package_name = u'testgres'
 copyright = u'2016-2023, Postgres Professional'
 author = u'Postgres Professional'
 

setup.py

@@ -11,7 +11,8 @@
     "port-for>=0.4",
     "six>=1.9.0",
     "psutil",
-    "packaging"
+    "packaging",
+    "testgres.os_ops>=0.0.2,<1.0.0"
 ]
 
 # Add compatibility enum class
@@ -29,7 +30,8 @@
 setup(
     version='1.11.1',
     name='testgres',
-    packages=['testgres', 'testgres.operations', 'testgres.impl'],
+    packages=['testgres', 'testgres.impl'],
+    package_dir={"testgres": "src"},
     description='Testing utility for PostgreSQL and its extensions',
     url='https://github.com/postgrespro/testgres',
     long_description=readme,

testgres/__init__.py → src/__init__.py

@@ -51,9 +51,9 @@
 
 from .config import testgres_config
 
-from .operations.os_ops import OsOperations, ConnectionParams
-from .operations.local_ops import LocalOperations
-from .operations.remote_ops import RemoteOperations
+from testgres.operations.os_ops import OsOperations, ConnectionParams
+from testgres.operations.local_ops import LocalOperations
+from testgres.operations.remote_ops import RemoteOperations
 
 __all__ = [
     "get_new_node",

testgres/backup.py → src/backup.py

@@ -13,7 +13,7 @@
 
 from .exceptions import BackupException
 
-from .operations.os_ops import OsOperations
+from testgres.operations.os_ops import OsOperations
 
 from .utils import \
     get_bin_path2, \

testgres/cache.py → src/cache.py

@@ -16,8 +16,8 @@
     get_bin_path2, \
     execute_utility2
 
-from .operations.local_ops import LocalOperations
-from .operations.os_ops import OsOperations
+from testgres.operations.local_ops import LocalOperations
+from testgres.operations.os_ops import OsOperations
 
 
 def cached_initdb(data_dir, logfile=None, params=None, os_ops: OsOperations = None, bin_path=None, cached=True):

testgres/config.py → src/config.py

@@ -9,8 +9,8 @@
 from contextlib import contextmanager
 
 from .consts import TMP_CACHE
-from .operations.os_ops import OsOperations
-from .operations.local_ops import LocalOperations
+from testgres.operations.os_ops import OsOperations
+from testgres.operations.local_ops import LocalOperations
 
 log_level = os.getenv('LOGGING_LEVEL', 'WARNING').upper()
 log_format = os.getenv('LOGGING_FORMAT', '%(asctime)s - %(levelname)s - %(message)s')

src/exceptions.py

@@ -0,0 +1,71 @@
+# coding: utf-8
+
+import six
+
+from testgres.operations.exceptions import TestgresException
+from testgres.operations.exceptions import ExecUtilException
+from testgres.operations.exceptions import InvalidOperationException
+
+
+class PortForException(TestgresException):
+    pass
+
+
+@six.python_2_unicode_compatible
+class QueryException(TestgresException):
+    def __init__(self, message=None, query=None):
+        super(QueryException, self).__init__(message)
+
+        self.message = message
+        self.query = query
+
+    def __str__(self):
+        msg = []
+
+        if self.message:
+            msg.append(self.message)
+
+        if self.query:
+            msg.append(u'Query: {}'.format(self.query))
+
+        return six.text_type('\n').join(msg)
+
+
+class TimeoutException(QueryException):
+    pass
+
+
+class CatchUpException(QueryException):
+    pass
+
+
+@six.python_2_unicode_compatible
+class StartNodeException(TestgresException):
+    def __init__(self, message=None, files=None):
+        super(StartNodeException, self).__init__(message)
+
+        self.message = message
+        self.files = files
+
+    def __str__(self):
+        msg = []
+
+        if self.message:
+            msg.append(self.message)
+
+        for f, lines in self.files or []:
+            msg.append(u'{}\n----\n{}\n'.format(f, lines))
+
+        return six.text_type('\n').join(msg)
+
+
+class InitNodeException(TestgresException):
+    pass
+
+
+class BackupException(TestgresException):
+    pass
+
+
+assert ExecUtilException.__name__ == "ExecUtilException"
+assert InvalidOperationException.__name__ == "InvalidOperationException"

testgres/impl/port_manager__generic.py → src/impl/port_manager__generic.py

@@ -1,4 +1,4 @@
-from ..operations.os_ops import OsOperations
+from testgres.operations.os_ops import OsOperations
 
 from ..port_manager import PortManager
 from ..exceptions import PortForException

testgres/node.py → src/node.py

@@ -104,9 +104,9 @@
 
 from .backup import NodeBackup
 
-from .operations.os_ops import ConnectionParams
-from .operations.os_ops import OsOperations
-from .operations.local_ops import LocalOperations
+from testgres.operations.os_ops import ConnectionParams
+from testgres.operations.os_ops import OsOperations
+from testgres.operations.local_ops import LocalOperations
 
 InternalError = pglib.InternalError
 ProgrammingError = pglib.ProgrammingError

testgres/utils.py → src/utils.py

@@ -15,10 +15,10 @@
 
 from .exceptions import ExecUtilException
 from .config import testgres_config as tconf
-from .operations.os_ops import OsOperations
-from .operations.remote_ops import RemoteOperations
-from .operations.local_ops import LocalOperations
-from .operations.helpers import Helpers as OsHelpers
+from testgres.operations.os_ops import OsOperations
+from testgres.operations.remote_ops import RemoteOperations
+from testgres.operations.local_ops import LocalOperations
+from testgres.operations.helpers import Helpers as OsHelpers
 
 from .impl.port_manager__generic import PortManager__Generic