Merge "alembic: Fix use_callerid_contact option add script." into 16

This commit is contained in:
George Joseph 2018-10-31 13:58:21 -05:00 committed by Gerrit Code Review
commit 021a425011
1 changed files with 10 additions and 5 deletions

View File

@ -14,21 +14,26 @@ from alembic import op
import sqlalchemy as sa
from sqlalchemy.dialects.postgresql import ENUM
YESNO_NAME = 'yesno_values'
YESNO_VALUES = ['yes', 'no']
AST_BOOL_NAME = 'ast_bool_values'
# We'll just ignore the n/y and f/t abbreviations as Asterisk does not write
# those aliases.
AST_BOOL_VALUES = [ '0', '1',
'off', 'on',
'false', 'true',
'no', 'yes' ]
def upgrade():
############################# Enums ##############################
# yesno_values have already been created, so use postgres enum object
# ast_bool_values has already been created, so use postgres enum object
# type to get around "already created" issue - works okay with mysql
yesno_values = ENUM(*YESNO_VALUES, name=YESNO_NAME, create_type=False)
ast_bool_values = ENUM(*AST_BOOL_VALUES, name=AST_BOOL_NAME, create_type=False)
op.add_column('ps_globals', sa.Column('use_callerid_contact', yesno_values))
def downgrade():
if op.get_context().bind.dialect.name == 'mssql':
op.drop_constraint('ck_ps_globals_use_callerid_contact_yesno_values','ps_globals')
op.drop_constraint('ck_ps_globals_use_callerid_contact_ast_bool_values', 'ps_globals')
op.drop_column('ps_globals', 'use_callerid_contact')