Dec 10, 2024 · Dec 8, 2024 · Dec 9, 2024 · Dec 9, 2024 · Dec 9, 2024 · Dec 9, 2024
diff --git a/testgres/operations/helpers.py b/testgres/operations/helpers.py
 import locale


 class Helpers:
    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()

    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'

    def PrepareProcessInput(input, encoding):
        if not input:
            return None

        if type(input) == str:  # noqa: E721
            if encoding is None:
                return input.encode()
                return input.encode(__class__.GetDefaultEncoding())

            assert type(encoding) == str  # noqa: E721
            return input.encode(encoding)
diff --git a/testgres/operations/raise_error.py b/testgres/operations/raise_error.py
 from ..exceptions import ExecUtilException
 from .helpers import Helpers


 class RaiseError:
        assert type(data) == bytes  # noqa: E721

        try:
            return data.decode('utf-8')
            return data.decode(Helpers.GetDefaultEncoding())
        except UnicodeDecodeError:
            pass
Original file line number	Diff line number	Diff line change
		@@ -1,11 +1,48 @@
		import locale


		class Helpers:
		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()

		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'

		def PrepareProcessInput(input, encoding):
		if not input:
		return None

		if type(input) == str: # noqa: E721
		if encoding is None:
		return input.encode()
		return input.encode(__class__.GetDefaultEncoding())

		assert type(encoding) == str # noqa: E721
		return input.encode(encoding)
Expand Down
Original file line number	Diff line number	Diff line change
		@@ -1,4 +1,5 @@
		from ..exceptions import ExecUtilException
		from .helpers import Helpers


		class RaiseError:
Expand DownExpand Up		@@ -29,7 +30,7 @@ def _TranslateDataIntoString__FromBinary(data):
		assert type(data) == bytes # noqa: E721

		try:
		return data.decode('utf-8')
		return data.decode(Helpers.GetDefaultEncoding())
		except UnicodeDecodeError:
		pass

Expand Down