diff options
| author | Dimitris Theodorou <dimitris.theodorou@gmail.com> | 2015-01-12 02:40:50 +0100 |
|---|---|---|
| committer | Dimitris Theodorou <dimitris.theodorou@gmail.com> | 2015-01-12 02:40:50 +0100 |
| commit | f4b7b02e31e6b49195c21da7221bcbda0bad02b9 (patch) | |
| tree | 33a7c94d79210a5223c517f5aa2789d58929f853 | |
| parent | b8a8cdd1ff47b5774662f4c61fe49382b967de02 (diff) | |
| download | sqlalchemy-f4b7b02e31e6b49195c21da7221bcbda0bad02b9.tar.gz | |
Add native_enum flag to Enum's repr() result
Needed for alembic autogenerate rendering.
| -rw-r--r-- | lib/sqlalchemy/sql/sqltypes.py | 1 | ||||
| -rw-r--r-- | test/sql/test_types.py | 4 |
2 files changed, 3 insertions, 2 deletions
diff --git a/lib/sqlalchemy/sql/sqltypes.py b/lib/sqlalchemy/sql/sqltypes.py index 9b0d26601..bd1914da3 100644 --- a/lib/sqlalchemy/sql/sqltypes.py +++ b/lib/sqlalchemy/sql/sqltypes.py @@ -1146,6 +1146,7 @@ class Enum(String, SchemaType): def __repr__(self): return util.generic_repr(self, + additional_kw=[('native_enum', True)], to_inspect=[Enum, SchemaType], ) diff --git a/test/sql/test_types.py b/test/sql/test_types.py index 26dc6c842..0212499c4 100644 --- a/test/sql/test_types.py +++ b/test/sql/test_types.py @@ -1157,8 +1157,8 @@ class EnumTest(AssertsCompiledSQL, fixtures.TestBase): def test_repr(self): e = Enum( "x", "y", name="somename", convert_unicode=True, quote=True, - inherit_schema=True) - eq_(repr(e), "Enum('x', 'y', name='somename', inherit_schema=True)") + inherit_schema=True, native_enum=False) + eq_(repr(e), "Enum('x', 'y', name='somename', inherit_schema=True, native_enum=False)") binary_table = MyPickleType = metadata = None |