1
# vim: tabstop=4 shiftwidth=4 softtabstop=4
3
# Copyright (c) 2012 Red Hat, Inc.
6
# Licensed under the Apache License, Version 2.0 (the "License"); you may
7
# not use this file except in compliance with the License. You may obtain
8
# a copy of the License at
10
# http://www.apache.org/licenses/LICENSE-2.0
12
# Unless required by applicable law or agreed to in writing, software
13
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
14
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
15
# License for the specific language governing permissions and limitations
18
from sqlalchemy import MetaData
21
def upgrade(migrate_engine):
23
meta.bind = migrate_engine
25
# NOTE(dprince): Need to rename the leftover zones stuff and quota_new
26
# stuff from Essex for PostgreSQL.
27
if migrate_engine.name == "postgresql":
28
sql = """ALTER TABLE zones_id_seq RENAME TO cells_id_seq;
29
ALTER TABLE ONLY cells DROP CONSTRAINT zones_pkey;
30
ALTER TABLE ONLY cells ADD CONSTRAINT cells_pkey
33
ALTER TABLE quotas_new_id_seq RENAME TO quotas_id_seq;
34
ALTER TABLE ONLY quotas DROP CONSTRAINT quotas_new_pkey;
35
ALTER TABLE ONLY quotas ADD CONSTRAINT quotas_pkey
37
migrate_engine.execute(sql)
40
def downgrade(migrate_engine):
42
meta.bind = migrate_engine
44
if migrate_engine.name == "postgresql":
45
sql = """ALTER TABLE cells_id_seq RENAME TO zones_id_seq;
46
ALTER TABLE ONLY cells DROP CONSTRAINT cells_pkey;
47
ALTER TABLE ONLY cells ADD CONSTRAINT zones_pkey
50
ALTER TABLE quotas_id_seq RENAME TO quotas_new_id_seq;
51
ALTER TABLE ONLY quotas DROP CONSTRAINT quotas_pkey;
52
ALTER TABLE ONLY quotas ADD CONSTRAINT quotas_new_pkey
54
migrate_engine.execute(sql)