1 """Add missing columns to system and registration
3 Revision ID: dbc44d5a908
5 Create Date: 2016-02-03 13:15:15.083043
10 revision =
'dbc44d5a908'
11 down_revision =
'423f34ad36e2'
13 from alembic
import op
14 import sqlalchemy
as sa
15 from sqlalchemy.dialects.postgresql
import ENUM
17 YESNO_NAME =
'yesno_values'
18 YESNO_VALUES = [
'yes',
'no']
25 yesno_values = ENUM(*YESNO_VALUES, name=YESNO_NAME, create_type=
False)
27 op.add_column(
'ps_systems', sa.Column(
'disable_tcp_switch', yesno_values))
28 op.add_column(
'ps_registrations', sa.Column(
'line', yesno_values))
29 op.add_column(
'ps_registrations', sa.Column(
'endpoint', sa.String(40)))
32 if op.get_context().bind.dialect.name ==
'mssql':
33 op.drop_constraint(
'ck_ps_systems_disable_tcp_switch_yesno_values',
'ps_systems')
34 op.drop_constraint(
'ck_ps_registrations_line_yesno_values',
'ps_registrations')
35 op.drop_column(
'ps_systems',
'disable_tcp_switch')
36 op.drop_column(
'ps_registrations',
'line')
37 op.drop_column(
'ps_registrations',
'endpoint')