summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/dialects/postgresql/ext.py
diff options
context:
space:
mode:
authorMike Bayer <mike_mp@zzzcomputing.com>2020-11-08 11:14:22 -0500
committerMike Bayer <mike_mp@zzzcomputing.com>2020-11-08 12:45:34 -0500
commitc6d869f814d2e8ffe03e519c59cf35f7a4927e1c (patch)
tree4017104ab8f89f451b60c788262069c91c0451aa /lib/sqlalchemy/dialects/postgresql/ext.py
parent89e748d7ad4426077313aaec916b41f999ae5a34 (diff)
downloadsqlalchemy-c6d869f814d2e8ffe03e519c59cf35f7a4927e1c.tar.gz
Allow dialect-specific stringification
Dialect-specific constructs such as :meth:`_postgresql.Insert.on_conflict_do_update` can now stringify in-place without the need to specify an explicit dialect object. The constructs, when called upon for ``str()``, ``print()``, etc. now have internal direction to call upon their appropriate dialect rather than the "default"dialect which doesn't know how to stringify these. The approach is also adapted to generic schema-level create/drop such as :class:`_schema.AddConstraint`, which will adapt its stringify dialect to one indicated by the element within it, such as the :class:`_postgresql.ExcludeConstraint` object. mostly towards being able to provide doctest-style examples for "on conflict" constructs using print statements. Change-Id: I4b855516fe6dee2df77744c1bb21a373d7fbab93
Diffstat (limited to 'lib/sqlalchemy/dialects/postgresql/ext.py')
-rw-r--r--lib/sqlalchemy/dialects/postgresql/ext.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/lib/sqlalchemy/dialects/postgresql/ext.py b/lib/sqlalchemy/dialects/postgresql/ext.py
index c139fe94f..78d9a96b6 100644
--- a/lib/sqlalchemy/dialects/postgresql/ext.py
+++ b/lib/sqlalchemy/dialects/postgresql/ext.py
@@ -52,6 +52,8 @@ class aggregate_order_by(expression.ColumnElement):
__visit_name__ = "aggregate_order_by"
+ stringify_dialect = "postgresql"
+
def __init__(self, target, *order_by):
self.target = coercions.expect(roles.ExpressionElementRole, target)
@@ -96,6 +98,8 @@ class ExcludeConstraint(ColumnCollectionConstraint):
where = None
+ create_drop_stringify_dialect = "postgresql"
+
@elements._document_text_coercion(
"where",
":class:`.ExcludeConstraint`",