From f127a1cd39d098647adf241ffda5fb78d0ace950 Mon Sep 17 00:00:00 2001
From: Josh Toft <josh@wealthfront.com>
Date: Tue, 10 Mar 2015 12:28:00 -0700
Subject: [PATCH] Add extra link module search paths

* Adds .eggs
* Adds recursive pathing for all user paths
---
 src/simian/util/link_module.sh | 33 +++++++++++++++++++++++++++++++--
 1 file changed, 31 insertions(+), 2 deletions(-)

diff --git a/src/simian/util/link_module.sh b/src/simian/util/link_module.sh
index e3b7966..d4133be 100755
--- a/src/simian/util/link_module.sh
+++ b/src/simian/util/link_module.sh
@@ -10,9 +10,26 @@ function find_module() {
   python <<EOF
 import imp
 try:
- print imp.find_module('$1')[1]
+  print imp.find_module('$1')[1]
 except ImportError:
- pass
+  pass
+EOF
+}
+
+function find_sitepackage_module() {
+  python <<EOF
+import os
+import re
+try:
+  user_paths = os.environ['PYTHONPATH'].split(os.pathsep)
+except KeyError:
+  user_paths = []
+
+pymod = '$1'
+for path in user_paths:
+  for f in os.listdir(path):
+    if re.search(r'(' + pymod + ')', os.path.join(path, f), re.I):
+      print os.path.join(path, f)
 EOF
 }
 
@@ -21,6 +38,11 @@ function find_egg_file() {
   if [[ ! -z "${egg}" ]]; then
     echo "${egg}"
   fi
+
+  egg=$(find .eggs -type f -maxdepth 1 -name ${module}-*.egg 2>/dev/null)
+  if [[ ! -z "${egg}" ]]; then
+    echo "${egg}"
+  fi
 }
 
 function find_egg_dir() {
@@ -53,6 +75,13 @@ function link_module() {
     return
   fi
 
+  local path=$(find_sitepackage_module ${module})
+  if [[ ! -z "${path}" ]]; then
+    rm -f "${GAE_BUNDLE}/${module}"
+    ln -s "${path}" "${GAE_BUNDLE}/${module}"
+    return
+  fi
+
   echo "ERROR: path not found for ${module}. symlink creation failure."
   exit 1
 }