diff options
| -rw-r--r-- | doc/build/changelog/unreleased_12/pg_partition.rst | 7 | ||||
| -rw-r--r-- | lib/sqlalchemy/dialects/postgresql/base.py | 13 | ||||
| -rw-r--r-- | test/dialect/postgresql/test_compiler.py | 20 |
3 files changed, 39 insertions, 1 deletions
diff --git a/doc/build/changelog/unreleased_12/pg_partition.rst b/doc/build/changelog/unreleased_12/pg_partition.rst new file mode 100644 index 000000000..57493d327 --- /dev/null +++ b/doc/build/changelog/unreleased_12/pg_partition.rst @@ -0,0 +1,7 @@ +.. change:: + :tags: feature, postgresql + :versions: 1.3.0b1 + + Added support for "PARTITION BY" in Postgresql table definitions, + using "postgresql_partition_by". Pull request courtesy + Vsevolod Solovyov. diff --git a/lib/sqlalchemy/dialects/postgresql/base.py b/lib/sqlalchemy/dialects/postgresql/base.py index 5ae27a996..1dffe8db9 100644 --- a/lib/sqlalchemy/dialects/postgresql/base.py +++ b/lib/sqlalchemy/dialects/postgresql/base.py @@ -780,7 +780,14 @@ dialect in conjunction with the :class:`.Table` construct: Table("some_table", metadata, ..., postgresql_inherits=("t1", "t2", ...)) -.. versionadded:: 1.0.0 + .. versionadded:: 1.0.0 + +* ``PARTITION BY``:: + + Table("some_table", metadata, ..., + postgresql_partition_by='LIST (part_column)') + + .. versionadded:: 1.2.6 .. seealso:: @@ -1829,6 +1836,9 @@ class PGDDLCompiler(compiler.DDLCompiler): ', '.join(self.preparer.quote(name) for name in inherits) + ' )') + if pg_opts['partition_by']: + table_opts.append('\n PARTITION BY %s' % pg_opts['partition_by']) + if pg_opts['with_oids'] is True: table_opts.append('\n WITH OIDS') elif pg_opts['with_oids'] is False: @@ -2166,6 +2176,7 @@ class PGDialect(default.DefaultDialect): (schema.Table, { "ignore_search_path": False, "tablespace": None, + "partition_by": None, "with_oids": None, "on_commit": None, "inherits": None diff --git a/test/dialect/postgresql/test_compiler.py b/test/dialect/postgresql/test_compiler.py index db142a657..66764fcc2 100644 --- a/test/dialect/postgresql/test_compiler.py +++ b/test/dialect/postgresql/test_compiler.py @@ -233,6 +233,26 @@ class CompileTest(fixtures.TestBase, AssertsCompiledSQL): 'CREATE TABLE atable (id INTEGER) INHERITS ' '( "Quote Me", "quote Me Too" )') + def test_create_table_partition_by_list(self): + m = MetaData() + tbl = Table( + 'atable', m, Column("id", Integer), Column("part_column", Integer), + postgresql_partition_by='LIST (part_column)') + self.assert_compile( + schema.CreateTable(tbl), + 'CREATE TABLE atable (id INTEGER, part_column INTEGER) ' + 'PARTITION BY LIST (part_column)') + + def test_create_table_partition_by_range(self): + m = MetaData() + tbl = Table( + 'atable', m, Column("id", Integer), Column("part_column", Integer), + postgresql_partition_by='RANGE (part_column)') + self.assert_compile( + schema.CreateTable(tbl), + 'CREATE TABLE atable (id INTEGER, part_column INTEGER) ' + 'PARTITION BY RANGE (part_column)') + def test_create_table_with_oids(self): m = MetaData() tbl = Table( |