From: Noufal Ibrahim <noufal@nibrahim.net.in>
Date: Sun, 16 Nov 2014 04:31:56 +0000 (+0530)
Subject: Adds cfg_vars to Config.
X-Git-Tag: rel_0_7_0~29^2~2
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=0586f71a641f3a685eee57c55bfdf9b9b1cdafa4;p=thirdparty%2Fsqlalchemy%2Falembic.git

Adds cfg_vars to Config.

This dictionary can be used to pass key value pairs that are used as
defaults while instantiating the SafeConfigParser.
---

diff --git a/alembic/config.py b/alembic/config.py
index f20abea7..d9dbca9b 100644
--- a/alembic/config.py
+++ b/alembic/config.py
@@ -51,10 +51,13 @@ class Config(object):
 
      ..versionadded:: 0.4
 
+    :param cfg_vars: A dictionary of keys and values that will be used
+    for substitution in the alembic config file.
+
     """
 
     def __init__(self, file_=None, ini_section='alembic', output_buffer=None,
-                 stdout=sys.stdout, cmd_opts=None):
+                 stdout=sys.stdout, cmd_opts=None, cfg_vars = {}):
         """Construct a new :class:`.Config`
 
         """
@@ -63,6 +66,7 @@ class Config(object):
         self.output_buffer = output_buffer
         self.stdout = stdout
         self.cmd_opts = cmd_opts
+        self.cfg_vars = cfg_vars
 
     cmd_opts = None
     """The command-line options passed to the ``alembic`` script.
@@ -113,7 +117,8 @@ class Config(object):
             here = os.path.abspath(os.path.dirname(self.config_file_name))
         else:
             here = ""
-        file_config = SafeConfigParser({'here': here})
+        self.cfg_vars['here'] = here
+        file_config = SafeConfigParser(self.cfg_vars)
         if self.config_file_name:
             file_config.read([self.config_file_name])
         else: