diff options
author | Joshua Colp <jcolp@digium.com> | 2017-09-06 11:23:19 -0500 |
---|---|---|
committer | Gerrit Code Review <gerrit2@gerrit.digium.api> | 2017-09-06 11:23:19 -0500 |
commit | 9a366d24242a0fc1e7d04b8859ac5773193aa932 (patch) | |
tree | fe524fb0f3d5eba29ac0cd3e5b303f21196387a3 /contrib | |
parent | f6d63a2539f4afc69725b2fdffa074a996444426 (diff) | |
parent | 15ddc9acb3160b24df8a0b56089dcb255aa5ef7d (diff) |
Merge "alembic: Fix enum creation for dtls_fingerprint" into 13
Diffstat (limited to 'contrib')
-rw-r--r-- | contrib/ast-db-manage/config/versions/b83645976fdd_add_dtls_fingerprint_to_ps_endpoints.py | 17 |
1 files changed, 16 insertions, 1 deletions
diff --git a/contrib/ast-db-manage/config/versions/b83645976fdd_add_dtls_fingerprint_to_ps_endpoints.py b/contrib/ast-db-manage/config/versions/b83645976fdd_add_dtls_fingerprint_to_ps_endpoints.py index 0efefa2fd..3d4f74de5 100644 --- a/contrib/ast-db-manage/config/versions/b83645976fdd_add_dtls_fingerprint_to_ps_endpoints.py +++ b/contrib/ast-db-manage/config/versions/b83645976fdd_add_dtls_fingerprint_to_ps_endpoints.py @@ -14,10 +14,25 @@ from alembic import op import sqlalchemy as sa from sqlalchemy.dialects.postgresql import ENUM +SHA_HASH_NAME = 'sha_hash_values' SHA_HASH_VALUES = ['SHA-1', 'SHA-256'] def upgrade(): - op.add_column('ps_endpoints', sa.Column('dtls_fingerprint', sa.Enum(*SHA_HASH_VALUES, name='sha_hash_values'))) + context = op.get_context() + + if context.bind.dialect.name == 'postgresql': + enum = ENUM(*SHA_HASH_VALUES, name=SHA_HASH_NAME) + enum.create(op.get_bind(), checkfirst=False) + + op.add_column('ps_endpoints', + sa.Column('dtls_fingerprint', ENUM(*SHA_HASH_VALUES, + name=SHA_HASH_NAME, create_type=False))) def downgrade(): + context = op.get_context() + op.drop_column('ps_endpoints', 'dtls_fingerprint') + + if context.bind.dialect.name == 'postgresql': + enum = ENUM(*SHA_HASH_VALUES, name=SHA_HASH_NAME) + enum.drop(op.get_bind(), checkfirst=False) |