From 3ebd1b30eb392ff62f08f0755e79a228d61ba7d2 Mon Sep 17 00:00:00 2001 From: Sergey Skopin Date: Tue, 31 May 2016 10:02:08 -0400 Subject: Add 'FOR NO KEY UPDATE' / 'FOR KEY SHARE' support for Postgresql Adds ``key_share=True`` for with_for_update(). Co-authored-by: Mike Bayer Change-Id: I74e0c3fcbc023e1dc98a1fa0c7db67b4c3693a31 Pull-request: https://github.com/zzzeek/sqlalchemy/pull/279 --- lib/sqlalchemy/dialects/postgresql/base.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) (limited to 'lib/sqlalchemy/dialects') diff --git a/lib/sqlalchemy/dialects/postgresql/base.py b/lib/sqlalchemy/dialects/postgresql/base.py index 2356458b9..d613aac92 100644 --- a/lib/sqlalchemy/dialects/postgresql/base.py +++ b/lib/sqlalchemy/dialects/postgresql/base.py @@ -1170,7 +1170,12 @@ class PGCompiler(compiler.SQLCompiler): def for_update_clause(self, select, **kw): if select._for_update_arg.read: - tmp = " FOR SHARE" + if select._for_update_arg.key_share: + tmp = " FOR KEY SHARE" + else: + tmp = " FOR SHARE" + elif select._for_update_arg.key_share: + tmp = " FOR NO KEY UPDATE" else: tmp = " FOR UPDATE" -- cgit v1.2.1