From e9ab3f31504a176f483c8340b3ad2ba7679fe285 Mon Sep 17 00:00:00 2001
From: Abhilash Raj
Date: Wed, 24 Sep 2014 23:13:12 +0530
Subject: added support for migrations via alembic

---
 src/mailman/database/factory.py | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'src/mailman/database/factory.py')

diff --git a/src/mailman/database/factory.py b/src/mailman/database/factory.py
index c06f75031..01a1357dd 100644
--- a/src/mailman/database/factory.py
+++ b/src/mailman/database/factory.py
@@ -29,6 +29,9 @@ __all__ = [
 import os
 import types
 
+from alembic.config import Config
+from alembic import command
+
 from flufl.lock import Lock
 from zope.interface import implementer
 from zope.interface.verify import verifyObject
@@ -39,6 +42,8 @@ from mailman.interfaces.database import IDatabase, IDatabaseFactory
 from mailman.utilities.modules import call_name
 
 
+alembic_cfg = Config("./alembic.ini")
+
 
 @implementer(IDatabaseFactory)
 class DatabaseFactory:
@@ -53,6 +58,7 @@ class DatabaseFactory:
             verifyObject(IDatabase, database)
             database.initialize()
             Model.metadata.create_all(database.engine)
+            command.stamp(alembic_cfg, "head")
             database.commit()
             return database
 
@@ -81,6 +87,7 @@ class DatabaseTestingFactory:
         verifyObject(IDatabase, database)
         database.initialize()
         Model.metadata.create_all(database.engine)
+        command.stamp(alembic_cfg, "head")
         database.commit()
         # Make _reset() a bound method of the database instance.
         database._reset = types.MethodType(_reset, database)
-- 
cgit v1.2.3-70-g09d2