summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/engine
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2011-01-11 15:22:46 -0500
committerMike Bayer <mike_mp@zzzcomputing.com>2011-01-11 15:22:46 -0500
commit67e0f356b2093fdc03303d50be1f89e75e847c7f (patch)
treee2209edb3a8aeb16702bc47573b9809a8e521db5 /lib/sqlalchemy/engine
parent0342a4886f00b34cf02e0d2d986a0896ba946788 (diff)
downloadsqlalchemy-67e0f356b2093fdc03303d50be1f89e75e847c7f.tar.gz
- A TypeDecorator of Integer can be used with a primary key
column, and the "autoincrement" feature of various dialects as well as the "sqlite_autoincrement" flag will honor the underlying database type as being Integer-based. [ticket:2005] - Result-row processors are applied to pre-executed SQL defaults, as well as cursor.lastrowid, when determining the contents of result.inserted_primary_key. [ticket:2006] - Bind parameters present in the "columns clause" of a select are now auto-labeled like other "anonymous" clauses, which among other things allows their "type" to be meaningful when the row is fetched, as in result row processors. - TypeDecorator is present in the "sqlalchemy" import space.
Diffstat (limited to 'lib/sqlalchemy/engine')
-rw-r--r--lib/sqlalchemy/engine/base.py2
-rw-r--r--lib/sqlalchemy/engine/default.py48
2 files changed, 31 insertions, 19 deletions
diff --git a/lib/sqlalchemy/engine/base.py b/lib/sqlalchemy/engine/base.py
index eb48c29d6..3bdcad2ac 100644
--- a/lib/sqlalchemy/engine/base.py
+++ b/lib/sqlalchemy/engine/base.py
@@ -1233,7 +1233,7 @@ class Connection(Connectable):
self._handle_dbapi_exception(e, None, None, None, None)
raise
- ret = ctx._exec_default(default)
+ ret = ctx._exec_default(default, None)
if self.should_close_with_result:
self.close()
return ret
diff --git a/lib/sqlalchemy/engine/default.py b/lib/sqlalchemy/engine/default.py
index 76077778c..eacbef8f9 100644
--- a/lib/sqlalchemy/engine/default.py
+++ b/lib/sqlalchemy/engine/default.py
@@ -418,9 +418,9 @@ class DefaultExecutionContext(base.ExecutionContext):
self.cursor = self.create_cursor()
if self.isinsert or self.isupdate:
- self.__process_defaults()
self.postfetch_cols = self.compiled.postfetch
self.prefetch_cols = self.compiled.prefetch
+ self.__process_defaults()
processors = compiled._bind_processors
@@ -532,7 +532,7 @@ class DefaultExecutionContext(base.ExecutionContext):
else:
return autocommit
- def _execute_scalar(self, stmt):
+ def _execute_scalar(self, stmt, proc):
"""Execute a string statement on the current cursor, returning a
scalar result.
@@ -553,7 +553,11 @@ class DefaultExecutionContext(base.ExecutionContext):
default_params = {}
conn._cursor_execute(self.cursor, stmt, default_params)
- return self.cursor.fetchone()[0]
+ r = self.cursor.fetchone()[0]
+ if proc:
+ return proc(r)
+ else:
+ return r
@property
def connection(self):
@@ -623,8 +627,14 @@ class DefaultExecutionContext(base.ExecutionContext):
table = self.compiled.statement.table
lastrowid = self.get_lastrowid()
self.inserted_primary_key = [
- c is table._autoincrement_column and lastrowid or v
- for c, v in zip(table.primary_key, self.inserted_primary_key)
+ c is table._autoincrement_column and (
+ proc and proc(lastrowid)
+ or lastrowid
+ ) or v
+ for c, v, proc in zip(
+ table.primary_key,
+ self.inserted_primary_key,
+ self.compiled._pk_processors)
]
def _fetch_implicit_returning(self, resultproxy):
@@ -688,9 +698,9 @@ class DefaultExecutionContext(base.ExecutionContext):
self.root_connection._handle_dbapi_exception(e, None, None, None, self)
raise
- def _exec_default(self, default):
+ def _exec_default(self, default, proc):
if default.is_sequence:
- return self.fire_sequence(default)
+ return self.fire_sequence(default, proc)
elif default.is_callable:
return default.arg(self)
elif default.is_clause_element:
@@ -702,17 +712,17 @@ class DefaultExecutionContext(base.ExecutionContext):
else:
return default.arg
- def get_insert_default(self, column):
+ def get_insert_default(self, column, proc):
if column.default is None:
return None
else:
- return self._exec_default(column.default)
+ return self._exec_default(column.default, proc)
- def get_update_default(self, column):
+ def get_update_default(self, column, proc):
if column.onupdate is None:
return None
else:
- return self._exec_default(column.onupdate)
+ return self._exec_default(column.onupdate, proc)
def __process_defaults(self):
"""Generate default values for compiled insert/update statements,
@@ -726,7 +736,7 @@ class DefaultExecutionContext(base.ExecutionContext):
# pre-determine scalar Python-side defaults
# to avoid many calls of get_insert_default()/
# get_update_default()
- for c in self.compiled.prefetch:
+ for c in self.prefetch_cols:
if self.isinsert and c.default and c.default.is_scalar:
scalar_defaults[c] = c.default.arg
elif self.isupdate and c.onupdate and c.onupdate.is_scalar:
@@ -734,13 +744,14 @@ class DefaultExecutionContext(base.ExecutionContext):
for param in self.compiled_parameters:
self.current_parameters = param
- for c in self.compiled.prefetch:
+ for c, proc in zip(self.prefetch_cols,
+ self.compiled._prefetch_processors):
if c in scalar_defaults:
val = scalar_defaults[c]
elif self.isinsert:
- val = self.get_insert_default(c)
+ val = self.get_insert_default(c, proc)
else:
- val = self.get_update_default(c)
+ val = self.get_update_default(c, proc)
if val is not None:
param[c.key] = val
del self.current_parameters
@@ -748,11 +759,12 @@ class DefaultExecutionContext(base.ExecutionContext):
self.current_parameters = compiled_parameters = \
self.compiled_parameters[0]
- for c in self.compiled.prefetch:
+ for c, proc in zip(self.compiled.prefetch,
+ self.compiled._prefetch_processors):
if self.isinsert:
- val = self.get_insert_default(c)
+ val = self.get_insert_default(c, proc)
else:
- val = self.get_update_default(c)
+ val = self.get_update_default(c, proc)
if val is not None:
compiled_parameters[c.key] = val