django_test_manage.py.diff

--- django_test_manage.py.orig	2014-10-21 21:50:06.000000000 -0700
+++ django_test_manage.py	2014-10-21 22:59:51.000000000 -0700
@@ -2,16 +2,42 @@
 
 import os
 import sys
+import re
 
 from django.core.management import ManagementUtility
 
 from pycharm_run_utils import import_system_module
 
+def read_env():
+    """Pulled from Honcho code with minor updates, reads local default
+    environment variables from a .env file located in the project root
+    directory.
+
+    """
+    try:
+        with open('.env') as f:
+            content = f.read()
+    except IOError:
+        content = ''
+
+    for line in content.splitlines():
+        m1 = re.match(r'\A([A-Za-z_0-9]+)=(.*)\Z', line)
+        if m1:
+            key, val = m1.group(1), m1.group(2)
+            m2 = re.match(r"\A'(.*)'\Z", val)
+            if m2:
+                val = m2.group(1)
+            m3 = re.match(r'\A"(.*)"\Z', val)
+            if m3:
+                val = re.sub(r'\\(.)', r'\1', m3.group(1))
+            os.environ.setdefault(key, val)
+
 inspect = import_system_module("inspect")
 
 #import settings to prevent circular dependencies later on import django.db
 try:
     from django.conf import settings
+    read_env()
     apps = settings.INSTALLED_APPS
 except:
     pass