Split Tensorflow into separate Python, C++ and Guide docs

metucclub · Nov 15, 2020 · c7a2a37 · c7a2a37
1 parent d7ca038
commit c7a2a37
Show file tree

Hide file tree

Showing 10 changed files with 44 additions and 13 deletions.
diff --git a/lib/docs/filters/tensorflow/entries.rb b/lib/docs/filters/tensorflow/entries.rb
@@ -15,9 +15,9 @@ def get_name
       end
 
       def get_type
-        if version == 'Guide' and base_url.path.start_with?('/guide')
+        if base_url.path.start_with?('/guide')
           'Guides'
-        elsif version == 'Guide' and base_url.path.start_with?('/tutorials')
+        elsif base_url.path.start_with?('/tutorials')
           'Tutorials'
         elsif slug.start_with?('tf/audio')
           'tf.audio'

diff --git a/lib/docs/scrapers/tensorflow.rb → lib/docs/scrapers/tensorflow/tensorflow.rb b/lib/docs/scrapers/tensorflow.rb → lib/docs/scrapers/tensorflow/tensorflow.rb
@@ -1,10 +1,7 @@
-# frozen_string_literal: true
-
 module Docs
   class Tensorflow < UrlScraper
     self.name = 'TensorFlow'
     self.type = 'tensorflow'
-    self.release = '2.1'
     self.root_path = 'index.html'
     self.links = {
       home: 'https://www.tensorflow.org/',
@@ -22,17 +19,19 @@ class Tensorflow < UrlScraper
       Code samples licensed under the Apache 2.0 License.
     HTML
 
-    version 'Python' do
-      self.base_url = 'https://www.tensorflow.org/api_docs/python/'
+    version '2.1' do
+      self.release = '2.1.0'
+      self.base_url = "https://www.tensorflow.org/versions/r#{version}/api_docs/python"
     end
 
-    version 'C++' do
-      self.base_url = 'https://www.tensorflow.org/api_docs/cc/'
+    version '2.0' do
+      self.release = '2.0.0'
+      self.base_url = "https://www.tensorflow.org/versions/r#{version}/api_docs/python"
     end
 
-    version 'Guide' do
-      include MultipleBaseUrls
-      self.base_urls = ['https://www.tensorflow.org/guide/', 'https://www.tensorflow.org/tutorials/']
+    version '1.15' do
+      self.release = '1.15.0'
+      self.base_url = "https://www.tensorflow.org/versions/r#{version}/api_docs/python"
     end
 
     def get_latest_version(opts)
@@ -42,7 +41,7 @@ def get_latest_version(opts)
     private
 
     def parse(response)
-      unless response.url == root_url || self.class.version == 'Guide'
+      unless response.url == root_url
         response.body.sub!(/<nav class="devsite-nav-responsive-sidebar.+?<\/nav>/m, '')
         response.body.gsub!(/<li class="devsite-nav-item">.+?<\/li>/m, '')
       end

diff --git a/lib/docs/scrapers/tensorflow/tensorflow_cpp.rb b/lib/docs/scrapers/tensorflow/tensorflow_cpp.rb
@@ -0,0 +1,21 @@
+module Docs
+  class TensorflowCpp < Tensorflow
+    self.name = 'TensorFlow C++'
+    self.slug = 'tensorflow_cpp'
+
+    version '2.1' do
+      self.release = '2.1.0'
+      self.base_url = "https://www.tensorflow.org/versions/r#{version}/api_docs/cc"
+    end
+
+    version '2.0' do
+      self.release = '2.0.0'
+      self.base_url = "https://www.tensorflow.org/versions/r#{version}/api_docs/cc"
+    end
+
+    version '1.15' do
+      self.release = '1.15.0'
+      self.base_url = "https://www.tensorflow.org/versions/r#{version}/api_docs/cc"
+    end
+  end
+end
diff --git a/lib/docs/scrapers/tensorflow/tensorflow_guide.rb b/lib/docs/scrapers/tensorflow/tensorflow_guide.rb
@@ -0,0 +1,9 @@
+module Docs
+  class TensorflowGuide < Tensorflow
+    include MultipleBaseUrls
+
+    self.name = 'TensorFlow Guides'
+    self.slug = 'tensorflow_guide'
+    self.base_urls = ['https://www.tensorflow.org/guide/', 'https://www.tensorflow.org/tutorials/']
+  end
+end
diff --git a/public/icons/docs/tensorflow_cpp/16.png b/public/icons/docs/tensorflow_cpp/16.png
diff --git a/public/icons/docs/tensorflow_cpp/16@2x.png b/public/icons/docs/tensorflow_cpp/16@2x.png
diff --git a/public/icons/docs/tensorflow_cpp/SOURCE b/public/icons/docs/tensorflow_cpp/SOURCE
@@ -0,0 +1 @@
+https://www.tensorflow.org/images/favicon.png
diff --git a/public/icons/docs/tensorflow_guide/16.png b/public/icons/docs/tensorflow_guide/16.png
diff --git a/public/icons/docs/tensorflow_guide/16@2x.png b/public/icons/docs/tensorflow_guide/16@2x.png
diff --git a/public/icons/docs/tensorflow_guide/SOURCE b/public/icons/docs/tensorflow_guide/SOURCE
@@ -0,0 +1 @@
+https://www.tensorflow.org/images/favicon.png