20def upgrade():
   21    context = op.get_context()
   22 
   23    if context.bind.dialect.name == 'postgresql':
   24        security_negotiation_values = ENUM(*SECURITY_NEGOTIATION_VALUES, name=SECURITY_NEGOTIATION_NAME)
   25        security_negotiation_values.create(op.get_bind(), checkfirst=False)
   26 
   27    op.add_column('ps_endpoints', sa.Column('security_negotiation',
   28        ENUM(*SECURITY_NEGOTIATION_VALUES, name=SECURITY_NEGOTIATION_NAME, create_type=False)))
   29    op.add_column('ps_endpoints', sa.Column('security_mechanisms', sa.String(512)))
   30 
   31    op.add_column('ps_registrations', sa.Column('security_negotiation',
   32        ENUM(*SECURITY_NEGOTIATION_VALUES, name=SECURITY_NEGOTIATION_NAME, create_type=False)))
   33    op.add_column('ps_registrations', sa.Column('security_mechanisms', sa.String(512)))
   34