summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2007-09-04 00:08:57 +0000
committerMike Bayer <mike_mp@zzzcomputing.com>2007-09-04 00:08:57 +0000
commit3126d464e7124cde24b18ba7efc318913d2ac40d (patch)
tree33dd8fdda1ea8a3aae5c75cfbbdded60a1b01997 /lib/sqlalchemy
parentc9924a4a145f06eac427fe60c54d4c58b894167f (diff)
downloadsqlalchemy-3126d464e7124cde24b18ba7efc318913d2ac40d.tar.gz
- removed "parameters" argument from clauseelement.compile(), replaced with
"column_keys". the parameters sent to execute() only interact with the insert/update statement compilation process in terms of the column names present but not the values for those columns. produces more consistent execute/executemany behavior, simplifies things a bit internally.
Diffstat (limited to 'lib/sqlalchemy')
-rw-r--r--lib/sqlalchemy/databases/mssql.py4
-rw-r--r--lib/sqlalchemy/engine/base.py66
-rw-r--r--lib/sqlalchemy/sql/compiler.py26
-rw-r--r--lib/sqlalchemy/sql/expression.py28
4 files changed, 64 insertions, 60 deletions
diff --git a/lib/sqlalchemy/databases/mssql.py b/lib/sqlalchemy/databases/mssql.py
index 3e8949e14..7b0defc2a 100644
--- a/lib/sqlalchemy/databases/mssql.py
+++ b/lib/sqlalchemy/databases/mssql.py
@@ -847,8 +847,8 @@ dialect_mapping = {
class MSSQLCompiler(compiler.DefaultCompiler):
- def __init__(self, dialect, statement, parameters, **kwargs):
- super(MSSQLCompiler, self).__init__(dialect, statement, parameters, **kwargs)
+ def __init__(self, *args, **kwargs):
+ super(MSSQLCompiler, self).__init__(*args, **kwargs)
self.tablealiases = {}
def get_select_precolumns(self, select):
diff --git a/lib/sqlalchemy/engine/base.py b/lib/sqlalchemy/engine/base.py
index bd6f5b97c..32bf7b780 100644
--- a/lib/sqlalchemy/engine/base.py
+++ b/lib/sqlalchemy/engine/base.py
@@ -421,7 +421,7 @@ class Compiled(object):
defaults.
"""
- def __init__(self, dialect, statement, parameters, bind=None):
+ def __init__(self, dialect, statement, column_keys=None, bind=None):
"""Construct a new ``Compiled`` object.
dialect
@@ -430,26 +430,16 @@ class Compiled(object):
statement
``ClauseElement`` to be compiled.
- parameters
- Optional dictionary indicating a set of bind parameters
- specified with this ``Compiled`` object. These parameters
- are the *default* values corresponding to the
- ``ClauseElement``'s ``_BindParamClauses`` when the
- ``Compiled`` is executed. In the case of an ``INSERT`` or
- ``UPDATE`` statement, these parameters will also result in
- the creation of new ``_BindParamClause`` objects for each
- key and will also affect the generated column list in an
- ``INSERT`` statement and the ``SET`` clauses of an
- ``UPDATE`` statement. The keys of the parameter dictionary
- can either be the string names of columns or
- ``_ColumnClause`` objects.
+ column_keys
+ a list of column names to be compiled into an INSERT or UPDATE
+ statement.
bind
Optional Engine or Connection to compile this statement against.
"""
self.dialect = dialect
self.statement = statement
- self.parameters = parameters
+ self.column_keys = column_keys
self.bind = bind
self.can_execute = statement.supports_execution()
@@ -778,8 +768,8 @@ class Connection(Connectable):
return self.execute(object, *multiparams, **params).scalar()
- def statement_compiler(self, statement, parameters, **kwargs):
- return self.dialect.statement_compiler(self.dialect, statement, parameters, bind=self, **kwargs)
+ def statement_compiler(self, statement, **kwargs):
+ return self.dialect.statement_compiler(self.dialect, statement, bind=self, **kwargs)
def execute(self, object, *multiparams, **params):
"""Executes and returns a ResultProxy."""
@@ -808,25 +798,43 @@ class Connection(Connectable):
parameters = list(multiparams)
return parameters
+ def __distill_params_and_keys(self, multiparams, params):
+ if multiparams is None or len(multiparams) == 0:
+ if params:
+ parameters = params
+ keys = params.keys()
+ else:
+ parameters = None
+ keys = []
+ executemany = False
+ elif len(multiparams) == 1 and isinstance(multiparams[0], (list, tuple, dict)):
+ parameters = multiparams[0]
+ if isinstance(parameters, dict):
+ keys = parameters.keys()
+ else:
+ keys = parameters[0].keys()
+ executemany = False
+ else:
+ parameters = list(multiparams)
+ keys = parameters[0].keys()
+ executemany = True
+ return (parameters, keys, executemany)
+
def _execute_function(self, func, multiparams, params):
return self._execute_clauseelement(func.select(), multiparams, params)
def _execute_clauseelement(self, elem, multiparams=None, params=None):
- if multiparams:
- param = multiparams[0]
- executemany = len(multiparams) > 1
- else:
- param = params
- executemany = False
- return self._execute_compiled(elem.compile(dialect=self.dialect, parameters=param, inline=executemany), multiparams, params)
+ (params, keys, executemany) = self.__distill_params_and_keys(multiparams, params)
+ return self._execute_compiled(elem.compile(dialect=self.dialect, column_keys=keys, inline=executemany), distilled_params=params)
- def _execute_compiled(self, compiled, multiparams=None, params=None):
+ def _execute_compiled(self, compiled, multiparams=None, params=None, distilled_params=None):
"""Execute a sql.Compiled object."""
if not compiled.can_execute:
raise exceptions.ArgumentError("Not an executeable clause: %s" % (str(compiled)))
- params = self.__distill_params(multiparams, params)
- context = self.__create_execution_context(compiled=compiled, parameters=params)
+ if distilled_params is None:
+ distilled_params = self.__distill_params(multiparams, params)
+ context = self.__create_execution_context(compiled=compiled, parameters=distilled_params)
context.pre_execution()
self.__execute_raw(context)
@@ -1119,8 +1127,8 @@ class Engine(Connectable):
connection = self.contextual_connect(close_with_result=True)
return connection._execute_compiled(compiled, multiparams, params)
- def statement_compiler(self, statement, parameters, **kwargs):
- return self.dialect.statement_compiler(self.dialect, statement, parameters, bind=self, **kwargs)
+ def statement_compiler(self, statement, **kwargs):
+ return self.dialect.statement_compiler(self.dialect, statement, bind=self, **kwargs)
def connect(self, **kwargs):
"""Return a newly allocated Connection object."""
diff --git a/lib/sqlalchemy/sql/compiler.py b/lib/sqlalchemy/sql/compiler.py
index 1cfebdc27..eb416803a 100644
--- a/lib/sqlalchemy/sql/compiler.py
+++ b/lib/sqlalchemy/sql/compiler.py
@@ -90,7 +90,7 @@ class DefaultCompiler(engine.Compiled, visitors.ClauseVisitor):
operators = OPERATORS
- def __init__(self, dialect, statement, parameters=None, inline=False, **kwargs):
+ def __init__(self, dialect, statement, column_keys=None, inline=False, **kwargs):
"""Construct a new ``DefaultCompiler`` object.
dialect
@@ -99,16 +99,12 @@ class DefaultCompiler(engine.Compiled, visitors.ClauseVisitor):
statement
ClauseElement to be compiled
- parameters
- optional dictionary indicating a set of bind parameters
- specified with this Compiled object. These parameters are
- the *default* key/value pairs when the Compiled is executed,
- and also may affect the actual compilation, as in the case
- of an INSERT where the actual columns inserted will
- correspond to the keys present in the parameters.
+ column_keys
+ a list of column names to be compiled into an INSERT or UPDATE
+ statement.
"""
- super(DefaultCompiler, self).__init__(dialect, statement, parameters, **kwargs)
+ super(DefaultCompiler, self).__init__(dialect, statement, column_keys, **kwargs)
# if we are insert/update. set to true when we visit an INSERT or UPDATE
self.isinsert = self.isupdate = False
@@ -217,12 +213,10 @@ class DefaultCompiler(engine.Compiled, visitors.ClauseVisitor):
to produce a ClauseParameters structure, representing the bind arguments
for a single statement execution, or one element of an executemany execution.
"""
-
+
d = sql_util.ClauseParameters(self.dialect, self.positiontup)
- pd = self.parameters or {}
- if params is not None:
- pd.update(params)
+ pd = params or {}
bind_names = self.bind_names
for key, bind in self.binds.iteritems():
@@ -658,15 +652,15 @@ class DefaultCompiler(engine.Compiled, visitors.ClauseVisitor):
# no parameters in the statement, no parameters in the
# compiled params - return binds for all columns
- if self.parameters is None and stmt.parameters is None:
+ if self.column_keys is None and stmt.parameters is None:
return [(c, create_bind_param(c, None)) for c in stmt.table.columns]
# if we have statement parameters - set defaults in the
# compiled params
- if self.parameters is None:
+ if self.column_keys is None:
parameters = {}
else:
- parameters = dict([(getattr(k, 'key', k), v) for k, v in self.parameters.iteritems()])
+ parameters = dict([(getattr(key, 'key', key), None) for key in self.column_keys])
if stmt.parameters is not None:
for k, v in stmt.parameters.iteritems():
diff --git a/lib/sqlalchemy/sql/expression.py b/lib/sqlalchemy/sql/expression.py
index ac56289e8..f88c418eb 100644
--- a/lib/sqlalchemy/sql/expression.py
+++ b/lib/sqlalchemy/sql/expression.py
@@ -963,19 +963,24 @@ class ClauseElement(object):
def execute(self, *multiparams, **params):
"""Compile and execute this ``ClauseElement``."""
-
- if multiparams:
- compile_params = multiparams[0]
+
+ if len(multiparams) == 0:
+ keys = params.keys()
+ elif isinstance(multiparams[0], dict):
+ keys = multiparams[0].keys()
+ elif isinstance(multiparams[0], (list, tuple)):
+ keys = multiparams[0][0].keys()
else:
- compile_params = params
- return self.compile(bind=self.bind, parameters=compile_params, inline=(len(multiparams) > 1)).execute(*multiparams, **params)
+ keys = None
+
+ return self.compile(bind=self.bind, column_keys=keys, inline=(len(multiparams) > 1)).execute(*multiparams, **params)
def scalar(self, *multiparams, **params):
"""Compile and execute this ``ClauseElement``, returning the result's scalar representation."""
return self.execute(*multiparams, **params).scalar()
- def compile(self, bind=None, parameters=None, compiler=None, dialect=None, inline=False):
+ def compile(self, bind=None, column_keys=None, compiler=None, dialect=None, inline=False):
"""Compile this SQL expression.
Uses the given ``Compiler``, or the given ``AbstractDialect``
@@ -999,21 +1004,18 @@ class ClauseElement(object):
``SET`` and ``VALUES`` clause of those statements.
"""
- if isinstance(parameters, (list, tuple)):
- parameters = parameters[0]
-
if compiler is None:
if dialect is not None:
- compiler = dialect.statement_compiler(dialect, self, parameters, inline=inline)
+ compiler = dialect.statement_compiler(dialect, self, column_keys=column_keys, inline=inline)
elif bind is not None:
- compiler = bind.statement_compiler(self, parameters, inline=inline)
+ compiler = bind.statement_compiler(self, column_keys=column_keys, inline=inline)
elif self.bind is not None:
- compiler = self.bind.statement_compiler(self, parameters, inline=inline)
+ compiler = self.bind.statement_compiler(self, column_keys=column_keys, inline=inline)
if compiler is None:
from sqlalchemy.engine.default import DefaultDialect
dialect = DefaultDialect()
- compiler = dialect.statement_compiler(dialect, self, parameters=parameters, inline=inline)
+ compiler = dialect.statement_compiler(dialect, self, column_keys=column_keys, inline=inline)
compiler.compile()
return compiler