From 0cbb17ce8f5c6afa7ac63e3b468ae40388d98b72 Mon Sep 17 00:00:00 2001
From: George Joseph <gjoseph@digium.com>
Date: Wed, 6 Sep 2017 06:54:00 -0600
Subject: [PATCH] alembic: Fix enum creation for dtls_fingerprint
Change-Id: Ic061c5066a146616a68376881c7e4cf6d6e7e7db
---
...6fdd_add_dtls_fingerprint_to_ps_endpoints.py | 17 ++++++++++++++++-
1 file changed, 16 insertions(+), 1 deletion(-)
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 0efefa2fda..3d4f74de51 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)
--
GitLab