Auto build pyplot from setup.

pelson · pelson · commit ade496c7fdee · 2012-06-06T19:32:57.000+01:00
diff --git a/lib/matplotlib/pyplot.py b/lib/matplotlib/pyplot.py
@@ -1,3 +1,6 @@
+# This file is entirely autogenerated by setup.py, and any changes to it will be lost.
+# The leading part of this file can be modified in src/pyplot_header.py
+# whilst the latter part is generated by boilerplate.py
 """
 Provides a MATLAB-like plotting framework.
 
@@ -2220,6 +2223,10 @@ def spy(Z, precision=0, marker=None, markersize=None, aspect='equal', hold=None,
         sci(ret)
     return ret
 
+
+################# REMAINING CONTENT GENERATED BY boilerplate.py ##############
+
+
 # This function was autogenerated by boilerplate.py.  Do not edit as
 # changes will be lost
 @autogen_docstring(Axes.acorr)
@@ -2925,7 +2932,7 @@ def step(x, y, *args, **kwargs):
 # This function was autogenerated by boilerplate.py.  Do not edit as
 # changes will be lost
 @autogen_docstring(Axes.streamplot)
-def streamplot(x, y, u, v, density=1, linewidth=None, color=None, cmap=None, norm=None, arrowsize=1, arrowstyle='-|>', minlength=0.10000000000000001, hold=None):
+def streamplot(x, y, u, v, density=1, linewidth=None, color=None, cmap=None, norm=None, arrowsize=1, arrowstyle='-|>', minlength=0.1, hold=None):
     ax = gca()
     # allow callers to override the hold state by passing hold=True|False
     washold = ax.ishold()
diff --git a/setup.py b/setup.py
@@ -248,6 +248,37 @@ def add_dateutil():
 template = template.replace("    use = 'Agg'", "    use = '%s'"%rc['backend'])
 open('lib/matplotlib/mpl-data/matplotlib.conf', 'w').write(template)
 
+def build_pyplot():
+    import StringIO
+    import os
+    import sys
+    
+    pyplot_path = os.path.join('lib', 'matplotlib', 'pyplot.py')
+    
+    pyplot_orig = file(pyplot_path, 'r').readlines()
+    
+    # this is the magic line that must exist in pyplot, after which the boilerplate content will be
+    # appended
+    bp_marker = '################# REMAINING CONTENT GENERATED BY boilerplate.py ##############\n'
+    
+    try:
+        pyplot_orig = pyplot_orig[:pyplot_orig.index(bp_marker)+1]
+    except IndexError:
+        raise ValueError('The pyplot.py file *must* have the exact line: %s' % bp_marker)
+        
+    pyplot = file(pyplot_path, 'w')
+    pyplot.writelines(pyplot_orig)
+    pyplot.write('\n\n')
+    
+    # importing boilerplate prints code. So capture the stdout and put it into the pyplot file.
+    old_stdout = sys.stdout
+    sys.stdout = pyplot
+    import boilerplate
+    sys.stdout = old_stdout
+    
+# Write the matplotlib.pyplot file
+build_pyplot()
+
 try: additional_params # has setupegg.py provided
 except NameError: additional_params = {}
 
