def downgrade(migrate_engine):
meta = MetaData(bind=migrate_engine)
service_flavors = Table('service_flavors', meta, autoload=True)
conn = migrate_engine.connect()
trans = conn.begin()
try:
delete = service_flavors.delete()\
.where(service_flavors.c.service_name=='database')
conn.execute(delete)
trans.commit()
except:
trans.rollback()
raise
service_flavors.c.ram.drop()
service_flavors.c.vcpus.drop()
try:
insert = service_flavors.insert()\
.execute(id="3", service_name="database", flavor_name="large",
flavor_id="103", deleted=0, created_at=datetime.datetime.now(),
updated_at=datetime.datetime.now())
trans.commit
except:
trans.rollback()
raise
def upgrade(migrate_engine):
meta = MetaData(bind=migrate_engine)
service_flavors = Table('service_flavors', meta, autoload=True)
conn = migrate_engine.connect()
trans = conn.begin()
try:
delete = service_flavors.delete()\
.where(service_flavors.c.service_name=='database')
conn.execute(delete)
trans.commit()
except:
trans.rollback()
raise
ramc = Column('ram', Integer())
ramc.create(service_flavors)
vcpusc = Column('vcpus', Integer())
vcpusc.create(service_flavors)
conn = migrate_engine.connect()
trans = conn.begin()
try:
for flavor in SERVICE_FLAVORS:
insert = service_flavors.insert()\
.execute(id=flavor['id'], service_name="database", flavor_name=flavor['flavor_name'],
flavor_id=flavor['flavor_id'], deleted=0, ram=flavor['ram'],
vcpus=flavor['vcpus'], created_at=datetime.datetime.now(), updated_at=datetime.datetime.now())
trans.commit
except:
trans.rollback()
raise