--- /dev/null
+"""more permission boolean columns
+
+Revision ID: 74dc751dfe8e
+Revises: bd335bae5d33
+Create Date: 2024-02-27 15:31:09.458313
+
+"""
+
+# revision identifiers, used by Alembic.
+revision = '74dc751dfe8e'
+down_revision = 'bd335bae5d33'
+
+import itertools
+import operator
+
+from alembic import op
+import sqlalchemy as sa
+from sqlalchemy import case, cast, or_, text
+from sqlalchemy.dialects.postgresql import ENUM
+from sqlalchemy.sql import table, column
+
+COLUMNS = [ ('ps_aors', 'authenticate_qualify'),
+ ('ps_aors', 'remove_existing'),
+ ('ps_aors', 'remove_unavailable'),
+ ('ps_aors', 'support_path'),
+ ('ps_asterisk_publications', 'device_state'),
+ ('ps_asterisk_publications', 'mailbox_state'),
+ ('ps_contacts', 'authenticate_qualify'),
+ ('ps_contacts', 'prune_on_boot'),
+ ('ps_endpoint_id_ips', 'srv_lookups'),
+ ('ps_endpoints', '100rel'),
+ ('ps_endpoints', 'accept_multiple_sdp_answers'),
+ ('ps_endpoints', 'aggregate_mwi'),
+ ('ps_endpoints', 'allow_overlap'),
+ ('ps_endpoints', 'allow_subscribe'),
+ ('ps_endpoints', 'allow_transfer'),
+ ('ps_endpoints', 'allow_unauthenticated_options'),
+ ('ps_endpoints', 'asymmetric_rtp_codec'),
+ ('ps_endpoints', 'bind_rtp_to_media_address'),
+ ('ps_endpoints', 'bundle'),
+ ('ps_endpoints', 'direct_media'),
+ ('ps_endpoints', 'disable_direct_media_on_nat'),
+ ('ps_endpoints', 'dtls_auto_generate_cert'),
+ ('ps_endpoints', 'fax_detect'),
+ ('ps_endpoints', 'follow_early_media_fork'),
+ ('ps_endpoints', 'force_avp'),
+ ('ps_endpoints', 'force_rport'),
+ ('ps_endpoints', 'g726_non_standard'),
+ ('ps_endpoints', 'ice_support'),
+ ('ps_endpoints', 'ignore_183_without_sdp'),
+ ('ps_endpoints', 'inband_progress'),
+ ('ps_endpoints', 'media_encryption_optimistic'),
+ ('ps_endpoints', 'media_use_received_transport'),
+ ('ps_endpoints', 'moh_passthrough'),
+ ('ps_endpoints', 'mwi_subscribe_replaces_unsolicited'),
+ ('ps_endpoints', 'notify_early_inuse_ringing'),
+ ('ps_endpoints', 'one_touch_recording'),
+ ('ps_endpoints', 'preferred_codec_only'),
+ ('ps_endpoints', 'refer_blind_progress'),
+ ('ps_endpoints', 'rewrite_contact'),
+ ('ps_endpoints', 'rpid_immediate'),
+ ('ps_endpoints', 'rtcp_mux'),
+ ('ps_endpoints', 'rtp_ipv6'),
+ ('ps_endpoints', 'rtp_symmetric'),
+ ('ps_endpoints', 'send_aoc'),
+ ('ps_endpoints', 'send_connected_line'),
+ ('ps_endpoints', 'send_diversion'),
+ ('ps_endpoints', 'send_history_info'),
+ ('ps_endpoints', 'send_pai'),
+ ('ps_endpoints', 'send_rpid'),
+ ('ps_endpoints', 'srtp_tag_32'),
+ ('ps_endpoints', 'stir_shaken'),
+ ('ps_endpoints', 'suppress_q850_reason_headers'),
+ ('ps_endpoints', 't38_bind_udptl_to_media_address'),
+ ('ps_endpoints', 't38_udptl'),
+ ('ps_endpoints', 't38_udptl_ipv6'),
+ ('ps_endpoints', 't38_udptl_nat'),
+ ('ps_endpoints', 'timers'),
+ ('ps_endpoints', 'trust_connected_line'),
+ ('ps_endpoints', 'trust_id_inbound'),
+ ('ps_endpoints', 'trust_id_outbound'),
+ ('ps_endpoints', 'use_avpf'),
+ ('ps_endpoints', 'use_ptime'),
+ ('ps_endpoints', 'user_eq_phone'),
+ ('ps_endpoints', 'webrtc'),
+ ('ps_globals', 'all_codecs_on_empty_reinvite'),
+ ('ps_globals', 'allow_sending_180_after_183'),
+ ('ps_globals', 'disable_multi_domain'),
+ ('ps_globals', 'ignore_uri_user_options'),
+ ('ps_globals', 'mwi_disable_initial_unsolicited'),
+ ('ps_globals', 'norefersub'),
+ ('ps_globals', 'send_contact_status_on_update_registration'),
+ ('ps_globals', 'use_callerid_contact'),
+ ('ps_outbound_publishes', 'multi_user'),
+ ('ps_registrations', 'auth_rejection_permanent'),
+ ('ps_registrations', 'line'),
+ ('ps_registrations', 'support_path'),
+ ('ps_resource_list', 'full_state'),
+ ('ps_resource_list', 'resource_display_name'),
+ ('ps_subscription_persistence', 'prune_on_boot'),
+ ('ps_systems', 'accept_multiple_sdp_answers'),
+ ('ps_systems', 'compact_headers'),
+ ('ps_systems', 'disable_rport'),
+ ('ps_systems', 'disable_tcp_switch'),
+ ('ps_systems', 'follow_early_media_fork'),
+ ('ps_transports', 'allow_reload'),
+ ('ps_transports', 'allow_wildcard_certs'),
+ ('ps_transports', 'require_client_cert'),
+ ('ps_transports', 'symmetric_transport'),
+ ('ps_transports', 'verify_client'),
+ ('ps_transports', 'verify_server') ]
+
+YESNO_NAME = 'yesno_values'
+YESNO_VALUES = ['yes', 'no']
+
+AST_BOOL_NAME = 'ast_bool_values'
+AST_BOOL_VALUES = [ '0', '1',
+ 'off', 'on',
+ 'false', 'true',
+ 'no', 'yes' ]
+
+yesno_values = ENUM(*YESNO_VALUES, name=YESNO_NAME, create_type=False)
+ast_bool_values = ENUM(*AST_BOOL_VALUES, name=AST_BOOL_NAME, create_type=False)
+
+def upgrade():
+ for table_name, column_list in itertools.groupby(COLUMNS, operator.itemgetter(0)):
+ with op.batch_alter_table(table_name) as batch_op:
+ for _, column_name in column_list:
+ batch_op.alter_column(column_name,
+ type_=ast_bool_values,
+ existing_type=yesno_values,
+ postgresql_using='"{}"::text::{}'.format(column_name, AST_BOOL_NAME))
+
+def downgrade():
+ for table_name, column_list in itertools.groupby(COLUMNS, operator.itemgetter(0)):
+ subject = table(table_name)
+ values_exprs = {}
+ for _, column_name in column_list:
+ subject.append_column(column(column_name))
+ values_exprs[column_name] = cast(
+ case((or_(subject.c[column_name] == text("'yes'"),
+ subject.c[column_name] == text("'1'"),
+ subject.c[column_name] == text("'on'"),
+ subject.c[column_name] == text("'true'")), text("'yes'")),
+ else_=text("'no'")),
+ ast_bool_values)
+
+ op.execute(
+ subject.update().values(values_exprs)
+ )
+
+ for table_name, column_list in itertools.groupby(COLUMNS, operator.itemgetter(0)):
+ with op.batch_alter_table(table_name) as batch_op:
+ for _, column_name in column_list:
+ batch_op.alter_column(column_name,
+ type_=yesno_values,
+ existing_type=ast_bool_values,
+ postgresql_using='"{}"::text::{}'.format(column_name, YESNO_NAME))