diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2009-10-15 18:41:02 +0000 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2009-10-15 18:41:02 +0000 |
| commit | c5571ab19a155f0c11381d65edc07c16902f3fad (patch) | |
| tree | 91d1177483fccf28f5527b1842f838c4623013fe /lib/sqlalchemy/sql | |
| parent | bc351a2dc423987f05f4bf88db4987be507ee0a1 (diff) | |
| download | sqlalchemy-c5571ab19a155f0c11381d65edc07c16902f3fad.tar.gz | |
- an executemany() now requires that all bound parameter
sets require that all keys are present which are
present in the first bound parameter set. The structure
and behavior of an insert/update statement is very much
determined by the first parameter set, including which
defaults are going to fire off, and a minimum of
guesswork is performed with all the rest so that performance
is not impacted. For this reason defaults would otherwise
silently "fail" for missing parameters, so this is now guarded
against. [ticket:1566]
Diffstat (limited to 'lib/sqlalchemy/sql')
| -rw-r--r-- | lib/sqlalchemy/sql/compiler.py | 23 | ||||
| -rw-r--r-- | lib/sqlalchemy/sql/expression.py | 19 |
2 files changed, 28 insertions, 14 deletions
diff --git a/lib/sqlalchemy/sql/compiler.py b/lib/sqlalchemy/sql/compiler.py index 61c6c214f..b204f42b1 100644 --- a/lib/sqlalchemy/sql/compiler.py +++ b/lib/sqlalchemy/sql/compiler.py @@ -231,7 +231,7 @@ class SQLCompiler(engine.Compiled): def is_subquery(self): return len(self.stack) > 1 - def construct_params(self, params=None): + def construct_params(self, params=None, _group_number=None): """return a dictionary of bind parameter keys and values""" if params: @@ -242,7 +242,12 @@ class SQLCompiler(engine.Compiled): pd[name] = params[paramname] break else: - if util.callable(bindparam.value): + if bindparam.required: + if _group_number: + raise exc.InvalidRequestError("A value is required for bind parameter %r, in parameter group %d" % (bindparam.key, _group_number)) + else: + raise exc.InvalidRequestError("A value is required for bind parameter %r" % bindparam.key) + elif util.callable(bindparam.value): pd[name] = bindparam.value() else: pd[name] = bindparam.value @@ -751,8 +756,8 @@ class SQLCompiler(engine.Compiled): return text - def _create_crud_bind_param(self, col, value): - bindparam = sql.bindparam(col.key, value, type_=col.type) + def _create_crud_bind_param(self, col, value, required=False): + bindparam = sql.bindparam(col.key, value, type_=col.type, required=required) self.binds[col.key] = bindparam return self.bindparam_string(self._truncate_bindparam(bindparam)) @@ -770,21 +775,23 @@ class SQLCompiler(engine.Compiled): self.postfetch = [] self.prefetch = [] self.returning = [] - + # no parameters in the statement, no parameters in the # compiled params - return binds for all columns if self.column_keys is None and stmt.parameters is None: return [ - (c, self._create_crud_bind_param(c, None)) + (c, self._create_crud_bind_param(c, None, required=True)) for c in stmt.table.columns ] + required = object() + # if we have statement parameters - set defaults in the # compiled params if self.column_keys is None: parameters = {} else: - parameters = dict((sql._column_as_key(key), None) + parameters = dict((sql._column_as_key(key), required) for key in self.column_keys) if stmt.parameters is not None: @@ -808,7 +815,7 @@ class SQLCompiler(engine.Compiled): if c.key in parameters: value = parameters[c.key] if sql._is_literal(value): - value = self._create_crud_bind_param(c, value) + value = self._create_crud_bind_param(c, value, required=value is required) else: self.postfetch.append(c) value = self.process(value.self_group()) diff --git a/lib/sqlalchemy/sql/expression.py b/lib/sqlalchemy/sql/expression.py index 0ece67e20..0a703ad36 100644 --- a/lib/sqlalchemy/sql/expression.py +++ b/lib/sqlalchemy/sql/expression.py @@ -743,7 +743,7 @@ def table(name, *columns): """ return TableClause(name, *columns) -def bindparam(key, value=None, shortname=None, type_=None, unique=False): +def bindparam(key, value=None, shortname=None, type_=None, unique=False, required=False): """Create a bind parameter clause with the given key. value @@ -762,11 +762,14 @@ def bindparam(key, value=None, shortname=None, type_=None, unique=False): underlying ``key`` modified to a uniquely generated name. mostly useful with value-based bind params. + required + A value is required at execution time. + """ if isinstance(key, ColumnClause): - return _BindParamClause(key.name, value, type_=key.type, unique=unique, shortname=shortname) + return _BindParamClause(key.name, value, type_=key.type, unique=unique, shortname=shortname, required=required) else: - return _BindParamClause(key, value, type_=type_, unique=unique, shortname=shortname) + return _BindParamClause(key, value, type_=type_, unique=unique, shortname=shortname, required=required) def outparam(key, type_=None): """Create an 'OUT' parameter for usage in functions (stored procedures), for @@ -2071,7 +2074,7 @@ class _BindParamClause(ColumnElement): __visit_name__ = 'bindparam' quote = None - def __init__(self, key, value, type_=None, unique=False, isoutparam=False, shortname=None): + def __init__(self, key, value, type_=None, unique=False, isoutparam=False, shortname=None, required=False): """Construct a _BindParamClause. key @@ -2100,7 +2103,10 @@ class _BindParamClause(ColumnElement): modified if another ``_BindParamClause`` of the same name already has been located within the containing ``ClauseElement``. - + + required + a value is required at execution time. + isoutparam if True, the parameter should be treated like a stored procedure "OUT" parameter. @@ -2115,7 +2121,8 @@ class _BindParamClause(ColumnElement): self.value = value self.isoutparam = isoutparam self.shortname = shortname - + self.required = required + if type_ is None: self.type = sqltypes.type_map.get(type(value), sqltypes.NullType)() elif isinstance(type_, type): |
