Support for setuptools/pkg_resources to build and use

astraw · astraw · commit 3f03686f0d31 · 2005-12-11T23:53:55.000Z
matplotlib as an egg. Still allows matplotlib to exist
using a traditional distutils install. - ADS

svn path=/trunk/matplotlib/; revision=1914
diff --git a/CHANGELOG b/CHANGELOG
@@ -1,4 +1,8 @@
-h2005-12-03 Modified setup to build matplotlibrc based on compile time
+2005-12-11 Support for setuptools/pkg_resources to build and use
+           matplotlib as an egg. Still allows matplotlib to exist
+           using a traditional distutils install. - ADS
+
+2005-12-03 Modified setup to build matplotlibrc based on compile time
            findings.  It will set numerix in the order of scipy,
            numarray, Numeric depending on which are founds, and
            backend as in preference order GTKAgg, WXAgg, TkAgg, GTK,
@@ -11,12 +15,12 @@ h2005-12-03 Modified setup to build matplotlibrc based on compile time
 	   than from scipy.fftpack import *
 
 2005-12-03 Applied some fixes to Nicholas Young's nonuniform image
-patch
+           patch
 
- 2005-12-01 Applied Alex Gontmakher hatch patch - PS only for now
+2005-12-01 Applied Alex Gontmakher hatch patch - PS only for now
 
 2005-11-30 Added Rob McMullen's EMF patch
-h
+
 2005-11-30 Added Daishi's patch for scipy
 
 2005-11-30 Fixed out of bounds draw markers segfault in agg
diff --git a/lib/matplotlib/__init__.py b/lib/matplotlib/__init__.py
@@ -181,6 +181,11 @@
 except ImportError: _havedate = False
 else: _havedate = True
 
+try:
+    import pkg_resources # pkg_resources is part of setuptools
+except ImportError: _have_pkg_resources = False
+else: _have_pkg_resources = True
+
 if not _python23:
     def enumerate(seq):
          for i in range(len(seq)):
@@ -363,6 +368,20 @@ def _get_configdir():
 def _get_data_path():
     'get the path to matplotlib data'
 
+    if _have_pkg_resources:
+        try:
+            dist = pkg_resources.get_distribution('matplotlib')
+        except pkg_resources.DistributionNotFound:
+            # pkg_resources is installed, but setuptools wasn't used into install matplotlib.
+            used_setuptools_to_install_MPL = False
+        else:
+            used_setuptools_to_install_MPL = True
+
+        if used_setuptools_to_install_MPL:
+            req = pkg_resources.Requirement.parse('matplotlib')
+            path = pkg_resources.resource_filename(req, 'share/matplotlib')
+            return path
+
     if os.environ.has_key('MATPLOTLIBDATA'):
         path = os.environ['MATPLOTLIBDATA']
         if os.path.isdir(path): return path
diff --git a/setup.py b/setup.py
@@ -58,6 +58,12 @@
 if os.path.exists('MANIFEST'): os.remove('MANIFEST')
 
 from distutils.core import setup
+
+try:
+    from setuptools import setup # use setuptools if possible
+except ImportError:
+    pass
+    
 import sys,os
 import glob
 from distutils.core import Extension
