From 0e66e77b3ee0400be39216379de22aebc7161116 Mon Sep 17 00:00:00 2001
From: "Fabian P. Schmidt" <kerel-fs@gmx.de>
Date: Sun, 31 Jan 2016 02:23:26 +0100
Subject: [PATCH] alembic: Link metadata to enable autogeneration of migrations

---
 alembic/env.py | 18 ++++++++----------
 1 file changed, 8 insertions(+), 10 deletions(-)

diff --git a/alembic/env.py b/alembic/env.py
index 2f7c635..36626d7 100644
--- a/alembic/env.py
+++ b/alembic/env.py
@@ -5,25 +5,23 @@ from alembic import context
 from sqlalchemy import engine_from_config, pool
 from logging.config import fileConfig
 
-# this is the Alembic Config object, which provides
-# access to the values within the .ini file in use.
-config = context.config
+
+# Provides access to the values within the .ini file in use.
+alembic_config = context.config
 
 # Interpret the config file for Python logging.
 # This line sets up loggers basically.
-fileConfig(config.config_file_name)
+fileConfig(alembic_config.config_file_name)
 
 # Get database path from ogn config
 os.environ.setdefault('OGN_CONFIG_MODULE', 'config.default')
 ogn_config = importlib.import_module(os.environ['OGN_CONFIG_MODULE'])
 
 alembic_config.set_main_option('sqlalchemy.url', ogn_config.SQLALCHEMY_DATABASE_URI)
-# add your model's MetaData object here
-# for 'autogenerate' support
-# from myapp import mymodel
-# target_metadata = mymodel.Base.metadata
-target_metadata = None
 
+# Import metadata for autogeneration of migrations
+from ogn.model import Base
+target_metadata = Base.metadata
 
 
 def run_migrations_offline():
@@ -54,7 +52,7 @@ def run_migrations_online():
 
     """
     connectable = engine_from_config(
-        config.get_section(config.config_ini_section),
+        alembic_config.get_section(alembic_config.config_ini_section),
         prefix='sqlalchemy.',
         poolclass=pool.NullPool)