Skip to content

Instantly share code, notes, and snippets.

@pleshevskiy
Created August 17, 2018 07:46
Show Gist options
  • Save pleshevskiy/ae39b9f556ec4c43cff6991d3294eef6 to your computer and use it in GitHub Desktop.
Save pleshevskiy/ae39b9f556ec4c43cff6991d3294eef6 to your computer and use it in GitHub Desktop.
Alembic migrate enum
from alembic import op
from sqlalchemy.dialects.postgresql import ENUM
from sqlalchemy import Column
def upgrade():
enum = ENUM("one", "two", "three", name="pgenum", create_type=False)
enum.create(op.get_bind(), checkfirst=False)
op.create_table("sometable",
Column("data", enum)
)
def downgrade():
op.drop_table("sometable")
ENUM(name="pgenum").drop(op.get_bind(), checkfirst=False)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment