Merge "alembic: Fix use_callerid_contact option add script."
This commit is contained in:
commit
5ea16f0cb5
|
@ -14,21 +14,26 @@ from alembic import op
|
||||||
import sqlalchemy as sa
|
import sqlalchemy as sa
|
||||||
from sqlalchemy.dialects.postgresql import ENUM
|
from sqlalchemy.dialects.postgresql import ENUM
|
||||||
|
|
||||||
YESNO_NAME = 'yesno_values'
|
AST_BOOL_NAME = 'ast_bool_values'
|
||||||
YESNO_VALUES = ['yes', 'no']
|
# 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():
|
def upgrade():
|
||||||
############################# Enums ##############################
|
############################# 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
|
# 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))
|
op.add_column('ps_globals', sa.Column('use_callerid_contact', yesno_values))
|
||||||
|
|
||||||
|
|
||||||
def downgrade():
|
def downgrade():
|
||||||
if op.get_context().bind.dialect.name == 'mssql':
|
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')
|
op.drop_column('ps_globals', 'use_callerid_contact')
|
||||||
|
|
Loading…
Reference in New Issue