mirror of
https://github.com/sqlalchemy/sqlalchemy.git
synced 2026-07-11 01:09:29 -04:00
Mike Bayer
b3105b7e3a
Added new execution option :paramref:`_engine.Connection.execution_options.driver_column_names`. This option disables the "name normalize" step that takes place against the DBAPI ``cursor.description`` for uppercase-default backends like Oracle, and will cause the keys of a result set (e.g. named tuple names, dictionary keys in :attr:`.Row._mapping`, etc.) to be exactly what was delivered in cursor.description. This is mostly useful for plain textual statements using :func:`_sql.text` or :meth:`_engine.Connection.exec_driver_sql`. Fixes: #10789 Change-Id: Ib647b25bb53492fa839af04dd032d9f061e630af
98 lines
3.2 KiB
Python
98 lines
3.2 KiB
Python
from sqlalchemy import testing
|
|
from sqlalchemy.engine.base import Connection
|
|
from sqlalchemy.engine.base import Engine
|
|
from sqlalchemy.engine.interfaces import CoreExecuteOptionsParameter
|
|
from sqlalchemy.ext.asyncio.engine import AsyncConnection
|
|
from sqlalchemy.ext.asyncio.engine import AsyncEngine
|
|
from sqlalchemy.orm._typing import OrmExecuteOptionsParameter
|
|
from sqlalchemy.orm.query import Query
|
|
from sqlalchemy.sql.base import Executable
|
|
from sqlalchemy.testing import fixtures
|
|
from sqlalchemy.testing.assertions import eq_
|
|
|
|
engine_execution_options = {
|
|
"compiled_cache": "Optional[CompiledCacheType]",
|
|
"logging_token": "str",
|
|
"isolation_level": "IsolationLevel",
|
|
"insertmanyvalues_page_size": "int",
|
|
"schema_translate_map": "Optional[SchemaTranslateMapType]",
|
|
"opt": "Any",
|
|
}
|
|
core_execution_options = {
|
|
**engine_execution_options,
|
|
"no_parameters": "bool",
|
|
"stream_results": "bool",
|
|
"max_row_buffer": "int",
|
|
"yield_per": "int",
|
|
"preserve_rowcount": "bool",
|
|
"driver_column_names": "bool",
|
|
}
|
|
|
|
orm_dql_execution_options = {
|
|
**core_execution_options,
|
|
"populate_existing": "bool",
|
|
"autoflush": "bool",
|
|
}
|
|
|
|
orm_dml_execution_options = {
|
|
"synchronize_session": "SynchronizeSessionArgument",
|
|
"dml_strategy": "DMLStrategyArgument",
|
|
"is_delete_using": "bool",
|
|
"is_update_from": "bool",
|
|
"render_nulls": "bool",
|
|
}
|
|
|
|
orm_execution_options = {
|
|
**orm_dql_execution_options,
|
|
**orm_dml_execution_options,
|
|
}
|
|
|
|
|
|
class OverloadTest(fixtures.TestBase):
|
|
# NOTE: get_overloads is python 3.11. typing_extensions implements it
|
|
# but for it to work the typing_extensions overload needs to be use and
|
|
# it can only be imported directly from typing_extensions in all modules
|
|
# that use it otherwise flake8 (pyflakes actually) will flag it with F811
|
|
__requires__ = ("python311",)
|
|
|
|
@testing.combinations(
|
|
(Engine, engine_execution_options),
|
|
(Connection, core_execution_options),
|
|
(AsyncEngine, engine_execution_options),
|
|
(AsyncConnection, core_execution_options),
|
|
(Query, orm_dql_execution_options),
|
|
(Executable, orm_execution_options),
|
|
)
|
|
def test_methods(self, class_, expected):
|
|
from typing import get_overloads
|
|
|
|
overloads = get_overloads(getattr(class_, "execution_options"))
|
|
eq_(len(overloads), 2)
|
|
annotations = overloads[0].__annotations__.copy()
|
|
annotations.pop("self", None)
|
|
annotations.pop("return", None)
|
|
eq_(annotations, expected)
|
|
annotations = overloads[1].__annotations__.copy()
|
|
annotations.pop("self", None)
|
|
annotations.pop("return", None)
|
|
eq_(annotations, {"opt": "Any"})
|
|
|
|
@testing.combinations(
|
|
(CoreExecuteOptionsParameter, core_execution_options),
|
|
(OrmExecuteOptionsParameter, orm_execution_options),
|
|
)
|
|
def test_typed_dicts(self, typ, expected):
|
|
# we currently expect these to be union types with first entry
|
|
# is the typed dict
|
|
|
|
typed_dict = typ.__args__[0]
|
|
|
|
expected = dict(expected)
|
|
expected.pop("opt")
|
|
|
|
assert_annotations = {
|
|
key: fwd_ref.__forward_arg__
|
|
for key, fwd_ref in typed_dict.__annotations__.items()
|
|
}
|
|
eq_(assert_annotations, expected)
|