Add TRUNCATE to postgres autocommit regexp

Extends AUTOCOMMIT_REGEXP for the postgres dialect to include `TRUNCATE`.

Change-Id: I315e03674b89bb89aae669b8655481e4d890491e
Pull-request: https://github.com/zzzeek/sqlalchemy/pull/407
(cherry picked from commit 756d5782870029f2d97b1aa171abd61dbf4cbcb4)

diff --git a/doc/build/changelog/unreleased_11/pg_truncate.rst b/doc/build/changelog/unreleased_11/pg_truncate.rst
new file mode 100644 (file)
index 0000000..4720c28
--- /dev/null
+++ b/doc/build/changelog/unreleased_11/pg_truncate.rst
@@ -0,0 +1,6 @@
+.. change::
+   :tags: bug, postgresql
+
+    Added "TRUNCATE" to the list of keywords accepted by the
+    Postgresql dialect as an "autocommit"-triggering keyword.
+    Pull request courtesy Jacob Hayes.

diff --git a/lib/sqlalchemy/dialects/postgresql/base.py b/lib/sqlalchemy/dialects/postgresql/base.py
index e5f164d57317b613417b5aaeaa0429346f49014f..627a62f9afe56ae6c922e12e8649160c885818ef 100644 (file)
--- a/lib/sqlalchemy/dialects/postgresql/base.py
+++ b/lib/sqlalchemy/dialects/postgresql/base.py
@@ -925,7 +925,7 @@ from sqlalchemy.types import INTEGER, BIGINT, SMALLINT, VARCHAR, \
 
 AUTOCOMMIT_REGEXP = re.compile(
     r'\s*(?:UPDATE|INSERT|CREATE|DELETE|DROP|ALTER|GRANT|REVOKE|'
-    'IMPORT FOREIGN SCHEMA|REFRESH MATERIALIZED VIEW)',
+    'IMPORT FOREIGN SCHEMA|REFRESH MATERIALIZED VIEW|TRUNCATE)',
     re.I | re.UNICODE)
 
 RESERVED_WORDS = set(

diff --git a/test/dialect/postgresql/test_dialect.py b/test/dialect/postgresql/test_dialect.py
index 5ab17ea8cd93ce3ba466f3ccee7a275c2efc044c..446765872e3da817cfcad5739a5eeb319e968252 100644 (file)
--- a/test/dialect/postgresql/test_dialect.py
+++ b/test/dialect/postgresql/test_dialect.py
@@ -322,3 +322,6 @@ class AutocommitTextTest(test_execute.AutocommitTextTest):
 
     def test_revoke(self):
         self._test_keyword("REVOKE USAGE ON SCHEMA fooschema FROM foorole")
+
+    def test_truncate(self):
+        self._test_keyword("TRUNCATE footable")