def downgrade():
op.execute(DropSequence(sa.Sequence('search_solr_seq')))
op.execute(DropSequence(sa.Sequence('sec_tls_ca_cert_seq')))
op.drop_column(model.HTTPSOAP.__tablename__, 'sec_tls_ca_cert_id')
op.drop_constraint('search_solr_name_cluster_id_key', 'search_solr')
op.drop_table(model.Solr.__tablename__)
op.drop_table(model.TLSCACert.__tablename__)
def downgrade():
op.execute(DropSequence(sa.Sequence('rbac_cli_rol_seq')))
op.execute(DropSequence(sa.Sequence('rbac_perm_seq')))
op.execute(DropSequence(sa.Sequence('rbac_role_perm_seq')))
op.drop_column(model.HTTPSOAP.__tablename__, 'has_rbac')
op.execute(DropSequence(sa.Sequence('rbac_role_seq')))
op.drop_table(model.RBACRolePermission.__tablename__)
op.drop_table(model.RBACClientRole.__tablename__)
op.drop_table(model.RBACRole.__tablename__)
op.drop_table(model.RBACPermission.__tablename__)
def downgrade():
### commands auto generated by Alembic - please adjust! ###
op.drop_table('tag_group')
op.drop_table('issue_version')
op.drop_table('image_group')
op.drop_table('tag')
op.drop_table('streetview')
op.drop_table('state')
op.drop_table('issue')
op.drop_table('image')
op.drop_table('description')
op.drop_table('coordinates')
op.execute(DropSequence(Sequence('tag_group_id_seq')))
op.execute(DropSequence(Sequence('image_group_id_seq')))
def test_create_drop_ddl(self):
self.assert_compile(
CreateSequence(Sequence('foo_seq')),
"CREATE SEQUENCE foo_seq",
)
self.assert_compile(
CreateSequence(Sequence('foo_seq', start=5)),
"CREATE SEQUENCE foo_seq START WITH 5",
)
self.assert_compile(
CreateSequence(Sequence('foo_seq', increment=2)),
"CREATE SEQUENCE foo_seq INCREMENT BY 2",
)
self.assert_compile(
CreateSequence(Sequence('foo_seq', increment=2, start=5)),
"CREATE SEQUENCE foo_seq INCREMENT BY 2 START WITH 5",
)
self.assert_compile(
DropSequence(Sequence('foo_seq')),
"DROP SEQUENCE foo_seq",
)
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.drop_index(op.f('ix_account_bcol_ids_account_id'),
table_name='account_bcol_ids')
op.drop_table('account_bcol_ids')
# Added manually
op.execute(DropSequence(Sequence('account_bcol_id_seq')))
def drop_dataset_table(dataset_data_table, sequence):
"""Drop dataset_<name>."""
try:
_db.session.execute(dataset_data_table.delete())
_db.session.execute(DropSequence(sequence))
_db.session.session.commit()
except BaseException as err:
logging.warning('Error while delete dataset table data')
def downgrade():
# Drop materialized views
op.execute(drop_all_dashboard_helper_views)
op.drop_constraint(
"data_source_in_measure_version_measure_version_id_fkey", "data_source_in_measure_version", type_="foreignkey"
)
op.drop_constraint("dimension_measure_version_id_fkey", "dimension", type_="foreignkey")
op.drop_constraint("upload_measure_version_id_fkey", "upload", type_="foreignkey")
op.drop_constraint("measure_version_parent_id_fkey", "measure_version", type_="foreignkey")
op.drop_constraint(
"data_source_in_measure_version_data_source_id_pkey", "data_source_in_measure_version", type_="primary"
)
op.drop_constraint("measure_version_id_pkey", "measure_version", type_="primary")
op.rename_table("data_source_in_measure_version", "data_source_in_page")
op.rename_table("measure_version", "page")
op.drop_column("dimension", "measure_version_id")
op.drop_column("upload", "measure_version_id")
op.drop_column("data_source_in_page", "measure_version_id")
op.drop_column("page", "parent_id")
op.drop_column("page", "id")
op.execute(DropSequence(Sequence("measure_version_id_seq")))
op.create_unique_constraint("uq_page_guid_version", "page", ["guid", "version"])
op.create_primary_key(op.f("page_guid_version_pkey"), "page", ["guid", "version"])
op.create_primary_key(
op.f("data_source_id_page_guid_version_pk"),
"data_source_in_page",
["data_source_id", "page_guid", "page_version"],
)
op.create_foreign_key(
op.f("data_source_in_page_page_guid_fkey"),
"data_source_in_page",
"page",
["page_guid", "page_version"],
["guid", "version"],
)
op.create_foreign_key(
op.f("dimension_page_id_version_fkey"), "dimension", "page", ["page_id", "page_version"], ["guid", "version"]
)
op.create_foreign_key(
op.f("upload_page_id_version_fkey"), "upload", "page", ["page_id", "page_version"], ["guid", "version"]
)
op.create_foreign_key(
op.f("page_parent_guid_version_fkey"), "page", "page", ["parent_guid", "parent_version"], ["guid", "version"]
)
# Create materialized views using old (pre-migration) spec
op.execute(OLD_latest_published_pages_view)
op.execute(OLD_ethnic_groups_by_dimension_view)
op.execute(OLD_categorisations_by_dimension)
op.execute(OLD_pages_by_geography_view)
def test_create_drop_ddl(self):
self.assert_compile(
CreateSequence(Sequence('foo_seq')),
"CREATE SEQUENCE foo_seq",
)
self.assert_compile(
CreateSequence(Sequence('foo_seq', start=5)),
"CREATE SEQUENCE foo_seq START WITH 5",
)
self.assert_compile(
CreateSequence(Sequence('foo_seq', increment=2)),
"CREATE SEQUENCE foo_seq INCREMENT BY 2",
)
self.assert_compile(
CreateSequence(Sequence('foo_seq', increment=2, start=5)),
"CREATE SEQUENCE foo_seq INCREMENT BY 2 START WITH 5",
)
self.assert_compile(
CreateSequence(Sequence(
'foo_seq', increment=2, start=0, minvalue=0)),
"CREATE SEQUENCE foo_seq INCREMENT BY 2 START WITH 0 MINVALUE 0",
)
self.assert_compile(
CreateSequence(Sequence(
'foo_seq', increment=2, start=1, maxvalue=5)),
"CREATE SEQUENCE foo_seq INCREMENT BY 2 START WITH 1 MAXVALUE 5",
)
self.assert_compile(
CreateSequence(Sequence(
'foo_seq', increment=2, start=1, nomaxvalue=True)),
"CREATE SEQUENCE foo_seq INCREMENT BY 2 START WITH 1 NO MAXVALUE",
)
self.assert_compile(
CreateSequence(Sequence(
'foo_seq', increment=2, start=0, nominvalue=True)),
"CREATE SEQUENCE foo_seq INCREMENT BY 2 START WITH 0 NO MINVALUE",
)
self.assert_compile(
CreateSequence(Sequence(
'foo_seq', start=1, maxvalue=10, cycle=True)),
"CREATE SEQUENCE foo_seq START WITH 1 MAXVALUE 10 CYCLE",
)
self.assert_compile(
DropSequence(Sequence('foo_seq')),
"DROP SEQUENCE foo_seq",
)
def drop_udl2_sequences():
'''
drop sequences according to configuration file
'''
try:
print("drop sequences")
with get_udl_connection() as conn:
for seq in generate_udl2_sequences():
conn.execute(DropSequence(seq))
except Exception as e:
print("Error occurs when tearing down sequence: " + e)
def downgrade():
### commands auto generated by Alembic - please adjust! ###
op.alter_column('users', 'userid', server_default=None)
op.execute(DropSequence(Sequence('users_userid_seq')))
op.alter_column(u'users',
'emailaddress',
existing_type=sa.TEXT(),
nullable=True)
op.alter_column(u'ao_users',
'emailaddress',
existing_type=sa.TEXT(),
nullable=False)
def downgrade():
op.create_unique_constraint(
'http_soap_url_path_connection_soap_action_cluster_id_key', model.HTTPSOAP.__tablename__, \
['url_path', 'connection', 'soap_action', 'cluster_id']
)
op.drop_table('django_openid_auth_association')
op.drop_table('django_openid_auth_nonce')
op.drop_table('django_openid_auth_useropenid')
op.drop_table(model.NotificationOpenStackSwift.__tablename__)
op.drop_table(model.Notification.__tablename__)
op.drop_table(model.ElasticSearch.__tablename__)
op.drop_table(model.OpenStackSecurity.__tablename__)
op.drop_column(model.OpenStackSwift.__tablename__, 'pool_size')
op.execute(DropSequence(sa.Sequence('conn_def_cassandra_seq')))
op.execute(
DropSequence(sa.Sequence('django_openid_auth_association_id_seq')))
op.execute(DropSequence(sa.Sequence('django_openid_auth_nonce_id_seq')))
op.execute(
DropSequence(sa.Sequence('django_openid_auth_useropenid_id_seq')))
op.execute(DropSequence(sa.Sequence('query_cassandra_seq')))
op.execute(DropSequence(sa.Sequence('search_es_seq')))
op.drop_constraint('conn_def_cassandra_name_cluster_id_key',
'conn_def_cassandra')
op.drop_constraint(
'http_soap_url_path_host_connection_soap_action_cluster_id_key',
model.HTTPSOAP.__tablename__)
op.drop_constraint('query_cassandra_name_cluster_id_key',
'query_cassandra')
op.drop_constraint('alembic_version_pkey', 'alembic_version')
def downgrade():
# Remove any non-MTG cards from the DB, and them drop game column
alembic.op.execute("""
DELETE FROM cards
WHERE game != 1
""")
alembic.op.drop_index("cards_name_idx")
alembic.op.drop_column('cards', 'game')
alembic.op.create_index("cards_name_idx", "cards", ["filteredname"], unique=True)
# Remove auto-increment sequence from cards.id
alembic.op.alter_column("cards", "id", server_default=None)
alembic.op.execute(DropSequence(Sequence('cards_id_seq')))
def downgrade():
for seqname in ('contact_pronouns_position', 'contact_name_position',
'contact_email_position',
):
op.execute(DropSequence(Sequence(seqname)))
# ### commands auto generated by Alembic - please adjust! ###
op.drop_table('contact_pronouns')
op.drop_table('contact_name')
op.drop_table('contact_email')
op.drop_table('roles_users')
op.drop_table('contact')
op.drop_index(op.f('ix_user_username'), table_name='user')
op.drop_table('user')
op.drop_table('role')
op.drop_table('pronouns')
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.drop_table("expense_parties")
op.drop_table("expenses")
op.drop_table("user_transactions")
op.drop_table("user_group_mapping")
op.drop_table("friends")
op.drop_table("users")
op.drop_table("merchant")
op.drop_table("groups")
op.execute(DropSequence(Sequence("expense_parties_id_seq")))
op.execute(DropSequence(Sequence("expenses_id_seq")))
op.execute(DropSequence(Sequence("user_transactions_id_seq")))
op.execute(DropSequence(Sequence("user_group_mapping_id_seq")))
op.execute(DropSequence(Sequence("friends_id_seq")))
op.execute(DropSequence(Sequence("users_id_seq")))
op.execute(DropSequence(Sequence("merchant_id_seq")))
op.execute(DropSequence(Sequence("groups_id_seq")))
def upgrade():
for tablename in tables_with_name_column:
# Create CHECK constraint on name
op.create_check_constraint(tablename + '_name_check', tablename,
"name <> ''")
# RENAME CONSTRAINT works in PostgreSQL >= 9.2
op.execute(
sa.DDL(
'ALTER TABLE comment RENAME CONSTRAINT ck_comment_state_valid TO comment_status_check;'
))
op.execute(
sa.DDL(
'ALTER TABLE proposal RENAME CONSTRAINT ck_proposal_state_valid TO proposal_status_check;'
))
op.execute(
sa.DDL(
'ALTER TABLE proposal_space RENAME CONSTRAINT ck_proposal_space_state_valid TO proposal_space_status_check;'
))
op.execute(
sa.DDL(
'ALTER TABLE rsvp RENAME CONSTRAINT ck_rsvp_state_valid TO rsvp_status_check;'
))
op.execute(
sa.DDL(
'ALTER TABLE contact_exchange RENAME CONSTRAINT contact_exchange_user_id_proposal_space_id_participant_id_key TO contact_exchange_pkey;'
))
op.execute(
sa.DDL(
'ALTER TABLE proposal_space RENAME CONSTRAINT proposal_space_proposal_space_id_fkey TO proposal_space_parent_space_id_fkey;'
))
op.alter_column('proposal_redirect',
'url_id',
existing_type=sa.INTEGER(),
server_default=None)
op.execute(DropSequence(Sequence('proposal_redirect_url_id_seq')))
op.alter_column(
'user',
'userinfo',
type_=JsonDict(),
existing_type=sa.TEXT(),
postgresql_using='userinfo::jsonb',
)
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.alter_column('search_requests',
'client_reference_id',
existing_type=sa.String(length=50),
type_=sa.VARCHAR(length=20),
existing_nullable=True)
op.alter_column('registrations',
'client_reference_id',
existing_type=sa.String(length=50),
type_=sa.VARCHAR(length=20),
existing_nullable=True)
op.drop_index(op.f('ix_user_extra_registrations_registration_number'),
table_name='user_extra_registrations')
op.drop_index(op.f('ix_user_extra_registrations_account_id'),
table_name='user_extra_registrations')
op.drop_table('user_extra_registrations')
op.execute(DropSequence(Sequence('user_extra_registration_seq')))
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
bind = op.get_bind()
session = Session(bind=bind)
from sqlalchemy.schema import Sequence, CreateSequence, DropSequence
op.execute(DropSequence(Sequence('role_id_seq')))
op.execute(CreateSequence(Sequence('role_id_seq')))
op.create_table('role',
sa.Column('id', sa.INTEGER(), server_default=sa.text("nextval('role_id_seq'::regclass)"),
nullable=False),
sa.Column('name', sa.VARCHAR(length=255), autoincrement=False, nullable=True),
sa.Column('description', sa.VARCHAR(length=255), autoincrement=False, nullable=True),
sa.PrimaryKeyConstraint('id', name='role_pkey'),
sa.UniqueConstraint('name', name='role_name_key'),
postgresql_ignore_search_path=False
)
op.create_table('roles_users',
sa.Column('user_id', sa.INTEGER(), autoincrement=False, nullable=True),
sa.Column('role_id', sa.INTEGER(), autoincrement=False, nullable=True),
sa.ForeignKeyConstraint(['role_id'], ['role.id'], name='roles_users_role_id_fkey'),
sa.ForeignKeyConstraint(['user_id'], ['users.id'], name='roles_users_user_id_fkey')
)
roles = [('ADMIN', 'Application administrator'),
('INTERNAL_USER', 'A user in the RDU team who can add, edit and view all pages'),
('DEPARTMENTAL_USER', 'A user that can view pages that have a status of departmental review')]
for r in roles:
role = Role(name=r[0], description=r[1])
session.add(role)
session.commit()
users = session.query(User).all()
for user in users:
for c in user.capabilities:
r = session.query(Role).filter(Role.name == c).one()
user.roles.append(r)
session.add(user)
session.commit()
op.drop_column('users', 'capabilities')
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
public_searchkey_aircraft = PGFunction(
schema="public",
signature="searchkey_aircraft(aircraft_number character varying)",
definition=
"returns character varying\n LANGUAGE plpgsql\nAS $function$\n DECLARE\n v_search_key VARCHAR(25);\n BEGIN\n v_search_key := TRIM(REGEXP_REPLACE(aircraft_number,'\\s|-','','gi'));\n IF (LENGTH(v_search_key) > 6) THEN\n v_search_key := RIGHT(v_search_key, 6);\n END IF;\n RETURN v_search_key;\n END\n ; \n $function$"
)
op.replace_entity(public_searchkey_aircraft)
op.drop_column('search_results', 'account_name')
op.drop_column('search_results', 'doc_storage_url')
op.drop_column('search_results', 'callback_url')
op.drop_index(op.f('ix_event_tracking_key_id'),
table_name='event_tracking')
op.drop_index(op.f('ix_event_tracking_event_ts'),
table_name='event_tracking')
op.drop_index(op.f('ix_event_tracking_event_tracking_type'),
table_name='event_tracking')
op.drop_table('event_tracking')
op.drop_table('event_tracking_types')
# Added manually
op.execute(DropSequence(Sequence('event_tracking_id_seq')))
def downgrade():
op.drop_table('menu_items_translation')
op.drop_table('menu_items')
op.execute(DropSequence(sa.Sequence(name='menu_items_id_seq')))
def drop_sequence(sequence):
if dialect_supports_sequences():
op.execute(DropSequence(sequence))
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.drop_index(op.f('ix_users_id'), table_name='users')
op.drop_index(op.f('ix_users_email'), table_name='users')
op.drop_table('users')
op.execute(DropSequence(Sequence("user_id_seq")))
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.drop_table('source_evidence', schema='city4age_ar')
op.execute(DropSequence(Sequence('source_evidence_id_seq', schema='city4age_ar')))
def downgrade():
# ### commands generated by Alembic ###
op.drop_table('error_detail')
op.drop_index('process_status_detail_idx', table_name='process_status_detail')
op.drop_table('process_status_detail')
op.execute(DropSequence(Sequence('process_status_detail_status_id_seq')))
def downgrade():
"""Downgrade database."""
op.drop_table('transaction')
if op._proxy.migration_context.dialect.supports_sequences:
op.execute(DropSequence(Sequence('transaction_id_seq')))
def downgrade():
op.drop_table("job")
op.execute(DropSequence(Sequence("job_id_seq")))
def downgrade():
op.drop_table(model.IMAP.__tablename__)
op.execute(DropSequence(sa.Sequence('email_imap_seq')))
def downgrade():
# ### commands auto generated by Alembic - please adjust! ###
op.drop_table("registry")
op.drop_table("integration_log")
op.execute(DropSequence(Sequence("registry_id_seq")))
def downgrade():
op.drop_constraint('sec_base_cluster_id_username_sec_type_key', model.SecurityBase.__tablename__)
op.drop_table(model.NTLM.__tablename__)
op.drop_table(model.AWSSecurity.__tablename__)
op.drop_table(model.APIKeySecurity.__tablename__)
op.drop_table(model.XPathSecurity.__tablename__)
op.execute(DropSequence(sa.Sequence('deliv_payl_seq')))
op.execute(DropSequence(sa.Sequence('msg_ns_seq')))
op.drop_table(model.HTTSOAPAudit.__tablename__)
op.execute(DropSequence(sa.Sequence('http_soap_audit_seq')))
op.drop_table(model.HTTSOAPAuditReplacePatternsJSONPointer.__tablename__)
op.execute(DropSequence(sa.Sequence('htp_sp_ad_rpl_p_jp_seq')))
op.drop_table(model.HTTSOAPAuditReplacePatternsXPath.__tablename__)
op.execute(DropSequence(sa.Sequence('htp_sp_ad_rpl_p_xp_seq')))
op.drop_table(model.PubSubConsumer.__tablename__)
op.execute(DropSequence(sa.Sequence('pub_sub_cons_seq')))
op.drop_table(model.PubSubProducer.__tablename__)
op.drop_table(model.OpenStackSwift.__tablename__)
op.execute(DropSequence(sa.Sequence('os_swift_seq')))
op.drop_table(model.AWSS3.__tablename__)
op.execute(DropSequence(sa.Sequence('aws_s3_seq')))
op.execute(DropSequence(sa.Sequence('deliv_seq')))
op.drop_table(model.PubSubTopic.__tablename__)
op.execute(DropSequence(sa.Sequence('pub_sub_topic_seq')))
op.execute(DropSequence(sa.Sequence('deliv_def_seq')))
op.drop_column(model.HTTPSOAP.__tablename__, 'serialization_type')
op.drop_column(model.HTTPSOAP.__tablename__, 'timeout')
op.alter_column(model.Cluster.__tablename__, 'odb_host', nullable=False)
op.alter_column(model.Cluster.__tablename__, 'odb_port', nullable=False)
op.alter_column(model.Cluster.__tablename__, 'odb_user', nullable=False)
op.alter_column(model.Cluster.__tablename__, 'odb_db_name', nullable=False)
def downgrade():
op.drop_table(model.XPath.__tablename__)
op.execute(DropSequence(sa.Sequence('msg_xpath_seq')))
op.drop_table(model.JSONPointer.__tablename__)
op.execute(DropSequence(sa.Sequence('msg_json_pointer_seq')))
def downgrade():
op.execute(DropSequence(Sequence('contact_tag_position')))
# ### commands auto generated by Alembic - please adjust! ###
op.drop_table('contact_tag')
op.drop_table('tag')