diff options
| author | Federico Caselli <cfederico87@gmail.com> | 2020-05-21 21:50:49 +0200 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2020-05-23 19:40:06 -0400 |
| commit | f79953a874c201a31a8972b999d18547bf227f25 (patch) | |
| tree | e71779e41fc3d7dc8a1bacfd0aea6716a80c5856 /lib/sqlalchemy/engine | |
| parent | 31d1846b76baec6ffa4c65bd72456a58f7b2ef1b (diff) | |
| download | sqlalchemy-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.py | 20 | ||||
| -rw-r--r-- | lib/sqlalchemy/engine/row.py | 43 |
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) |
