testgres.os-ops 0.0.1__tar.gz

testgres_os_ops-0.0.1/PKG-INFO

@@ -0,0 +1,21 @@
+Metadata-Version: 2.4
+Name: testgres.os_ops
+Version: 0.0.1
+Summary: Testgres subsystem to work with OS
+Home-page: https://github.com/postgrespro/testgres.os_ops
+Author: Postgres Professional
+Author-email: testgres@postgrespro.ru
+License: PostgreSQL
+Keywords: testgres
+Description-Content-Type: text/markdown
+Requires-Dist: psutil
+Requires-Dist: six>=1.9.0
+Requires-Dist: testgres.common==0.0.1
+Dynamic: author
+Dynamic: author-email
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license
+Dynamic: requires-dist
+Dynamic: summary

testgres_os_ops-0.0.1/README.md

@@ -0,0 +1,10 @@
+[![Build Status](https://api.travis-ci.com/postgrespro/testgres.os_ops.svg?token=HgEMhyw9n7RCQkHCis4T&branch=master)](https://travis-ci.com/github/postgrespro/testgres.os_ops)
+
+
+# testgres - os_ops
+
+Subsystem of testgres to work with OS.
+
+## Authors
+
+[Postgres Professional](https://postgrespro.ru/about)

testgres_os_ops-0.0.1/setup.cfg

@@ -0,0 +1,11 @@
+[metadata]
+description-file = README.md
+
+[flake8]
+ignore = E501
+exclude = .git,__pycache__,env,venv
+
+[egg_info]
+tag_build = 
+tag_date = 0
+

testgres_os_ops-0.0.1/setup.py

@@ -0,0 +1,28 @@
+try:
+    from setuptools import setup
+except ImportError:
+    from distutils.core import setup
+
+# Basic dependencies
+install_requires = [
+    "psutil",
+    "six>=1.9.0",
+    "testgres.common==0.0.1",
+]
+
+setup(
+    version="0.0.1",
+    name="testgres.os_ops",
+    packages=[
+        "testgres.operations",
+    ],
+    package_dir={"testgres.operations": "src"},
+    description='Testgres subsystem to work with OS',
+    url='https://github.com/postgrespro/testgres.os_ops',
+    long_description_content_type='text/markdown',
+    license='PostgreSQL',
+    author='Postgres Professional',
+    author_email='testgres@postgrespro.ru',
+    keywords=['testgres'],
+    install_requires=install_requires,
+)

testgres_os_ops-0.0.1/src/exceptions.py

@@ -0,0 +1,53 @@
+# coding: utf-8
+
+from testgres.common.exceptions import TestgresException
+from testgres.common.exceptions import InvalidOperationException
+import six
+
+
+class ExecUtilException(TestgresException):
+    def __init__(self, message=None, command=None, exit_code=0, out=None, error=None):
+        super(ExecUtilException, self).__init__(message)
+
+        self.message = message
+        self.command = command
+        self.exit_code = exit_code
+        self.out = out
+        self.error = error
+
+    def __str__(self):
+        msg = []
+
+        if self.message:
+            msg.append(self.message)
+
+        if self.command:
+            command_s = ' '.join(self.command) if isinstance(self.command, list) else self.command
+            msg.append(u'Command: {}'.format(command_s))
+
+        if self.exit_code:
+            msg.append(u'Exit code: {}'.format(self.exit_code))
+
+        if self.error:
+            msg.append(u'---- Error:\n{}'.format(self.error))
+
+        if self.out:
+            msg.append(u'---- Out:\n{}'.format(self.out))
+
+        return self.convert_and_join(msg)
+
+    @staticmethod
+    def convert_and_join(msg_list):
+        # Convert each byte element in the list to str
+        str_list = [six.text_type(item, 'utf-8') if isinstance(item, bytes) else six.text_type(item) for item in
+                    msg_list]
+
+        # Join the list into a single string with the specified delimiter
+        return six.text_type('\n').join(str_list)
+
+
+__all__ = [
+    type(TestgresException).__name__,
+    type(InvalidOperationException).__name__,
+    type(ExecUtilException).__name__,
+]

testgres_os_ops-0.0.1/src/helpers.py

@@ -0,0 +1,55 @@
+import locale
+
+
+class Helpers:
+    @staticmethod
+    def _make_get_default_encoding_func():
+        # locale.getencoding is added in Python 3.11
+        if hasattr(locale, 'getencoding'):
+            return locale.getencoding
+
+        # It must exist
+        return locale.getpreferredencoding
+
+    # Prepared pointer on function to get a name of system codepage
+    _get_default_encoding_func = _make_get_default_encoding_func.__func__()
+
+    @staticmethod
+    def GetDefaultEncoding():
+        #
+        #   Original idea/source was:
+        #
+        #   def os_ops.get_default_encoding():
+        #       if not hasattr(locale, 'getencoding'):
+        #       locale.getencoding = locale.getpreferredencoding
+        #       return locale.getencoding() or 'UTF-8'
+        #
+
+        assert __class__._get_default_encoding_func is not None
+
+        r = __class__._get_default_encoding_func()
+
+        if r:
+            assert r is not None
+            assert type(r) == str  # noqa: E721
+            assert r != ""
+            return r
+
+        # Is it an unexpected situation?
+        return 'UTF-8'
+
+    @staticmethod
+    def PrepareProcessInput(input, encoding):
+        if not input:
+            return None
+
+        if type(input) == str:  # noqa: E721
+            if encoding is None:
+                return input.encode(__class__.GetDefaultEncoding())
+
+            assert type(encoding) == str  # noqa: E721
+            return input.encode(encoding)
+
+        # It is expected!
+        assert type(input) == bytes  # noqa: E721
+        return input