tableau_server_client 0.0.7 → 0.0.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 886e88eedf330cdb09d92a5fe2bd7e1452c19c9e413fb5767098376c17d03608
-  data.tar.gz: 5884c44fc430ff3ac02fa6d94e3f5a4edc426de9597281f019694a5fb2a76ace
+  metadata.gz: 0e288311bd523063625df63be79cff2295f1d48b0a17c2dc8712145be68d44ec
+  data.tar.gz: 02db553763acd7e4de9fa06966dde66d2a92a6bcae70ed1734c16e7ddbe474bd
 SHA512:
-  metadata.gz: 32050b19c3d0ef603e5476f879070cf3c14cdefc522beb4a4367fb33272e79c62a1a12cc65e2799a53aa6035f8c9e3ea5cd43eac3dc59c5f4db38beee53d13cb
-  data.tar.gz: a682c5b349b0ff0a36e15535b2da7fc52a77ba2275a2c58ee0cd8d820b3a2b1f389223ed6eb446b421234c4111f4c25f2f2e6ea8ca96623a2b18ba750afbd624
+  metadata.gz: 8c105d2be4dd5e12fc6230fe3edc7c3927c5ed900912e8130a68cbc8dfd948b66b74ffd187a3bf8e66cc877d7edb2e0a84bd11681c11ba1f588b4c6e6dfb64dc
+  data.tar.gz: 66d0710f717c507e1f5e32ac7dcebbb46d6e07c9f4bc71afb9d09f181ccb9636004f1fe82bf3bf694e01bc77f2b38e9ee4ae875a31907e3df38d47732636c4fc

data/lib/tableau_server_client/client.rb

@@ -5,6 +5,9 @@ require 'tableau_server_client/request_url'
 require 'tableau_server_client/request_builder'
 require 'tableau_server_client/token'
 require 'tableau_server_client/paginatable_response'
+require 'tempfile'
+require 'zip'
+require 'stringio'
 
 module TableauServerClient
 
@@ -46,6 +49,25 @@ module TableauServerClient
     def create(resource)
     end
 
+    def download(resource_location)
+      req_url = request_url("#{resource_location.path}/content", resource_location.query_params)
+      response = session.get req_url.to_s
+      type, disposition = response.headers.values_at('content-type', 'content-disposition')
+      case type
+      when 'application/xml'
+        return Nokogiri::XML(response.body)
+      when 'application/octet-stream'
+        Zip::InputStream.open(StringIO.new(response.body)) do |io|
+          while entry = io.get_next_entry
+            return Nokogiri::XML(io.read) if entry.name =~ /.*\.(tds|twb)/
+          end
+          raise "TDS or TWB file not found for: #{resource_location.path}"
+        end
+      else
+        raise "Unknown content-type: #{type}"
+      end
+    end
+
     def update(resource)
       session.put do |req|
         req.url request_url(resource.path).to_s

data/lib/tableau_server_client/resources/resource.rb

@@ -26,7 +26,7 @@ module TableauServerClient
 
       def self.location(prefix, id=nil, filter: [])
         path = [prefix, "#{resource_name}s", id].compact.join("/")
-        Location.new(self, path, filter)
+        Location.new(self, path, filter.empty? ? {} : {filter: filter.join(',')})
       end
 
       def self.extract_attributes(xml)
@@ -49,6 +49,10 @@ module TableauServerClient
         @path
       end
 
+      def location(query_params: {})
+        Location.new(self, path, query_params)
+      end
+
       def site_path
         self.class.extract_site_path(path)
       end
@@ -59,21 +63,13 @@ module TableauServerClient
 
       class Location
 
-        def initialize(klass, path, filter)
+        def initialize(klass, path, query_params)
           @klass = klass
           @path = path
-          @filter = filter
-        end
-
-        attr_reader :klass, :path
-
-        def filter
-          @filter.empty? ? {} : {filter: @filter.join(",")}
+          @query_params = query_params
         end
 
-        def query_params
-          filter
-        end
+        attr_reader :klass, :path, :query_params
       end
 
       class Attribute < String

data/lib/tableau_server_client/resources/workbook.rb

@@ -49,6 +49,70 @@ module TableauServerClient
         @client.update self
       end
 
+      def custom_queries
+        relations.select {|r| r['type'] == 'text' }.map {|c| c.content }
+      end
+
+      def tables
+        tables  = []
+        redshift_connections = named_connections.select {|c| c.class == 'redshift' }.map {|c| c.name }
+        relations.each do |rel|
+          next unless redshift_connections.include? rel['connection']
+          case rel['type']
+          when 'table'
+            tables << rel['table']
+          when 'text'
+            tables.concat extract_tables(rel.content)
+          else
+            next
+          end
+        end
+        tables.map {|t| t.gsub(/[\[\]")]/, '')}.uniq
+      end
+
+      private
+
+      NamedConnection = Struct.new("NamedConnections", :class, :caption, :name)
+      def named_connections
+        download.xpath('//named-connection').map do |c|
+          NamedConnection.new(c.first_element_child['class'], c['caption'], c['name'])
+        end
+      end
+
+      def download
+        @twb ||= @client.download location(query_params: {"includeExtract": "False"})
+      end
+
+      def relations
+        download.xpath('//datasources//datasource//relation')
+      end
+
+      def extract_tables(query)
+        q = query.dup
+        q.gsub!(/(\<\[Parameters\]\.\[.*?\]\>)/, "'\\1'")\
+          .gsub!(/(--[^\r\n]*)|(\/\*[\w\W]*?(?=\*\/)\*\/)/m, '')\
+          .gsub!(/[\t\r\n]/, ' ')\
+          .gsub!(/\s+/, ' ')
+
+        tables = []
+        may_be_table = false
+        q.split(' ').each do |t|
+          t.downcase!
+          if may_be_table
+            tables << t unless t =~ /(^select|^\(.*)/
+            may_be_table = false
+          end
+          if ['from', 'join'].include?(t)
+             may_be_table = true
+          end
+        end
+        tables
+        # ParseError with sub-query without alias name
+        #PgQuery.parse(no_parameter_query).tables.each do |t|
+        #  yield t
+        #end
+      end
+
     end
   end
 end

data/lib/tableau_server_client/version.rb

@@ -1,3 +1,3 @@
 module TableauServerClient
-  VERSION = "0.0.7"
+  VERSION = "0.0.8"
 end

data/tableau_server_client.gemspec

@@ -29,4 +29,5 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'faraday', '>= 0.14.0'
   spec.add_dependency 'nokogiri', '>= 1.8.2'
   spec.add_dependency 'pry', '>= 0.11.3'
+  spec.add_dependency 'rubyzip', '>= 1.2.1'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tableau_server_client
 version: !ruby/object:Gem::Version
-  version: 0.0.7
+  version: 0.0.8
 platform: ruby
 authors:
 - shimpeko
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-07-27 00:00:00.000000000 Z
+date: 2018-08-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -108,6 +108,20 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: 0.11.3
+- !ruby/object:Gem::Dependency
+  name: rubyzip
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.2.1
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 1.2.1
 description: REST API Client for Tableau Server.
 email:
 - shimpeko@swimmingpython.com