Matplotlib toolkits support

japborst committed 9 years ago

Commit 718fa991 parent 79a20252 files · +16 −5Browse files

Unified · Split

2 files

         name = at_css('h1').content.strip
         name.remove! "\u{00b6}"
         name.remove! 'matplotlib.'
+        name.remove! 'mpl_toolkits.'
+        name.remove! ' API'
         name.remove! %r{ \(.*\)}
         name.downcase!
         name
         entries = []
         css('.class > dt[id]', '.exception > dt[id]', '.attribute > dt[id]').each do |node|
-          entries << [node['id'].remove('matplotlib.'), node['id']]
+          entry_name = node['id'].remove('matplotlib.').remove('mpl_toolkits.')
+          entries << [entry_name, node['id']]
         end
         css('.data > dt[id]').each do |node|
           if node['id'].split('.').last.upcase! # skip constants
-            entries << [node['id'].remove('matplotlib.'), node['id']]
+            entry_name = node['id'].remove('matplotlib.').remove('mpl_toolkits.')
+            entries << [entry_name, node['id']]
           end
         end
         css('.function > dt[id]', '.method > dt[id]', '.classmethod > dt[id]').each do |node|
-          entries << [node['id'].remove('matplotlib.') + '()', node['id']]
+          entry_name = node['id'].remove('matplotlib.').remove('mpl_toolkits.')
+          entries << [entry_name + '()', node['id']]
         end
         entries

 module Docs
   class Matplotlib < UrlScraper
+    include MultipleBaseUrls
     self.name = 'Matplotlib'
     self.type = 'sphinx'
     self.root_path = 'index.html'
     html_filters.push 'matplotlib/entries', 'sphinx/clean_html'
     options[:container] = '.body'
-    options[:skip] = %w(api_changes.html)
+    options[:skip] = %w(api_changes.html tutorial.html faq.html)
     options[:attribution] = <<-HTML
       &copy; 2012&ndash;2016 Matplotlib Development Team. All rights reserved.<br>
     version '1.5' do
       self.release = '1.5.1'
-      self.base_url = 'http://matplotlib.org/1.5.1/api/'
+      self.base_urls = [
+        'http://matplotlib.org/1.5.1/api/',
+        'http://matplotlib.org/1.5.1/mpl_toolkits/mplot3d/',
+        'http://matplotlib.org/1.5.1/mpl_toolkits/axes_grid/api/'
+      ]
     end
   end
 end