Optionally load connection arguments from the environment.

tonyandrewmeyer · tonyandrewmeyer · commit d30d54c26b10 · 2016-10-28T12:19:05.000+13:00
diff --git a/MySQLdb/connections.py b/MySQLdb/connections.py
@@ -11,6 +11,7 @@
      NotSupportedError, ProgrammingError
 import _mysql
 import re
+import os
 
 
 if not PY2:
@@ -144,7 +145,12 @@ class object, used to create cursors (keyword only)
         from MySQLdb.converters import conversions
         from weakref import proxy
 
-        kwargs2 = kwargs.copy()
+		# Load values from the environment if provided.
+		# Anything passed locally overrides the environment.
+		kwargs2 = {key.split("PY_MYSQL_CONNECT_", 1)[1].lower(): value
+		           for key, value in os.environ.items
+		           if key.startswith("PY_MYSQL_CONNECT_")}
+		kwargs2.update(kwargs)
 
         if 'database' in kwargs2:
             kwargs2['db'] = kwargs2.pop('database')
