summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/future/result.py
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2019-06-04 17:29:20 -0400
committerMike Bayer <mike_mp@zzzcomputing.com>2020-02-21 17:53:33 -0500
commitf559f378c47811b5528ad1769cb86925e85fd1e5 (patch)
treefd8325501a96cf1e4280c15f267f63b2af7b5f97 /lib/sqlalchemy/future/result.py
parent93b7767d00267ebe149cabcae7246b6796352eb8 (diff)
downloadsqlalchemy-f559f378c47811b5528ad1769cb86925e85fd1e5.tar.gz
Result initial introduction
This builds on cc718cccc0bf8a01abdf4068c7ea4f3 which moved RowProxy to Row, allowing Row to be more like a named tuple. - KeyedTuple in ORM is replaced with Row - ResultSetMetaData broken out into "simple" and "cursor" versions for ORM and Core, as well as LegacyCursor version. - Row now has _mapping attribute that supplies full mapping behavior. Row and SimpleRow both have named tuple behavior otherwise. LegacyRow has some mapping features on the tuple which emit deprecation warnings (e.g. keys(), values(), etc). the biggest change for mapping->tuple is the behavior of __contains__ which moves from testing of "key in row" to "value in row". - ResultProxy breaks into ResultProxy and FutureResult (interim), the latter has the newer APIs. Made available to dialects using execution options. - internal reflection methods and most tests move off of implicit Row mapping behavior and move to row._mapping, result.mappings() method using future result - a new strategy system for cursor handling replaces the various subclasses of RowProxy - some execution context adjustments. We will leave EC in but refined things like get_result_proxy() and out parameter handling. Dialects for 1.4 will need to adjust from get_result_proxy() to get_result_cursor_strategy(), if they are using this method - out parameter handling now accommodated by get_out_parameter_values() EC method. Oracle changes for this. external dialect for DB2 for example will also need to adjust for this. - deprecate case_insensitive flag for engine / result, this feature is not used mapping-methods on Row are deprecated, and replaced with Row._mapping.<meth>, including: row.keys() -> use row._mapping.keys() row.items() -> use row._mapping.items() row.values() -> use row._mapping.values() key in row -> use key in row._mapping int in row -> use int < len(row) Fixes: #4710 Fixes: #4878 Change-Id: Ieb9085e9bcff564359095b754da9ae0af55679f0
Diffstat (limited to 'lib/sqlalchemy/future/result.py')
-rw-r--r--lib/sqlalchemy/future/result.py171
1 files changed, 171 insertions, 0 deletions
diff --git a/lib/sqlalchemy/future/result.py b/lib/sqlalchemy/future/result.py
new file mode 100644
index 000000000..583ff957a
--- /dev/null
+++ b/lib/sqlalchemy/future/result.py
@@ -0,0 +1,171 @@
+import operator
+
+from .. import util
+from ..engine.result import _baserow_usecext
+from ..engine.result import BaseResult
+from ..engine.result import CursorResultMetaData
+from ..engine.result import DefaultCursorFetchStrategy
+from ..engine.result import Row
+from ..sql import util as sql_util
+from ..sql.base import _generative
+from ..sql.base import Generative
+
+
+class Result(Generative, BaseResult):
+ """Interim "future" result proxy so that dialects can build on
+ upcoming 2.0 patterns.
+
+
+ """
+
+ _process_row = Row
+ _cursor_metadata = CursorResultMetaData
+ _cursor_strategy_cls = DefaultCursorFetchStrategy
+
+ _column_slice_filter = None
+ _post_creational_filter = None
+
+ def close(self):
+ """Close this :class:`.Result`.
+
+ This closes out the underlying DBAPI cursor corresponding
+ to the statement execution, if one is still present. Note that the
+ DBAPI cursor is automatically released when the :class:`.Result`
+ exhausts all available rows. :meth:`.Result.close` is generally
+ an optional method except in the case when discarding a
+ :class:`.Result` that still has additional rows pending for fetch.
+
+ After this method is called, it is no longer valid to call upon
+ the fetch methods, which will raise a :class:`.ResourceClosedError`
+ on subsequent use.
+
+ .. seealso::
+
+ :ref:`connections_toplevel`
+
+ """
+ self._soft_close(hard=True)
+
+ def columns(self, *col_expressions):
+ indexes = []
+ for key in col_expressions:
+ try:
+ rec = self._keymap[key]
+ except KeyError:
+ rec = self._key_fallback(key, True)
+ if rec is None:
+ return None
+
+ index, obj = rec[0:2]
+
+ if index is None:
+ self._metadata._raise_for_ambiguous_column_name(obj)
+ indexes.append(index)
+ return self._column_slices(indexes)
+
+ def scalars(self):
+ result = self._column_slices(0)
+ result._post_creational_filter = operator.itemgetter(0)
+ return result
+
+ @_generative
+ def _column_slices(self, indexes):
+ if _baserow_usecext:
+ self._column_slice_filter = self._metadata._tuplegetter(*indexes)
+ else:
+ self._column_slice_filter = self._metadata._pure_py_tuplegetter(
+ *indexes
+ )
+
+ @_generative
+ def mappings(self):
+ self._post_creational_filter = operator.attrgetter("_mapping")
+
+ def _row_getter(self):
+ process_row = self._process_row
+ metadata = self._metadata
+ keymap = metadata._keymap
+ processors = metadata._processors
+
+ fns = ()
+
+ if self._echo:
+ log = self.context.engine.logger.debug
+
+ def log_row(row):
+ log("Row %r", sql_util._repr_row(row))
+ return row
+
+ fns += (log_row,)
+
+ if self._column_slice_filter:
+ fns += (self._column_slice_filter,)
+
+ if self._post_creational_filter:
+ fns += (self._post_creational_filter,)
+
+ def make_row(row):
+ row = process_row(metadata, processors, keymap, row)
+ for fn in fns:
+ row = fn(row)
+ return row
+
+ return make_row
+
+ def _safe_fetchone_impl(self):
+ try:
+ return self.cursor_strategy.fetchone()
+ except BaseException as e:
+ self.connection._handle_dbapi_exception(
+ e, None, None, self.cursor, self.context
+ )
+
+ def _safe_fetchall_impl(self):
+ try:
+ result = self.cursor_strategy.fetchall()
+ self._soft_close()
+ return result
+ except BaseException as e:
+ self.connection._handle_dbapi_exception(
+ e, None, None, self.cursor, self.context
+ )
+
+ def _safe_fetchmany_impl(self, size=None):
+ try:
+ l = self.process_rows(self.cursor_strategy.fetchmany(size))
+ if len(l) == 0:
+ self._soft_close()
+ return l
+ except BaseException as e:
+ self.connection._handle_dbapi_exception(
+ e, None, None, self.cursor, self.context
+ )
+
+ def __iter__(self):
+ getter = self._row_getter()
+ return (getter(r) for r in self._safe_fetchall_impl())
+
+ def _onerow(self):
+ getter = self._row_getter()
+ row = self._safe_fetchone_impl()
+ if row is None:
+ return None
+ else:
+ return getter(row)
+
+ def all(self):
+ getter = self._row_getter()
+ return [getter(r) for r in self._safe_fetchall_impl()]
+
+ def first(self):
+ getter = self._row_getter()
+ row = self._safe_fetchone_impl()
+ if row is None:
+ return None
+ else:
+ row = getter(row)
+ second_row = self._safe_fetchone_impl()
+ if second_row is not None:
+ self._soft_close()
+ util.warn("Additional rows remain")
+ return row