From 744a8944b0e79a79b0801616d2a6f339be0c8cca Mon Sep 17 00:00:00 2001
From: Mike McQuaid <mike@mikemcquaid.com>
Date: Sat, 4 Jan 2014 13:18:27 +0000
Subject: [PATCH] caveats: show python caveats if modules installed.

---
 Library/Homebrew/caveats.rb | 18 +++++++++++++++++-
 Library/Homebrew/keg.rb     |  4 ++++
 2 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/Library/Homebrew/caveats.rb b/Library/Homebrew/caveats.rb
index 466d0b0b5b..db95fe86e4 100644
--- a/Library/Homebrew/caveats.rb
+++ b/Library/Homebrew/caveats.rb
@@ -7,11 +7,12 @@ class Caveats
 
   def caveats
     caveats = []
-    caveats << f.caveats
+    caveats << f.caveats if f.caveats.to_s.length > 0
     caveats << f.keg_only_text if f.keg_only? && f.respond_to?(:keg_only_text)
     caveats << bash_completion_caveats
     caveats << zsh_completion_caveats
     caveats << plist_caveats
+    caveats << python_caveats
     caveats.compact.join("\n")
   end
 
@@ -43,6 +44,21 @@ class Caveats
     end
   end
 
+  def python_caveats
+    site_packages = if f.keg_only?
+      "#{f.opt_prefix}/lib/python2.7/site-packages"
+    else
+      "#{HOMEBREW_PREFIX}/lib/python2.7/site-packages"
+    end
+    if keg and keg.python_site_packages_installed? \
+      and !ENV['PYTHONPATH'].to_s.include? site_packages
+      <<-EOS.undent
+        Set PYTHONPATH if you want Python to find your site-packages:
+          export PYTHONPATH=#{site_packages}:$PYTHONPATH
+      EOS
+    end
+  end
+
   def plist_caveats
     s = []
     if f.plist or (keg and keg.plist_installed?)
diff --git a/Library/Homebrew/keg.rb b/Library/Homebrew/keg.rb
index e1178d5dd2..8ad5ffaeae 100644
--- a/Library/Homebrew/keg.rb
+++ b/Library/Homebrew/keg.rb
@@ -92,6 +92,10 @@ class Keg < Pathname
     end
   end
 
+  def python_site_packages_installed?
+    (self/'lib/python2.7/site-packages').directory?
+  end
+
   def version
     require 'version'
     Version.new(basename.to_s)