diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/mailman/database/alembic/versions/429e08420177_initial.py | 29 | ||||
| -rw-r--r-- | src/mailman/database/factory.py | 60 |
2 files changed, 87 insertions, 2 deletions
diff --git a/src/mailman/database/alembic/versions/429e08420177_initial.py b/src/mailman/database/alembic/versions/429e08420177_initial.py new file mode 100644 index 000000000..e8d612676 --- /dev/null +++ b/src/mailman/database/alembic/versions/429e08420177_initial.py @@ -0,0 +1,29 @@ +"""Initial migration + +This empty migration file makes sure there is always an alembic_version in the +database. As a consequence, if the DB version is reported as None, it means the +database needs to be created from scratch with SQLAlchemy itself. + +It also removes the "version" table left over from Storm (if it exists). + + +Revision ID: 429e08420177 +Revises: None +Create Date: 2014-10-02 10:18:17.333354 + +""" + +# revision identifiers, used by Alembic. +revision = '429e08420177' +down_revision = None + +from alembic import op +import sqlalchemy as sa + + +def upgrade(): + op.drop_table('version') + + +def downgrade(): + pass diff --git a/src/mailman/database/factory.py b/src/mailman/database/factory.py index 0a295331a..1976962d3 100644 --- a/src/mailman/database/factory.py +++ b/src/mailman/database/factory.py @@ -29,7 +29,12 @@ __all__ = [ import os import types +from alembic import command +from alembic.config import Config as AlembicConfig +from alembic.migration import MigrationContext +from alembic.script import ScriptDirectory from flufl.lock import Lock +from sqlalchemy import MetaData from zope.interface import implementer from zope.interface.verify import verifyObject @@ -52,13 +57,64 @@ class DatabaseFactory: database = call_name(database_class) verifyObject(IDatabase, database) database.initialize() - Model.metadata.create_all(database.engine) - database.stamp() + schema_mgr = SchemaManager(database) + schema_mgr.setup_db() database.commit() return database +class SchemaManager: + + LAST_STORM_SCHEMA_VERSION = '20130406000000' + + def __init__(self, database): + self.database = database + self.alembic_cfg = AlembicConfig() + self.alembic_cfg.set_main_option( + "script_location", config.alembic['script_location']) + self.script = ScriptDirectory.from_config(self.alembic_cfg) + + def get_storm_schema_version(self): + md = MetaData() + md.reflect(bind=self.database.engine) + if "version" not in md.tables: + return None + Version = md.tables["version"] + last_version = self.database.store.query(Version.c.version).filter( + Version.c.component == "schema" + ).order_by(Version.c.version.desc()).first() + return last_version + + def setup_db(self): + context = MigrationContext.configure(self.database.store.connection()) + current_rev = context.get_current_revision() + head_rev = self.script.get_current_head() + if current_rev == head_rev: + return head_rev # already at the latest revision, nothing to do + if current_rev == None: + # no alembic information + storm_version = self.get_storm_schema_version() + if storm_version is None: + # initial DB creation + Model.metadata.create_all(self.database.engine) + command.stamp(self.alembic_cfg, "head") + else: + # DB from a previous version managed by Storm + if storm_version.version < self.LAST_STORM_SCHEMA_VERSION: + raise RuntimeError( + "Upgrading while skipping beta version is " + "unsupported, please install the previous " + "Mailman beta release") + # Run migrations to remove the Storm-specific table and + # upgrade to SQLAlchemy & Alembic + command.upgrade(self.alembic_cfg, "head") + elif db_rev != head_rev: + command.upgrade(self.alembic_cfg, "head") + return head_rev + + + def _reset(self): """See `IDatabase`.""" # Avoid a circular import at module level. |