diff options
Diffstat (limited to 'lib/sqlalchemy/dialects/postgresql/base.py')
-rw-r--r-- | lib/sqlalchemy/dialects/postgresql/base.py | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/lib/sqlalchemy/dialects/postgresql/base.py b/lib/sqlalchemy/dialects/postgresql/base.py index 4d302dabe..3781a7ba2 100644 --- a/lib/sqlalchemy/dialects/postgresql/base.py +++ b/lib/sqlalchemy/dialects/postgresql/base.py @@ -929,6 +929,7 @@ from ...sql import compiler from ...sql import elements from ...sql import expression from ...sql import sqltypes +from ...sql import util as sql_util from ...types import BIGINT from ...types import BOOLEAN from ...types import CHAR @@ -1681,10 +1682,11 @@ class PGCompiler(compiler.SQLCompiler): tmp = " FOR UPDATE" if select._for_update_arg.of: - tables = util.OrderedSet( - c.table if isinstance(c, expression.ColumnClause) else c - for c in select._for_update_arg.of - ) + + tables = util.OrderedSet() + for c in select._for_update_arg.of: + tables.update(sql_util.surface_selectables_only(c)) + tmp += " OF " + ", ".join( self.process(table, ashint=True, use_schema=False, **kw) for table in tables |