summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/sql/dml.py
diff options
context:
space:
mode:
Diffstat (limited to 'lib/sqlalchemy/sql/dml.py')
-rw-r--r--lib/sqlalchemy/sql/dml.py20
1 files changed, 18 insertions, 2 deletions
diff --git a/lib/sqlalchemy/sql/dml.py b/lib/sqlalchemy/sql/dml.py
index 171a2cc2c..2349bfd03 100644
--- a/lib/sqlalchemy/sql/dml.py
+++ b/lib/sqlalchemy/sql/dml.py
@@ -37,6 +37,18 @@ class DMLState(CompileState):
isdelete = False
isinsert = False
+ @classmethod
+ def _create_insert(cls, statement, compiler, **kw):
+ return DMLState(statement, compiler, isinsert=True, **kw)
+
+ @classmethod
+ def _create_update(cls, statement, compiler, **kw):
+ return DMLState(statement, compiler, isupdate=True, **kw)
+
+ @classmethod
+ def _create_delete(cls, statement, compiler, **kw):
+ return DMLState(statement, compiler, isdelete=True, **kw)
+
def __init__(
self,
statement,
@@ -181,8 +193,6 @@ class UpdateBase(
_hints = util.immutabledict()
named_with_column = False
- _compile_state_cls = DMLState
-
@classmethod
def _constructor_20_deprecations(cls, fn_name, clsname, names):
@@ -717,6 +727,8 @@ class Insert(ValuesBase):
_supports_multi_parameters = True
+ _compile_state_factory = DMLState._create_insert
+
select = None
include_insert_from_select_defaults = False
@@ -915,6 +927,8 @@ class Update(DMLWhereBase, ValuesBase):
__visit_name__ = "update"
+ _compile_state_factory = DMLState._create_update
+
_traverse_internals = (
[
("table", InternalTraversal.dp_clauseelement),
@@ -1153,6 +1167,8 @@ class Delete(DMLWhereBase, UpdateBase):
__visit_name__ = "delete"
+ _compile_state_factory = DMLState._create_delete
+
_traverse_internals = (
[
("table", InternalTraversal.dp_clauseelement),