1 """Fix tos and cos field types
3 Revision ID: 4c573e7135bd
5 Create Date: 2014-03-05 12:16:56.618630
10 revision =
'4c573e7135bd'
11 down_revision =
'28887f25a46f'
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']
21 op.alter_column(
'ps_endpoints',
'tos_audio', type_=sa.String(10))
22 op.alter_column(
'ps_endpoints',
'tos_video', type_=sa.String(10))
23 op.drop_column(
'ps_endpoints',
'cos_audio')
24 op.drop_column(
'ps_endpoints',
'cos_video')
25 op.add_column(
'ps_endpoints', sa.Column(
'cos_audio', sa.Integer))
26 op.add_column(
'ps_endpoints', sa.Column(
'cos_video', sa.Integer))
28 op.alter_column(
'ps_transports',
'tos', type_=sa.String(10))
31 op.drop_column(
'ps_transports',
'cos', schema=
None, mssql_drop_check=
True)
32 op.add_column(
'ps_transports', sa.Column(
'cos', sa.Integer))
36 yesno_values = ENUM(*YESNO_VALUES, name=YESNO_NAME, create_type=
False)
39 op.drop_column(
'ps_endpoints',
'tos_audio')
40 op.drop_column(
'ps_endpoints',
'tos_video')
41 op.add_column(
'ps_endpoints', sa.Column(
'tos_audio', yesno_values))
42 op.add_column(
'ps_endpoints', sa.Column(
'tos_video', yesno_values))
43 op.drop_column(
'ps_endpoints',
'cos_audio')
44 op.drop_column(
'ps_endpoints',
'cos_video')
45 op.add_column(
'ps_endpoints', sa.Column(
'cos_audio', yesno_values))
46 op.add_column(
'ps_endpoints', sa.Column(
'cos_video', yesno_values))
48 if op.get_context().bind.dialect.name ==
'mssql':
49 op.drop_constraint(
'ck_ps_transports_tos_yesno_values',
'ps_transports')
50 op.drop_column(
'ps_transports',
'tos')
51 op.add_column(
'ps_transports', sa.Column(
'tos', yesno_values))
53 op.drop_column(
'ps_transports',
'cos')
54 op.add_column(
'ps_transports', sa.Column(
'cos', yesno_values))