291 lines
9.2 KiB
Python
291 lines
9.2 KiB
Python
# GNU MediaGoblin -- federated, autonomous media hosting
|
|
# Copyright (C) 2012, 2012 MediaGoblin contributors. See AUTHORS.
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import copy
|
|
|
|
from sqlalchemy import (
|
|
Table, Column, MetaData, Index
|
|
Integer, Float, Unicode, UnicodeText, DateTime, Boolean,
|
|
ForeignKey, UniqueConstraint, PickleType)
|
|
from sqlalchemy.orm import sessionmaker
|
|
from sqlalchemy.ext.declarative import declarative_base
|
|
from sqlalchemy.sql import select, insert
|
|
from migrate import changeset
|
|
|
|
from mediagoblin.db.sql.base import GMGTableBase
|
|
|
|
|
|
# This one will get filled with local migrations
|
|
FULL_MIGRATIONS = {}
|
|
|
|
|
|
#######################################################
|
|
# Migration set 1: Define initial models, no migrations
|
|
#######################################################
|
|
|
|
Base1 = declarative_base(cls=GMGTableBase)
|
|
|
|
class Creature1(Base1):
|
|
__tablename__ = "creature"
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
name = Column(Unicode, unique=True, nullable=False, index=True)
|
|
num_legs = Column(Integer, nullable=False)
|
|
is_demon = Column(Boolean)
|
|
|
|
class Level1(Base1):
|
|
__tablename__ = "level"
|
|
|
|
id = Column(Unicode, primary_key=True)
|
|
name = Column(Unicode, unique=True, nullable=False, index=True)
|
|
description = Column(UnicodeText)
|
|
exits = Column(PickleType)
|
|
|
|
SET1_MODELS = [Creature1, Level1]
|
|
|
|
SET1_MIGRATIONS = []
|
|
|
|
#######################################################
|
|
# Migration set 2: A few migrations and new model
|
|
#######################################################
|
|
|
|
Base2 = declarative_base(cls=GMGTableBase)
|
|
|
|
class Creature2(Base2):
|
|
__tablename__ = "creature"
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
name = Column(Unicode, unique=True, nullable=False, index=True)
|
|
num_legs = Column(Integer, nullable=False)
|
|
|
|
class CreaturePower2(Base2):
|
|
__tablename__ = "creature_power"
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
creature = Column(
|
|
Integer, ForeignKey('creature.id'), nullable=False)
|
|
name = Column(Unicode)
|
|
description = Column(Unicode)
|
|
hitpower = Column(Integer, nullable=False)
|
|
|
|
class Level2(Base2):
|
|
__tablename__ = "level"
|
|
|
|
id = Column(Unicode, primary_key=True)
|
|
name = Column(Unicode)
|
|
description = Column(UnicodeText)
|
|
|
|
class LevelExit2(Base2):
|
|
__tablename__ = "level_exit"
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
name = Column(Unicode)
|
|
from_level = Column(
|
|
Unicode, ForeignKey('level.id'), nullable=False)
|
|
to_level = Column(
|
|
Unicode, ForeignKey('level.id'), nullable=False)
|
|
|
|
SET2_MODELS = [Creature2, CreaturePower2, Level2, LevelExit2]
|
|
|
|
|
|
@RegisterMigration(1, FULL_MIGRATIONS)
|
|
def creature_remove_is_demon(db_conn):
|
|
metadata = MetaData(bind=db_conn.engine)
|
|
creature_table = Table(
|
|
'creature', metadata,
|
|
autoload=True, autoload_with=db_conn.engine)
|
|
creature_table.drop_column('is_demon')
|
|
|
|
|
|
@RegisterMigration(2, FULL_MIGRATIONS)
|
|
def creature_powers_new_table(db_conn):
|
|
metadata = MetaData(bind=db_conn.engine)
|
|
creature_powers = Table(
|
|
'creature_power', metadata,
|
|
Column('id', Integer, primary_key=True),
|
|
Column('creature',
|
|
Integer, ForeignKey('creature.id'), nullable=False),
|
|
Column('name', Unicode),
|
|
Column('description', Unicode),
|
|
Column('hitpower', Integer, nullable=False))
|
|
metadata.create_all(db_conn.engine)
|
|
|
|
|
|
@RegisterMigration(3, FULL_MIGRATIONS)
|
|
def level_exits_new_table(db_conn):
|
|
# First, create the table
|
|
# -----------------------
|
|
metadata = MetaData(bind=db_conn.engine)
|
|
level_exits = Table(
|
|
'level_exit', metadata,
|
|
Column('id', Integer, primary_key=True),
|
|
Column('name', Unicode),
|
|
Column('from_level',
|
|
Integer, ForeignKey('level.id'), nullable=False),
|
|
Column('to_level',
|
|
Integer, ForeignKey('level.id'), nullable=False))
|
|
metadata.create_all(db_conn.engine)
|
|
|
|
# And now, convert all the old exit pickles to new level exits
|
|
# ------------------------------------------------------------
|
|
|
|
# Minimal representation of level table.
|
|
# Not auto-introspecting here because of pickle table. I'm not
|
|
# sure sqlalchemy can auto-introspect pickle columns.
|
|
levels = Table(
|
|
'level', metadata,
|
|
Column('id', Integer, primary_key=True),
|
|
Column('exits', PickleType))
|
|
|
|
# query over and insert
|
|
result = db_conn.execute(
|
|
select([levels], levels.c.exits!=None))
|
|
|
|
for level in result:
|
|
this_exit = level['exits']
|
|
|
|
# Insert the level exit
|
|
db_conn.execute(
|
|
level_exits.insert().values(
|
|
name=this_exit['name'],
|
|
from_level=this_exit['from_level'],
|
|
to_level=this_exit['to_level']))
|
|
|
|
# Finally, drop the old level exits pickle table
|
|
# ----------------------------------------------
|
|
levels.drop_column('exits')
|
|
|
|
|
|
# A hack! At this point we freeze-fame and get just a partial list of
|
|
# migrations
|
|
|
|
SET2_MIGRATIONS = copy.copy(FULL_MIGRATIONS)
|
|
|
|
#######################################################
|
|
# Migration set 3: Final migrations
|
|
#######################################################
|
|
|
|
Base3 = declarative_base(cls=GMGTableBase)
|
|
|
|
class Creature3(Base3):
|
|
__tablename__ = "creature"
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
name = Column(Unicode, unique=True, nullable=False, index=True)
|
|
num_limbs= Column(Integer, nullable=False)
|
|
|
|
class CreaturePower3(Base3):
|
|
__tablename__ = "creature_power"
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
creature = Column(
|
|
Integer, ForeignKey('creature.id'), nullable=False, index=True)
|
|
name = Column(Unicode)
|
|
description = Column(Unicode)
|
|
hitpower = Column(Float, nullable=False)
|
|
|
|
class Level3(Base3):
|
|
__tablename__ = "level"
|
|
|
|
id = Column(Unicode, primary_key=True)
|
|
name = Column(Unicode)
|
|
description = Column(UnicodeText)
|
|
|
|
class LevelExit3(Base3):
|
|
__tablename__ = "level_exit"
|
|
|
|
id = Column(Integer, primary_key=True)
|
|
name = Column(Unicode)
|
|
from_level = Column(
|
|
Unicode, ForeignKey('level.id'), nullable=False, index=True)
|
|
to_level = Column(
|
|
Unicode, ForeignKey('level.id'), nullable=False, index=True)
|
|
|
|
|
|
SET3_MODELS = [Creature3, CreaturePower3, Level3, LevelExit3]
|
|
|
|
|
|
@RegisterMigration(4, FULL_MIGRATIONS)
|
|
def creature_num_legs_to_num_limbs(db_conn):
|
|
metadata = MetaData(bind=db_conn.engine)
|
|
creature_table = Table(
|
|
'creature', metadata,
|
|
autoload=True, autoload_with=db_conn.engine)
|
|
creature_table.c.num_legs.alter(name="num_limbs")
|
|
|
|
|
|
@RegisterMigration(5, FULL_MIGRATIONS)
|
|
def level_exit_index_from_and_to_level(db_conn):
|
|
metadata = MetaData(bind=db_conn.engine)
|
|
level_exit = Table(
|
|
'level_exit', metadata,
|
|
autoload=True, autoload_with=db_conn.engine)
|
|
Index('ix_from_level', level_exit.c.from_level).create(engine)
|
|
Index('ix_to_exit', level_exit.c.to_exit).create(engine)
|
|
|
|
|
|
@RegisterMigration(6, FULL_MIGRATIONS)
|
|
def creature_power_index_creature(db_conn):
|
|
metadata = MetaData(bind=db_conn.engine)
|
|
creature_power = Table(
|
|
'creature_power', metadata,
|
|
autoload=True, autoload_with=db_conn.engine)
|
|
Index('ix_creature', creature_power.c.creature).create(engine)
|
|
|
|
|
|
@RegisterMigration(7, FULL_MIGRATIONS)
|
|
def creature_power_hitpower_to_float(db_conn):
|
|
metadata = MetaData(bind=db_conn.engine)
|
|
creature_power = Table(
|
|
'creature_power', metadata,
|
|
autoload=True, autoload_with=db_conn.engine)
|
|
creature_power.c.hitpower.alter(type=Float)
|
|
|
|
|
|
def _insert_migration1_objects(session):
|
|
# Insert creatures
|
|
session.add_all(
|
|
[Creature1(name='centipede',
|
|
num_legs=100,
|
|
is_demon=False),
|
|
Creature1(name='wolf',
|
|
num_legs=4,
|
|
is_demon=False),
|
|
# don't ask me what a wizardsnake is.
|
|
Creature1(name='wizardsnake',
|
|
num_legs=0,
|
|
is_demon=True)])
|
|
|
|
session.add_all(
|
|
[Level1(id='necroplex',
|
|
name='The Necroplex',
|
|
description='A complex full of pure deathzone.',
|
|
exits={
|
|
'deathwell': 'evilstorm',
|
|
'portal': 'central_park'}),
|
|
Level1(id='evilstorm',
|
|
name='Evil Storm',
|
|
description='A storm full of pure evil.',
|
|
exits={}), # you can't escape the evilstorm
|
|
Level1(id='central_park'
|
|
name='Central Park, NY, NY',
|
|
description="New York's friendly Central Park.",
|
|
exits={
|
|
'portal': 'necroplex'})])
|
|
|
|
session.commit()
|
|
|