?HAMA

Edit File Size: 4.65 KB

//proc/thread-self/root/opt/hc_python/lib/python3.12/site-packages/sqlalchemy/engine/characteristics.py

Text Base64

# engine/characteristics.py
# Copyright (C) 2005-2025 the SQLAlchemy authors and contributors
# <see AUTHORS file>
#
# This module is part of SQLAlchemy and is released under
# the MIT License: https://www.opensource.org/licenses/mit-license.php
from __future__ import annotations

import abc
import typing
from typing import Any
from typing import ClassVar

if typing.TYPE_CHECKING:
    from .base import Connection
    from .interfaces import DBAPIConnection
    from .interfaces import Dialect

class ConnectionCharacteristic(abc.ABC):
    """An abstract base for an object that can set, get and reset a
    per-connection characteristic, typically one that gets reset when the
    connection is returned to the connection pool.

transaction isolation is the canonical example, and the
    ``IsolationLevelCharacteristic`` implementation provides this for the
    ``DefaultDialect``.

The ``ConnectionCharacteristic`` class should call upon the ``Dialect`` for
    the implementation of each method.   The object exists strictly to serve as
    a dialect visitor that can be placed into the
    ``DefaultDialect.connection_characteristics`` dictionary where it will take
    effect for calls to :meth:`_engine.Connection.execution_options` and
    related APIs.

.. versionadded:: 1.4

"""

__slots__ = ()

transactional: ClassVar[bool] = False

@abc.abstractmethod
    def reset_characteristic(
        self, dialect: Dialect, dbapi_conn: DBAPIConnection
    ) -> None:
        """Reset the characteristic on the DBAPI connection to its default
        value."""

@abc.abstractmethod
    def set_characteristic(
        self, dialect: Dialect, dbapi_conn: DBAPIConnection, value: Any
    ) -> None:
        """set characteristic on the DBAPI connection to a given value."""

def set_connection_characteristic(
        self,
        dialect: Dialect,
        conn: Connection,
        dbapi_conn: DBAPIConnection,
        value: Any,
    ) -> None:
        """set characteristic on the :class:`_engine.Connection` to a given
        value.

.. versionadded:: 2.0.30 - added to support elements that are local
           to the :class:`_engine.Connection` itself.

"""
        self.set_characteristic(dialect, dbapi_conn, value)

@abc.abstractmethod
    def get_characteristic(
        self, dialect: Dialect, dbapi_conn: DBAPIConnection
    ) -> Any:
        """Given a DBAPI connection, get the current value of the
        characteristic.

"""

def get_connection_characteristic(
        self, dialect: Dialect, conn: Connection, dbapi_conn: DBAPIConnection
    ) -> Any:
        """Given a :class:`_engine.Connection`, get the current value of the
        characteristic.

.. versionadded:: 2.0.30 - added to support elements that are local
           to the :class:`_engine.Connection` itself.

"""
        return self.get_characteristic(dialect, dbapi_conn)

class IsolationLevelCharacteristic(ConnectionCharacteristic):
    """Manage the isolation level on a DBAPI connection"""

transactional: ClassVar[bool] = True

def reset_characteristic(
        self, dialect: Dialect, dbapi_conn: DBAPIConnection
    ) -> None:
        dialect.reset_isolation_level(dbapi_conn)

def set_characteristic(
        self, dialect: Dialect, dbapi_conn: DBAPIConnection, value: Any
    ) -> None:
        dialect._assert_and_set_isolation_level(dbapi_conn, value)

def get_characteristic(
        self, dialect: Dialect, dbapi_conn: DBAPIConnection
    ) -> Any:
        return dialect.get_isolation_level(dbapi_conn)

class LoggingTokenCharacteristic(ConnectionCharacteristic):
    """Manage the 'logging_token' option of a :class:`_engine.Connection`.

.. versionadded:: 2.0.30

"""

transactional: ClassVar[bool] = False

def reset_characteristic(
        self, dialect: Dialect, dbapi_conn: DBAPIConnection
    ) -> None:
        pass

def set_characteristic(
        self, dialect: Dialect, dbapi_conn: DBAPIConnection, value: Any
    ) -> None:
        raise NotImplementedError()

def set_connection_characteristic(
        self,
        dialect: Dialect,
        conn: Connection,
        dbapi_conn: DBAPIConnection,
        value: Any,
    ) -> None:
        if value:
            conn._message_formatter = lambda msg: "[%s] %s" % (value, msg)
        else:
            del conn._message_formatter

def get_characteristic(
        self, dialect: Dialect, dbapi_conn: DBAPIConnection
    ) -> Any:
        raise NotImplementedError()

def get_connection_characteristic(
        self, dialect: Dialect, conn: Connection, dbapi_conn: DBAPIConnection
    ) -> Any:
        return conn._execution_options.get("logging_token", None)

Exit & Reset Text mode: syntax highlighting auto-detects file type.

Name	Size	Perms	Modified	Actions
__pycache__ DIR	-	drwxr-xr-x	2025-04-04 08:02:02	Edit Download recursive recursive
base.py	120.02 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
characteristics.py	4.65 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
create.py	32.44 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
cursor.py	74.52 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
default.py	82.72 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
events.py	36.52 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
interfaces.py	110.38 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
mock.py	4.08 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
processors.py	2.32 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
reflection.py	73.62 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
result.py	75.82 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
row.py	11.75 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
strategies.py	442 B	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
url.py	30.34 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
util.py	5.55 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
_py_processors.py	3.66 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
_py_row.py	3.70 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
_py_util.py	2.43 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive
__init__.py	2.75 KB	lrw-r--r--	2025-04-04 08:02:01	Edit Download recursive recursive

Directory Contents