summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy
diff options
context:
space:
mode:
authormike bayer <mike_mp@zzzcomputing.com>2023-03-28 22:05:23 +0000
committerGerrit Code Review <gerrit@bbpush.zzzcomputing.com>2023-03-28 22:05:23 +0000
commitd878f9074c77aebc698ecc492babba96edae5c05 (patch)
treec5e78e0ca891ceaad5d089f186b74d0cafb80186 /lib/sqlalchemy
parent50d17f80503176430cd7927a8d27aadc85a6377a (diff)
parent2dc24ced0e94c50623aab41c75e42883d68a846a (diff)
downloadsqlalchemy-d878f9074c77aebc698ecc492babba96edae5c05.tar.gz
Merge "Improve from_select typing" into main
Diffstat (limited to 'lib/sqlalchemy')
-rw-r--r--lib/sqlalchemy/sql/dml.py10
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)