pjsip_add_use_callerid_contact: fixed alembic script
[asterisk/asterisk.git] / contrib / ast-db-manage / config / versions / 2bb1a85135ad_pjsip_add_use_callerid_contact.py
1 """pjsip add use_callerid_contact
2
3 Revision ID: 2bb1a85135ad
4 Revises: 7f85dd44c775
5 Create Date: 2018-10-18 15:13:40.462354
6
7 """
8
9 # revision identifiers, used by Alembic.
10 revision = '2bb1a85135ad'
11 down_revision = '465f47f880be'
12
13 from alembic import op
14 import sqlalchemy as sa
15 from sqlalchemy.dialects.postgresql import ENUM
16
17 AST_BOOL_NAME = 'ast_bool_values'
18 # We'll just ignore the n/y and f/t abbreviations as Asterisk does not write
19 # those aliases.
20 AST_BOOL_VALUES = [ '0', '1',
21                     'off', 'on',
22                     'false', 'true',
23                     'no', 'yes' ]
24
25
26 def upgrade():
27     ############################# Enums ##############################
28
29     # ast_bool_values has already been created, so use postgres enum object
30     # type to get around "already created" issue - works okay with mysql
31     ast_bool_values = ENUM(*AST_BOOL_VALUES, name=AST_BOOL_NAME, create_type=False)
32
33     op.add_column('ps_globals', sa.Column('use_callerid_contact', ast_bool_values))
34
35
36 def downgrade():
37     if op.get_context().bind.dialect.name == 'mssql':
38         op.drop_constraint('ck_ps_globals_use_callerid_contact_ast_bool_values', 'ps_globals')
39     op.drop_column('ps_globals', 'use_callerid_contact')