Asterisk - The Open Source Telephony Project  21.4.1
5a2247c957d2_add_aoc_option.py
1 """add aoc option
2 
3 Revision ID: 5a2247c957d2
4 Revises: ccf795ee535f
5 Create Date: 2022-10-30 12:47:56.173511
6 
7 """
8 
9 # revision identifiers, used by Alembic.
10 revision = '5a2247c957d2'
11 down_revision = 'ccf795ee535f'
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 def upgrade():
26  ############################# Enums ##############################
27 
28  # ast_bool_values has already been created, so use postgres enum object
29  # type to get around "already created" issue - works okay with mysql
30  ast_bool_values = ENUM(*AST_BOOL_VALUES, name=AST_BOOL_NAME, create_type=False)
31 
32  op.add_column('ps_endpoints', sa.Column('send_aoc', ast_bool_values))
33 
34 def downgrade():
35  if op.get_context().bind.dialect.name == 'mssql':
36  op.drop_constraint('ck_ps_endpoints_send_aoc_ast_bool_values', 'ps_endpoints')
37  op.drop_column('ps_endpoints', 'send_aoc')
38  pass