From 71c45937f9adbb64482fffcda75f8fe4d063e027 Mon Sep 17 00:00:00 2001
From: Mario Lassnig <mario@lassnig.net>
Date: Tue, 12 Nov 2013 23:08:51 +0100
Subject: add psql FOR UPDATE OF functionality

---
 lib/sqlalchemy/sql/compiler.py | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'lib/sqlalchemy/sql/compiler.py')

diff --git a/lib/sqlalchemy/sql/compiler.py b/lib/sqlalchemy/sql/compiler.py
index 4f3dbba36..51ec0d9eb 100644
--- a/lib/sqlalchemy/sql/compiler.py
+++ b/lib/sqlalchemy/sql/compiler.py
@@ -1571,6 +1571,8 @@ class SQLCompiler(Compiled):
 
     def for_update_clause(self, select):
         if select.for_update:
+            if select.for_update_of is not None:
+                return " FOR UPDATE OF " + select.for_update_of
             return " FOR UPDATE"
         else:
             return ""
-- 
cgit v1.2.1