alembic: Update list of TLS methods available on ps_transports.

Related to #221 and #222.

Also adds `*.ini` to the `.gitignore` file in ast-db-manage for
convenience.
This commit is contained in:
Sean Bright 2023-11-14 17:49:04 -05:00
parent 9211fb5e97
commit c7838a352a
2 changed files with 59 additions and 0 deletions

1
contrib/ast-db-manage/.gitignore vendored Normal file
View File

@ -0,0 +1 @@
*.ini

View File

@ -0,0 +1,58 @@
"""update pjsip tls method list
Revision ID: 37a5332640e2
Revises: dac2b4c328b8
Create Date: 2023-11-14 18:02:18.857452
"""
# revision identifiers, used by Alembic.
revision = '37a5332640e2'
down_revision = 'dac2b4c328b8'
from alembic import op
from sqlalchemy.dialects.postgresql import ENUM
import sqlalchemy as sa
PJSIP_TRANSPORT_METHOD_OLD_NAME = 'pjsip_transport_method_values'
PJSIP_TRANSPORT_METHOD_NEW_NAME = 'pjsip_transport_method_values_v2'
PJSIP_TRANSPORT_METHOD_OLD_VALUES = ['default', 'unspecified', 'tlsv1', 'sslv2',
'sslv3', 'sslv23']
PJSIP_TRANSPORT_METHOD_NEW_VALUES = ['default', 'unspecified',
'tlsv1', 'tlsv1_1', 'tlsv1_2', 'tlsv1_3',
'sslv2', 'sslv23', 'sslv3']
PJSIP_TRANSPORT_METHOD_OLD_TYPE = sa.Enum(*PJSIP_TRANSPORT_METHOD_OLD_VALUES,
name=PJSIP_TRANSPORT_METHOD_OLD_NAME)
PJSIP_TRANSPORT_METHOD_NEW_TYPE = sa.Enum(*PJSIP_TRANSPORT_METHOD_NEW_VALUES,
name=PJSIP_TRANSPORT_METHOD_NEW_NAME)
def upgrade():
if op.get_context().bind.dialect.name == 'postgresql':
enum = PJSIP_TRANSPORT_METHOD_NEW_TYPE
enum.create(op.get_bind(), checkfirst=False)
op.alter_column('ps_transports', 'method',
type_=PJSIP_TRANSPORT_METHOD_NEW_TYPE,
existing_type=PJSIP_TRANSPORT_METHOD_OLD_TYPE,
postgresql_using='method::text::' + PJSIP_TRANSPORT_METHOD_NEW_NAME)
if op.get_context().bind.dialect.name == 'postgresql':
ENUM(name=PJSIP_TRANSPORT_METHOD_OLD_NAME).drop(op.get_bind(), checkfirst=False)
def downgrade():
# First we need to ensure that columns are not using the enum values
# that are going away.
op.execute("UPDATE ps_transports SET method = 'tlsv1' WHERE method IN ('tlsv1_1', 'tlsv1_2', 'tlsv1_3')")
if op.get_context().bind.dialect.name == 'postgresql':
enum = PJSIP_TRANSPORT_METHOD_OLD_TYPE
enum.create(op.get_bind(), checkfirst=False)
op.alter_column('ps_transports', 'method',
type_=PJSIP_TRANSPORT_METHOD_OLD_TYPE,
existing_type=PJSIP_TRANSPORT_METHOD_NEW_TYPE,
postgresql_using='method::text::' + PJSIP_TRANSPORT_METHOD_OLD_NAME)
if op.get_context().bind.dialect.name == 'postgresql':
ENUM(name=PJSIP_TRANSPORT_METHOD_NEW_NAME).drop(op.get_bind(), checkfirst=False)