diff options
| author | Mike Bayer <mike_mp@zzzcomputing.com> | 2007-09-04 00:08:57 +0000 |
|---|---|---|
| committer | Mike Bayer <mike_mp@zzzcomputing.com> | 2007-09-04 00:08:57 +0000 |
| commit | 3126d464e7124cde24b18ba7efc318913d2ac40d (patch) | |
| tree | 33dd8fdda1ea8a3aae5c75cfbbdded60a1b01997 /lib/sqlalchemy | |
| parent | c9924a4a145f06eac427fe60c54d4c58b894167f (diff) | |
| download | sqlalchemy-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.py | 4 | ||||
| -rw-r--r-- | lib/sqlalchemy/engine/base.py | 66 | ||||
| -rw-r--r-- | lib/sqlalchemy/sql/compiler.py | 26 | ||||
| -rw-r--r-- | lib/sqlalchemy/sql/expression.py | 28 |
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 |
