summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/engine
diff options
context:
space:
mode:
authorFederico Caselli <cfederico87@gmail.com>2020-05-21 21:50:49 +0200
committerMike Bayer <mike_mp@zzzcomputing.com>2020-05-23 19:40:06 -0400
commitf79953a874c201a31a8972b999d18547bf227f25 (patch)
treee71779e41fc3d7dc8a1bacfd0aea6716a80c5856 /lib/sqlalchemy/engine
parent31d1846b76baec6ffa4c65bd72456a58f7b2ef1b (diff)
downloadsqlalchemy-f79953a874c201a31a8972b999d18547bf227f25.tar.gz
Avoid proxy functions in row functions
This streamlines a bit for non-C implementations, however also adds and tests behavioral contracts that mappings should not allow integer or slice access and should behave like a Python mapping in that it raises KeyError for an integer and TypeError for a slice. Py3/Py2/C/noC :) References: #5340 Change-Id: Id3cef452dc8a526b8371c90c5ca2bbb240b25c26
Diffstat (limited to 'lib/sqlalchemy/engine')
-rw-r--r--lib/sqlalchemy/engine/result.py20
-rw-r--r--lib/sqlalchemy/engine/row.py43
2 files changed, 37 insertions, 26 deletions
diff --git a/lib/sqlalchemy/engine/result.py b/lib/sqlalchemy/engine/result.py
index 109ab41fe..ce844eb40 100644
--- a/lib/sqlalchemy/engine/result.py
+++ b/lib/sqlalchemy/engine/result.py
@@ -40,7 +40,7 @@ else:
operator.methodcaller("_get_by_key_impl_mapping", index)
for index in indexes
]
- return lambda rec: tuple(getter(rec) for getter in getters)
+ return lambda rec: tuple([getter(rec) for getter in getters])
class ResultMetaData(object):
@@ -775,15 +775,19 @@ class Result(InPlaceGenerative):
uniques, strategy = self._unique_strategy
def filterrows(make_row, rows, strategy, uniques):
+ if strategy:
+ made_rows = (
+ (made_row, strategy(made_row))
+ for made_row in [make_row(row) for row in rows]
+ )
+ else:
+ made_rows = (
+ (made_row, made_row)
+ for made_row in [make_row(row) for row in rows]
+ )
return [
made_row
- for made_row, sig_row in [
- (
- made_row,
- strategy(made_row) if strategy else made_row,
- )
- for made_row in [make_row(row) for row in rows]
- ]
+ for made_row, sig_row in made_rows
if sig_row not in uniques and not uniques.add(sig_row)
]
diff --git a/lib/sqlalchemy/engine/row.py b/lib/sqlalchemy/engine/row.py
index d279776ce..70f45c82c 100644
--- a/lib/sqlalchemy/engine/row.py
+++ b/lib/sqlalchemy/engine/row.py
@@ -103,16 +103,19 @@ except ImportError:
def __getitem__(self, key):
return self._data[key]
- def _subscript_impl(self, key, ismapping):
+ def _get_by_key_impl(self, key):
+ if self._key_style == KEY_INTEGER_ONLY:
+ return self._data[key]
+
+ # the following is all LegacyRow support. none of this
+ # should be called if not LegacyRow
+ # assert isinstance(self, LegacyRow)
+
try:
rec = self._keymap[key]
except KeyError as ke:
rec = self._parent._key_fallback(key, ke)
except TypeError:
- # the non-C version detects a slice using TypeError.
- # this is pretty inefficient for the slice use case
- # but is more efficient for the integer use case since we
- # don't have to check it up front.
if isinstance(key, slice):
return tuple(self._data[key])
else:
@@ -124,7 +127,6 @@ except ImportError:
elif (
self._key_style == KEY_OBJECTS_BUT_WARN
- and not ismapping
and mdindex != key
and not isinstance(key, int)
):
@@ -132,14 +134,22 @@ except ImportError:
return self._data[mdindex]
- def _get_by_key_impl(self, key):
- return self._subscript_impl(key, False)
-
def _get_by_key_impl_mapping(self, key):
- # the C code has two different methods so that we can distinguish
- # between tuple-like keys (integers, slices) and mapping-like keys
- # (strings, objects)
- return self._subscript_impl(key, True)
+ try:
+ rec = self._keymap[key]
+ except KeyError as ke:
+ rec = self._parent._key_fallback(key, ke)
+
+ mdindex = rec[MD_INDEX]
+ if mdindex is None:
+ self._parent._raise_for_ambiguous_column_name(rec)
+ elif (
+ self._key_style == KEY_OBJECTS_ONLY
+ and int in key.__class__.__mro__
+ ):
+ raise KeyError(key)
+
+ return self._data[mdindex]
def __getattr__(self, name):
try:
@@ -348,9 +358,7 @@ class LegacyRow(Row):
return self._parent._contains(key, self)
if not _baserow_usecext:
-
- def __getitem__(self, key):
- return self._get_by_key_impl(key)
+ __getitem__ = BaseRow._get_by_key_impl
@util.deprecated(
"1.4",
@@ -510,8 +518,7 @@ class RowMapping(BaseRow, collections_abc.Mapping):
if not _baserow_usecext:
- def __getitem__(self, key):
- return self._get_by_key_impl(key)
+ __getitem__ = BaseRow._get_by_key_impl_mapping
def _values_impl(self):
return list(self._data)