get env to have all the arguments before and after context is set up

1 files changed, 5 insertions, 5 deletions

diff --git a/alembic/templates/pylons/env.py b/alembic/templates/pylons/env.py
index e202cea..799ba02 100644
--- a/alembic/templates/pylons/env.py
+++ b/alembic/templates/pylons/env.py
@@ -4,19 +4,19 @@ Place 'pylons_config_file' into alembic.ini, and the application will
 be loaded from there.
 
 """
-from alembic import options, context
+from alembic import config, context
 from paste.deploy import loadapp
 import logging
 
 try:
     # if pylons app already in, don't create a new app
-    from pylons import config
-    config['__file__']
+    from pylons import config as pylons_config
+    pylons_config['__file__']
 except:
     # can use config['__file__'] here, i.e. the Pylons
     # ini file, instead of alembic.ini
-    config_file = options.get_main_option('pylons_config_file')
-    config_file = options.config_file_name
+    config_file = config.get_main_option('pylons_config_file')
+    config_file = config.config_file_name
     logging.config.fileConfig(config_file)
     wsgi_app = loadapp('config:%s' % config_file, relative_to='.')