diff options
| author | mike bayer <mike_mp@zzzcomputing.com> | 2023-03-28 22:05:23 +0000 |
|---|---|---|
| committer | Gerrit Code Review <gerrit@bbpush.zzzcomputing.com> | 2023-03-28 22:05:23 +0000 |
| commit | d878f9074c77aebc698ecc492babba96edae5c05 (patch) | |
| tree | c5e78e0ca891ceaad5d089f186b74d0cafb80186 /lib/sqlalchemy | |
| parent | 50d17f80503176430cd7927a8d27aadc85a6377a (diff) | |
| parent | 2dc24ced0e94c50623aab41c75e42883d68a846a (diff) | |
| download | sqlalchemy-d878f9074c77aebc698ecc492babba96edae5c05.tar.gz | |
Merge "Improve from_select typing" into main
Diffstat (limited to 'lib/sqlalchemy')
| -rw-r--r-- | lib/sqlalchemy/sql/dml.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/lib/sqlalchemy/sql/dml.py b/lib/sqlalchemy/sql/dml.py index dbbf09f1b..316392c0c 100644 --- a/lib/sqlalchemy/sql/dml.py +++ b/lib/sqlalchemy/sql/dml.py @@ -228,8 +228,7 @@ class DMLState(CompileState): def _process_select_values(self, statement: ValuesBase) -> None: assert statement._select_names is not None parameters: MutableMapping[_DMLColumnElement, Any] = { - coercions.expect(roles.DMLColumnRole, name, as_key=True): Null() - for name in statement._select_names + name: Null() for name in statement._select_names } if self._no_parameters: @@ -1195,7 +1194,7 @@ class Insert(ValuesBase): @_generative def from_select( self, - names: List[str], + names: Sequence[_DMLColumnArgument], select: Selectable, include_defaults: bool = True, ) -> Self: @@ -1253,7 +1252,10 @@ class Insert(ValuesBase): "This construct already inserts value expressions" ) - self._select_names = names + self._select_names = [ + coercions.expect(roles.DMLColumnRole, name, as_key=True) + for name in names + ] self._inline = True self.include_insert_from_select_defaults = include_defaults self.select = coercions.expect(roles.DMLSelectRole, select) |
