everysk-lib 1.9.7__tar.gz → 1.9.9__tar.gz

{everysk_lib-1.9.7/src/everysk_lib.egg-info → everysk_lib-1.9.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: everysk-lib
-Version: 1.9.7
+Version: 1.9.9
 Summary: Generic lib to share python code on Everysk.
 License-Expression: LicenseRef-Proprietary
 Project-URL: Homepage, https://everysk.com/
@@ -12,7 +12,7 @@ Classifier: Programming Language :: Python :: 3.11
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE.txt
-Requires-Dist: python-dateutil<=2.9.0,>=2.8.2
+Requires-Dist: python-dateutil<=3.0.0,>=2.8.2
 Requires-Dist: holidays==0.51
 Requires-Dist: tzdata<=2025.2,>=2024.1
 Provides-Extra: requests
@@ -115,7 +115,9 @@ Requires-Dist: lark==1.2.2; extra == "expression"
 Requires-Dist: numpy==1.26.4; extra == "expression"
 Requires-Dist: pandas==2.1.4; extra == "expression"
 Provides-Extra: postgresql
-Requires-Dist: psqlpy==0.11.9; extra == "postgresql"
+Requires-Dist: psycopg-binary==3.3.0; extra == "postgresql"
+Requires-Dist: psycopg-pool==3.3.0; extra == "postgresql"
+Requires-Dist: psycopg==3.3.0; extra == "postgresql"
 Dynamic: license-file
 
 

{everysk_lib-1.9.7 → everysk_lib-1.9.9}/pyproject.toml

@@ -38,7 +38,7 @@ classifiers = [
 
 # These are the assumed default dependencies from pip
 dependencies = [
-    "python-dateutil>=2.8.2,<=2.9.0",
+    "python-dateutil>=2.8.2,<=3.0.0",
     "holidays==0.51",
     "tzdata>=2024.1,<=2025.2",
 ]
@@ -178,7 +178,9 @@ expression = [
 ]
 
 postgresql = [
-    "psqlpy==0.11.9"
+    "psycopg-binary==3.3.0",
+    "psycopg-pool==3.3.0",
+    "psycopg==3.3.0"
 ]
 
 [build-system]

everysk_lib-1.9.9/src/everysk/core/retry.py

@@ -0,0 +1,51 @@
+###############################################################################
+#
+# (C) Copyright 2025 EVERYSK TECHNOLOGIES
+#
+# This is an unpublished work containing confidential and proprietary
+# information of EVERYSK TECHNOLOGIES. Disclosure, use, or reproduction
+# without authorization of EVERYSK TECHNOLOGIES is prohibited.
+#
+###############################################################################
+from typing import Any
+
+from everysk.config import settings
+from everysk.core.log import Logger
+
+log = Logger('everysk-lib-core-retry')
+
+
+def retry(
+    func: callable, params: dict, retry_count: int = 0, retries: int = 5, exceptions: tuple | Exception = Exception
+) -> Any:
+    """
+    Retries a function call a number of times if it raises an exception.
+
+    Args:
+        func (callable): The function to be called.
+        params (dict): The parameters to be passed to the function.
+        retry_count (int, optional): The current retry count. Defaults to 0.
+        retries (int, optional): The maximum number of retries. Defaults to 5.
+        exceptions (tuple | Exception, optional): The exceptions to catch. Defaults to Exception.
+
+    Raises:
+        Exception: If the maximum number of retries is reached.
+    """
+    try:
+        return func(**params)
+    except exceptions:
+        if retry_count < retries:
+            if settings.RETRY_SHOW_LOGS:
+                msg = f'Retry {retry_count + 1} of {retries} for function {func.__name__} due to exception.'
+                log.warning(
+                    msg,
+                    extra={
+                        'function': func.__name__,
+                        'params': params,
+                        'retry_count': retry_count + 1,
+                        'max_retries': retries,
+                    },
+                )
+            return retry(func, params, retry_count + 1, retries, exceptions)
+
+        raise

everysk_lib-1.9.9/src/everysk/core/tests.py

@@ -0,0 +1,240 @@
+###############################################################################
+#
+# (C) Copyright 2023 EVERYSK TECHNOLOGIES
+#
+# This is an unpublished work containing confidential and proprietary
+# information of EVERYSK TECHNOLOGIES. Disclosure, use, or reproduction
+# without authorization of EVERYSK TECHNOLOGIES is prohibited.
+#
+###############################################################################
+# ruff: noqa: F401
+
+## Remember to prefix all import with EveryskLib to avoid clash with other tests
+
+## Cloud function Test Cases
+try:
+    from everysk.core.cloud_function.tests import CloudFunctionTestCase as EveryskLibCloudFunctionTestCase
+except ModuleNotFoundError as error:
+    # This will prevent running these tests if redis is not installed
+    if not error.args[0].startswith("No module named 'redis'"):
+        raise
+
+## Compress Test Cases
+from everysk.core._tests.compress import CompressGzipJsonTestCase as EveryskLibCompressGzipJsonTestCase
+from everysk.core._tests.compress import CompressGzipPickleTestCase as EveryskLibCompressGzipPickleTestCase
+from everysk.core._tests.compress import CompressTestCase as EveryskLibCompressTestCase
+from everysk.core._tests.compress import CompressZlibJsonTestCase as EveryskLibCompressZlibJsonTestCase
+from everysk.core._tests.compress import CompressZlibPickleTestCase as EveryskLibCompressZlibPickleTestCase
+from everysk.core._tests.compress import FileHandlingTestCase as EveryskLibFileHandlingTestCase
+
+## Config Test Cases
+from everysk.core._tests.config import SettingsManagerTestCase as EveryskLibSettingsManagerTestCase
+from everysk.core._tests.config import SettingsModulesTestCase as EveryskLibSettingsModulesTestCase
+from everysk.core._tests.config import SettingsTestCase as EveryskLibSettingsTestCase
+
+## Exceptions Test Cases
+from everysk.core._tests.exceptions import BaseExceptionTestCase as EveryskLibBaseExceptionTestCase
+from everysk.core._tests.exceptions import DefaultErrorTestCase as EveryskLibDefaultErrorTestCase
+from everysk.core._tests.exceptions import FieldValueErrorTestCase as EveryskLibFieldValueErrorTestCase
+from everysk.core._tests.exceptions import HandledExceptionTestCase as EveryskLibHandledExceptionTestCase
+from everysk.core._tests.exceptions import HttpErrorTestCase as EveryskLibHttpErrorTestCase
+from everysk.core._tests.exceptions import ReadonlyErrorTestCase as EveryskLibReadonlyErrorTestCase
+from everysk.core._tests.exceptions import RequiredErrorTestCase as EveryskLibRequiredErrorTestCase
+from everysk.core._tests.exceptions import SDKExceptionsTestCase as EveryskLibSDKExceptionsTestCase
+from everysk.core._tests.exceptions import TestAPIError as EveryskLibTestAPIError
+
+## Fields Test Cases
+from everysk.core._tests.fields import BoolFieldTestCase as EveryskLibBoolFieldTestCase
+from everysk.core._tests.fields import ChoiceFieldTestCase as EveryskLibChoiceFieldTestCase
+from everysk.core._tests.fields import COD3770TestCase as EveryskLibCOD3770TestCase
+from everysk.core._tests.fields import DateFieldTestCase as EveryskLibDateFieldTestCase
+from everysk.core._tests.fields import DateTimeFieldTestCase as EveryskLibDateTimeFieldTestCase
+from everysk.core._tests.fields import DictFieldTestCase as EveryskLibDictFieldTestCase
+from everysk.core._tests.fields import EmailFieldTestCase as EveryskLibEmailFieldTestCase
+from everysk.core._tests.fields import FieldTestCase as EveryskLibFieldTestCase
+from everysk.core._tests.fields import FieldUndefinedTestCase as EveryskLibFieldUndefinedTestCase
+from everysk.core._tests.fields import FloatFieldTestCase as EveryskLibFloatFieldTestCase
+from everysk.core._tests.fields import IntFieldTestCase as EveryskLibIntFieldTestCase
+from everysk.core._tests.fields import IteratorFieldTestCase as EveryskLibIteratorFieldTestCase
+from everysk.core._tests.fields import ListFieldTestCase as EveryskLibListFieldTestCase
+from everysk.core._tests.fields import ObjectInitPropertyTestCase as EveryskLibObjectInitPropertyTestCase
+from everysk.core._tests.fields import SetFieldTestCase as EveryskLibSetFieldTestCase
+from everysk.core._tests.fields import StrFieldTestCase as EveryskLibStrFieldTestCase
+from everysk.core._tests.fields import TupleFieldTestCase as EveryskLibTupleFieldTestCase
+from everysk.core._tests.fields import URLFieldTestCase as EveryskLibURLFieldTestCase
+
+## Date, DateTime Test Cases
+from everysk.core.datetime.tests.calendar import CalendarTestCase as EveryskLibCalendarTestCase
+from everysk.core.datetime.tests.date import DateTestCase as EveryskLibDateTestCase
+from everysk.core.datetime.tests.date_mixin import GetHolidaysTestCase as EveryskLibDateMixinGetHolidaysTestCase
+from everysk.core.datetime.tests.datetime import DateTimeTestCase as EveryskLibDateTimeTestCase
+
+## Firestore Test Cases
+try:
+    from everysk.core._tests.firestore import (
+        BaseDocumentCachedConfigTestCase as EveryskLibBaseDocumentCachedConfigTestCase,
+    )
+    from everysk.core._tests.firestore import BaseDocumentConfigTestCase as EveryskLibBaseDocumentConfigTestCase
+    from everysk.core._tests.firestore import DocumentCachedTestCase as EveryskLibDocumentCachedTestCase
+    from everysk.core._tests.firestore import DocumentTestCase as EveryskLibDocumentTestCase
+    from everysk.core._tests.firestore import FirestoreClientTestCase as EveryskLibFirestoreClientTestCase
+    from everysk.core._tests.firestore import LoadsPaginatedTestCase as EveryskLibLoadsPaginatedTestCase
+except ModuleNotFoundError as error:
+    # This will prevent running these tests if google-cloud-firestore is not installed
+    if not error.args[0].startswith("No module named 'google"):
+        raise
+
+## Http Test Cases
+try:
+    from everysk.core._tests.http import HttpConnectionConfigTestCase as EveryskLibHttpConnectionConfigTestCase
+    from everysk.core._tests.http import HttpConnectionTestCase as EveryskLibHttpConnectionTestCase
+    from everysk.core._tests.http import HttpDELETEConnectionTestCase as EveryskLibHttpDELETEConnectioNTestCase
+    from everysk.core._tests.http import HttpGETConnectionTestCase as EveryskLibHttpGETConnectionTestCase
+    from everysk.core._tests.http import HttpHEADConnectionTestCase as EveryskLibHttpHEADConnectionTestCase
+    from everysk.core._tests.http import HttpOPTIONSConnectionTestCase as EveryskLibHttpOPTIONSConnectionTestCase
+    from everysk.core._tests.http import HttpPATCHConnectionTestCase as EveryskLibHttpPATCHConnectionTestCase
+    from everysk.core._tests.http import (
+        HttpPOSTCompressedConnectionTestCase as EveryskLibHttpPOSTCompressedConnectionTestCase,
+    )
+    from everysk.core._tests.http import HttpPOSTConnectionTestCase as EveryskLibHttpPOSTConnectionTestCase
+    from everysk.core._tests.http import HttpPUTConnectionTestCase as EveryskLibHttpPUTCompressedConnectionTestCase
+    from everysk.core._tests.http import HttpSDKPOSTConnectionTestCase as EveryskLibHttpSDKPOSTConnectionTestCase
+except ModuleNotFoundError as error:
+    # This will prevent running these tests if requests is not installed
+    if not error.args[0].startswith("No module named 'requests'"):
+        raise
+
+## Lists Test Cases
+from everysk.core._tests.lists import SlicesTestCase as EveryskLibSlicesTestCase
+from everysk.core._tests.lists import SortListDictTestCase as EveryskLibSortListDictTestCase
+from everysk.core._tests.lists import SplitInSlicesTestCase as EveryskLibSplitInSlicesTestCase
+
+## Log Test Cases
+from everysk.core._tests.log import LoggerExtraDataTestCase as EveryskLibLoggerExtraDataTestCase
+from everysk.core._tests.log import LoggerFormatterTestCase as EveryskLibLoggerFormatterTestCase
+from everysk.core._tests.log import LoggerJsonTestCase as EveryskLibLoggerJsonTestCase
+from everysk.core._tests.log import LoggerManagerTestCase as EveryskLibLoggerManagerTestCase
+from everysk.core._tests.log import LoggerMethodsTestCase as EveryskLibLoggerMethodsTestCase
+from everysk.core._tests.log import LoggerStackLevelTestCase as EveryskLibLoggerStackLevelTestCase
+from everysk.core._tests.log import LoggerStdoutTestCase as EveryskLibLoggerStdoutTestCase
+from everysk.core._tests.log import LoggerTestCase as EveryskLibLoggerTestCase
+from everysk.core._tests.log import LoggerTraceTestCase as EveryskLibLogTraceTestCase
+
+try:
+    # We need requests to run this test
+    from everysk.core._tests.log import LoggerSlackTestCase as EveryskLibLoggerSlackTestCase
+except ModuleNotFoundError as error:
+    # This will prevent running these tests if requests is not installed
+    if not error.args[0].startswith("No module named 'requests'"):
+        raise
+
+## Number Test Cases
+from everysk.core._tests.number import NumberTestCase as EveryskLibNumberTestCase
+
+## Object Test Cases
+from everysk.core._tests.object import AfterInitTestCase as EveryskLibAfterInitTestCase
+from everysk.core._tests.object import BaseDictPropertyTestCase as EveryskLibBaseDictPropertyTestCase
+from everysk.core._tests.object import BaseDictSuperTestCase as EveryskLibBaseDictSuperTestCase
+from everysk.core._tests.object import BaseDictTestCase as EveryskLibBaseDictTestCase
+from everysk.core._tests.object import BaseFieldTestCase as EveryskLibBaseFieldTestCase
+from everysk.core._tests.object import BaseObjectTestCase as EveryskLibBaseObjectTestCase
+from everysk.core._tests.object import BeforeInitTestCase as EveryskLibBeforeInitTestCase
+from everysk.core._tests.object import ConfigHashTestCase as EveryskLibConfigHashTestCase
+from everysk.core._tests.object import FrozenDictTestCase as EveryskLibFrozenDictTestCase
+from everysk.core._tests.object import FrozenObjectTestCase as EveryskLibFrozenObjectTestCase
+from everysk.core._tests.object import MetaClassAttributesTestCase as EveryskLibMetaClassAttributesTestCase
+from everysk.core._tests.object import MetaClassConfigTestCase as EveryskLibMetaClassConfigTestCase
+from everysk.core._tests.object import NpArrayTestCase as EveryskLibNpArrayTestCase
+from everysk.core._tests.object import RequiredTestCase as EveryskLibRequiredTestCase
+from everysk.core._tests.object import SilentTestCase as EveryskLibSilentTestCase
+from everysk.core._tests.object import TypingCheckingTestCase as EveryskLibTypingCheckingTestCase
+from everysk.core._tests.object import ValidateTestCase as EveryskLibValidateTestCase
+
+## Redis Test Cases
+try:
+    from everysk.core._tests.redis import CacheDecoratorTestCase as EveryskLibCacheDecoratorTestCase
+    from everysk.core._tests.redis import RedisCacheCompressedTestCase as EveryskLibRedisCacheCompressedTestCase
+    from everysk.core._tests.redis import RedisCacheGetSetTestCase as EveryskLibRedisCacheGetSetTestCase
+    from everysk.core._tests.redis import RedisCacheTestCase as EveryskLibRedisCacheTestCase
+    from everysk.core._tests.redis import RedisChannelTestCase as EveryskLibRedisChannelTestCase
+    from everysk.core._tests.redis import RedisClientTestCase as EveryskLibRedisClientTestCase
+    from everysk.core._tests.redis import RedisListTestCase as EveryskLibRedisListTestCase
+    from everysk.core._tests.redis import RedisLockTestCase as EveryskLibRedisLockTestCase
+except ModuleNotFoundError as error:
+    # This will prevent running these tests if redis is not installed
+    if not error.args[0].startswith("No module named 'redis'"):
+        raise
+
+## Retry Test Cases
+from everysk.core._tests.retry import RetryTestCase as EveryskLibRetryTestCase
+
+## Serialize Test Cases
+from everysk.core._tests.serialize.test_json import SerializeJsonDumpsTestCase as EveryskLibSerializeJsonDumpsTestCase
+from everysk.core._tests.serialize.test_json import SerializeJsonLoadsTestCase as EveryskLibSerializeJsonLoadsTestCase
+from everysk.core._tests.serialize.test_pickle import (
+    SerializePickleDumpsTestCase as EveryskLibSerializePickleDumpsTestCase,
+)
+from everysk.core._tests.serialize.test_pickle import (
+    SerializePickleLoadsTestCase as EveryskLibSerializePickleLoadsTestCase,
+)
+
+try:
+    from everysk.core._tests.serialize.test_orjson import (
+        SerializeOrjsonDumpsTestCase as EveryskLibSerializeOrjsonDumpsTestCase,
+    )
+    from everysk.core._tests.serialize.test_orjson import (
+        SerializeOrjsonLoadsTestCase as EveryskLibSerializeOrjsonLoadsTestCase,
+    )
+except ModuleNotFoundError as error:
+    # This will prevent running these tests if orjson is not installed
+    if not error.args[0].startswith("No module named 'orjson'"):
+        raise
+
+## SFTP Test Cases
+try:
+    from everysk.core._tests.sftp import KnownHostsTestCase as EveryskLibKnownHostsTestCase
+    from everysk.core._tests.sftp import SFTPTestCase as EveryskLibSFTPTestCase
+except ModuleNotFoundError as error:
+    # This will prevent running these tests if Paramiko is not installed
+    if not error.args[0].startswith("No module named 'paramiko'"):
+        raise
+
+## Signing Test Cases
+from everysk.core._tests.signing import SignTestCase as EveryskLibSignTestCase
+from everysk.core._tests.signing import UnsignTestCase as EveryskLibUnsignTestCase
+
+## String Test Cases
+from everysk.core._tests.string import StringTestCase as EveryskLibStringTestCase
+
+## Slack Test Cases
+try:
+    from everysk.core._tests.slack import SlackTestCase as EveryskLibSlackTestCase
+except ModuleNotFoundError as error:
+    # This will prevent running these tests if requests is not installed
+    if not error.args[0].startswith("No module named 'requests'"):
+        raise
+
+
+## Thread Test Cases
+from everysk.core._tests.threads import ThreadPoolTestCase as EveryskLibThreadPoolTestCase
+from everysk.core._tests.threads import ThreadTestCase as EveryskLibThreadTestCase
+
+## Undefined Test Cases
+from everysk.core._tests.undefined import UndefinedTestCase as EveryskLibUndefinedTestCase
+
+## Unittest Test Cases
+from everysk.core._tests.unittests import SDKUnittestTestCase as EveryskLibSDKUnittestTestCase
+
+## Utils Test Cases
+from everysk.core._tests.utils import BoolConverterTestCase as EveryskLibBoolConverterTestCase
+from everysk.core._tests.utils import SearchKeyTestCase as EveryskLibSearchKeyTestCase
+
+## Workers Test Cases
+try:
+    from everysk.core._tests.workers import BaseGoogleTestCase as EveryskLibBaseGoogleTestCase
+    from everysk.core._tests.workers import TaskGoogleTestCase as EveryskLibTaskGoogleTestCase
+    from everysk.core._tests.workers import WorkerGoogleTestCase as EveryskLibWorkerGoogleTestCase
+except ModuleNotFoundError as error:
+    # This will prevent running these tests if google-cloud-tasks is not installed
+    if not error.args[0].startswith("No module named 'google"):
+        raise

{everysk_lib-1.9.7 → everysk_lib-1.9.9}/src/everysk/sdk/entities/secrets/base.py

@@ -119,7 +119,10 @@ class Secrets(BaseEntity, metaclass=SecretsScriptMetaClass):
             path (str): The path to retrieve the value from.
 
         Returns:
-            Any: The value retrieved from the path.
+            Any | None: The value retrieved from the path or None if the path is empty.
         """
+        if not path:
+            return None
+
         _, _ = cls._split_path(path)
         return cls.get_response(params={'path': path})

{everysk_lib-1.9.7 → everysk_lib-1.9.9}/src/everysk/settings.py

@@ -93,3 +93,6 @@ HTTP_USE_RANDOM_USER_AGENT = BoolField(default=False)
 
 # Default directory to use the known_hosts file and other SFTP configurations
 EVERYSK_SFTP_DIR = StrField(default=f'{tempfile.gettempdir()}/sftp')
+
+# Enable to show retry logs
+RETRY_SHOW_LOGS = BoolField(default=False)

{everysk_lib-1.9.7 → everysk_lib-1.9.9}/src/everysk/sql/connection.py

@@ -7,18 +7,21 @@
 # without authorization of EVERYSK TECHNOLOGIES is prohibited.
 #
 ###############################################################################
+from collections.abc import Callable
 from contextvars import ContextVar, Token
 from os import getpid
 from types import TracebackType
 from typing import Literal
 
-from psqlpy import Connection, ConnectionPool, QueryResult, SslMode
-from psqlpy import Transaction as _Transaction
+from psycopg import Connection, OperationalError
+from psycopg_pool import ConnectionPool as _ConnectionPool
 
 from everysk.config import settings
 from everysk.core.log import Logger
+from everysk.core.retry import retry
+from everysk.sql.row_factory import cls_row, dict_row
 
-_CONNECTIONS: dict[str, ConnectionPool] = {}
+_CONNECTIONS: dict[str, 'ConnectionPool'] = {}
 log = Logger('everysk-lib-sql-query')
 
 
@@ -27,37 +30,49 @@ def _log(message: str, extra: dict | None = None) -> None:
         log.debug(message, extra=extra)
 
 
+class ConnectionPool(_ConnectionPool):
+    def __del__(self) -> None:
+        # To close the connections when the pool is deleted
+        # https://everysk.atlassian.net/browse/COD-8885
+        try:
+            return super().__del__()
+        except RuntimeError:
+            # The connection is already closed or discarded because we cannot join the current thread
+            # RuntimeError: cannot join current thread
+            pass
+
+        return None
+
+
 class Transaction:
     ## Private attributes
     _connection: Connection
     _pool: ConnectionPool
     _token: Token
-    _transaction: _Transaction
 
     ## Public attributes
-    connection: ContextVar[_Transaction] = ContextVar('postgresql-psqlpy-transaction', default=None)
+    connection: ContextVar[Connection] = ContextVar('postgresql-psqlpy-transaction', default=None)
 
     def __init__(self, dsn: str | None = None) -> None:
         self._pool: ConnectionPool = get_pool(dsn=dsn)
 
-    async def __aenter__(self) -> None:
-        self._connection = await self._pool.connection()
-        self._transaction = self._connection.transaction()
-        await self._transaction.begin()
-        self._token = self.connection.set(self._transaction)
+    def __enter__(self) -> None:
+        self._connection = self._pool.getconn()
+        self._token = self.connection.set(self._connection)
 
         return self
 
-    async def __aexit__(
+    def __exit__(
         self, exc_type: type[BaseException] | None, exc_value: BaseException | None, traceback: TracebackType | None
     ) -> None:
         if exc_type is None:
-            await self._transaction.commit()
+            self._connection.commit()
         else:
-            await self._transaction.rollback()
+            self._connection.rollback()
 
         self.connection.reset(self._token)
-        self._connection.close()
+        # Return the connection to the pool
+        self._pool.putconn(self._connection)
 
         return False
 
@@ -89,7 +104,7 @@ def make_connection_dsn(
     return 'postgresql://{user}:{password}@{host}:{port}/{database}'.format(**options)
 
 
-def get_pool(dsn: str | None = None) -> ConnectionPool:
+def get_pool(dsn: str | None = None, **kwargs) -> ConnectionPool:
     """
     Retrieve a database connection pool for the given DSN.
 
@@ -102,28 +117,38 @@ def get_pool(dsn: str | None = None) -> ConnectionPool:
 
     Args:
         dsn (str | None): The Data Source Name for the database connection. If None, a default DSN is used.
+        **kwargs: Additional keyword arguments to configure the connection pool.
 
     Returns:
         ConnectionPool: The connection pool associated with the given DSN.
     """
     dsn = dsn or make_connection_dsn()
+    # https://www.psycopg.org/psycopg3/docs/api/pool.html
+    kwargs['check'] = ConnectionPool.check_connection
+    kwargs['min_size'] = kwargs.get('min_size', settings.POSTGRESQL_POOL_MIN_SIZE)
+    kwargs['max_size'] = kwargs.get('max_size', settings.POSTGRESQL_POOL_MAX_SIZE)
+    kwargs['max_idle'] = kwargs.get('max_idle', settings.POSTGRESQL_POOL_MAX_IDLE)
+    kwargs['max_lifetime'] = kwargs.get('max_lifetime', settings.POSTGRESQL_POOL_MAX_LIFETIME)
+    kwargs['max_waiting'] = kwargs.get('max_waiting', settings.POSTGRESQL_POOL_MAX_WAITING)
+    kwargs['reconnect_timeout'] = kwargs.get('reconnect_timeout', settings.POSTGRESQL_POOL_RECONNECT_TIMEOUT)
+    kwargs['timeout'] = kwargs.get('timeout', settings.POSTGRESQL_POOL_TIMEOUT)
+    kwargs['open'] = kwargs.get('open', settings.POSTGRESQL_POOL_OPEN)
+
     key = f'{getpid()}:{hash(dsn)}'
     if key not in _CONNECTIONS:
-        _CONNECTIONS[key] = ConnectionPool(
-            dsn=dsn,
-            max_db_pool_size=settings.POSTGRESQL_POOL_MAX_SIZE,
-            ssl_mode=getattr(SslMode, settings.POSTGRESQL_CONNECTION_ENCRYPTION, None),
-        )
+        _CONNECTIONS[key] = ConnectionPool(conninfo=dsn, **kwargs)
+
     return _CONNECTIONS[key]
 
 
-async def execute(
+def execute(
     query: str,
     params: dict | None = None,
     return_type: Literal['dict', 'list'] = 'list',
     dsn: str | None = None,
     cls: type | None = None,
-) -> list[dict] | list[object] | dict:
+    loads: Callable | None = None,
+) -> list[dict] | list[object] | dict | None:
     """
     Execute a query and return the results.
     If return_type is a class, return a list of instances of that class.
@@ -136,25 +161,48 @@ async def execute(
         return_type (Literal['dict', 'list'], optional): The type of return value. Defaults to 'list'.
         dsn (str | None, optional): The DSN to use for the connection. Defaults to None.
         cls (type | None, optional): The class to map the results to. Defaults to None.
+        loads (Callable | None, optional): Optional function to process each value. Defaults to None.
+        retry (int, optional): The current retry count. Defaults to 0.
     """
-    conn = Transaction.connection.get()
+    conn: Connection = Transaction.connection.get()
     if not conn:
         pool: ConnectionPool = get_pool(dsn=dsn)
-        conn = await pool.connection()
+        conn: Connection = pool.getconn()
+        is_transactional = False
         log_message = 'PostgreSQL query executed.'
     else:
+        is_transactional = True
         log_message = 'PostgreSQL query executed within transaction.'
 
     _log(log_message, extra={'labels': {'query': query, 'params': params}})
-    result: QueryResult = await conn.execute(query, params)
-    if not Transaction.connection.get():
-        conn.close()
 
-    if cls:
-        result = result.as_class(cls)
-        if return_type == 'dict':
-            return {row[cls._primary_key]: row for row in result}
+    row_factory = cls_row(cls, loads) if cls else dict_row(loads)
+    # For transactions we let it be controlled externally by the context manager
+    try:
+        with conn.cursor(row_factory=row_factory) as cur:
+            result = retry(cur.execute, {'query': query, 'params': params}, retries=3, exceptions=OperationalError)
+
+            if result.description:
+                result = cur.fetchall()
+            else:
+                result = None
+    except Exception:
+        # On error we need to rollback
+        if not is_transactional:
+            conn.rollback()
+        raise
+
+    else:
+        # Block that only executes if no exception was raised in the try block
+        if not is_transactional:
+            conn.commit()
+
+    finally:
+        # We only return the connection to the pool if we are not in a transaction
+        if not is_transactional:
+            pool.putconn(conn)
 
-        return result
+    if result and cls and return_type == 'dict':
+        return {row[cls._primary_key]: row for row in result}
 
-    return result.result()
+    return result

{everysk_lib-1.9.7 → everysk_lib-1.9.9}/src/everysk/sql/model.py

@@ -7,7 +7,6 @@
 # without authorization of EVERYSK TECHNOLOGIES is prohibited.
 #
 ###############################################################################
-import asyncio
 import inspect
 from copy import deepcopy
 from types import GenericAlias, UnionType
@@ -260,12 +259,8 @@ class BaseModel(dict, metaclass=BaseModelMetaClass):
         return_type: Literal['dict', 'list'] = 'list',
         klass: type | None = None,
     ) -> Any:
-        loop = asyncio.get_event_loop()
         kwargs = {'query': query, 'params': params, 'return_type': return_type, 'dsn': cls._dsn, 'cls': klass}
-        if loop.is_running():
-            return asyncio.run_coroutine_threadsafe(execute(**kwargs), loop).result()
-
-        return loop.run_until_complete(execute(**kwargs))
+        return execute(**kwargs)
 
     @classmethod
     def _generate_attributes(cls) -> None: