gazer 0.2.9

data/lib/gzr/commands/query/runquery.rb

@@ -0,0 +1,102 @@
+# The MIT License (MIT)
+
+# Copyright (c) 2018 Mike DeAngelo Looker Data Sciences, Inc.
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+# frozen_string_literal: true
+
+require_relative '../../command'
+require 'json'
+
+module Gzr
+  module Commands
+    class Query
+      class RunQuery < Gzr::Command
+        def initialize(query_def,options)
+          super()
+          @query_def = query_def
+          @options = options
+        end
+
+        def execute(input: $stdin, output: $stdout)
+          unless @query_def
+            raise Gzr::CLI::Error, "No query specified. A query id, query slug, or a query in json formation must be specified."
+          end
+
+          case @options[:format]
+          when 'png','jpg','xlsx'
+            raise Gzr::CLI::Error, "Output file must be specified with '--file=filename' when using '--format=#{@options[:format]}" unless @options[:file]
+          when 'json', 'json_detail', 'csv', 'txt', 'html', 'md', 'sql'
+            # these formats can be output to stdout
+          else
+            raise Gzr::CLI::Error, "'--format=#{@options[:format]}' not understood. The format must be one of json,json_detail,csv,txt,html,md,xlsx,sql,png,jpg"
+          end
+
+          case @query_def
+          when /^[0-9]+$/
+            query_id = @query_def.to_i
+          when /^[A-Za-z0-9]+$/
+            query_slug = @query_def
+          else
+            begin
+              query_hash = JSON.parse(@query_def,{:symbolize_names => true})
+            rescue JSON::ParserError => e
+              raise Gzr::CLI::Error, "The query specification is not a valid id, slug, or json document"
+            end
+          end
+
+          f = File.open(@options[:file], "w") if @options[:file]
+          with_session do
+            if query_id || query_slug then
+              begin
+                @sdk.query(query_id)
+              rescue LookerSDK::NotFound => e
+                raise Gzr::CLI::Error, "Query with the id #{query_id} not found"
+              end if query_id
+
+              begin
+                query_id = @sdk.query_for_slug(query_slug, { :fields => 'id' }).id.to_i
+              rescue LookerSDK::NotFound => e
+                raise Gzr::CLI::Error, "Query with the slug #{query_slug} not found"
+              end if query_slug
+
+              begin
+                @sdk.run_query(query_id,@options[:format]) { |data,progress| (f || output).write(data) }
+              rescue LookerSDK::Error => e
+                say_error "Error in run_query(#{query_id},#{@options[:format]})})"
+                say_error e.message
+                raise
+              end
+            else
+              begin
+                @sdk.run_inline_query(@options[:format],query_hash) { |data,progress| (f || output).write(data) } 
+              rescue LookerSDK::Error => e
+                say_error "Error in run_inline_query(#{@options[:format]},#{JSON.pretty_generate(query_hash)})"
+                say_error e.message
+                raise
+              end
+            end
+          end
+        ensure
+          f.close if f
+        end
+      end
+    end
+  end
+end

data/lib/gzr/commands/role.rb

@@ -0,0 +1,163 @@
+# The MIT License (MIT)
+
+# Copyright (c) 2018 Mike DeAngelo Looker Data Sciences, Inc.
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+# frozen_string_literal: true
+
+require 'thor'
+
+module Gzr
+  module Commands
+    class Role < Thor
+
+      namespace :role
+
+      desc 'group_rm ROLE_ID GROUP_ID GROUP_ID GROUP_ID ...', 'Remove indicated groups from role'
+      method_option :help, aliases: '-h', type: :boolean,
+                           desc: 'Display usage information'
+      def group_rm(role_id, *groups)
+        if options[:help]
+          invoke :help, ['group_rm']
+        else
+          require_relative 'role/group_rm'
+          Gzr::Commands::Role::GroupRm.new(role_id, groups, options).execute
+        end
+      end
+
+      desc 'user_rm ROLE_ID USER_ID USER_ID USER_ID ...', 'Remove indicated users from role'
+      method_option :help, aliases: '-h', type: :boolean,
+                           desc: 'Display usage information'
+      def user_rm(role_id,*users)
+        if options[:help]
+          invoke :help, ['user_rm']
+        else
+          require_relative 'role/user_rm'
+          Gzr::Commands::Role::UserRm.new(role_id,users,options).execute
+        end
+      end
+
+      desc 'group_add ROLE_ID GROUP_ID GROUP_ID GROUP_ID ...', 'Add indicated groups to role'
+      method_option :help, aliases: '-h', type: :boolean,
+                           desc: 'Display usage information'
+      def group_add(role_id,*groups)
+        if options[:help]
+          invoke :help, ['group_add']
+        else
+          require_relative 'role/group_add'
+          Gzr::Commands::Role::GroupAdd.new(role_id, groups, options).execute
+        end
+      end
+
+      desc 'user_add ROLE_ID USER_ID USER_ID USER_ID ...', 'Add indicated users to role'
+      method_option :help, aliases: '-h', type: :boolean,
+                           desc: 'Display usage information'
+      def user_add(role_id,*users)
+        if options[:help]
+          invoke :help, ['user_add']
+        else
+          require_relative 'role/user_add'
+          Gzr::Commands::Role::UserAdd.new(role_id,users,options).execute
+        end
+      end
+
+      desc 'group_ls ROLE_ID', 'List the groups assigned to a role'
+      method_option :help, aliases: '-h', type: :boolean,
+                           desc: 'Display usage information'
+      method_option :fields, type: :string, default: 'id,name,external_group_id',
+                           desc: 'Fields to display'
+      method_option :plain, type: :boolean, default: false,
+                           desc: 'print without any extra formatting'
+      method_option :csv, type: :boolean, default: false,
+                           desc: 'output in csv format per RFC4180'
+      def group_ls(role_id)
+        if options[:help]
+          invoke :help, ['group_ls']
+        else
+          require_relative 'role/group_ls'
+          Gzr::Commands::Role::GroupLs.new(role_id,options).execute
+        end
+      end
+
+      desc 'user_ls ROLE_ID', 'List the users assigned to a role'
+      method_option :help, aliases: '-h', type: :boolean,
+                           desc: 'Display usage information'
+      method_option :fields, type: :string, default: 'id,first_name,last_name,email',
+                           desc: 'Fields to display'
+      method_option :plain, type: :boolean, default: false,
+                           desc: 'print without any extra formatting'
+      method_option :csv, type: :boolean, default: false,
+                           desc: 'output in csv format per RFC4180'
+      method_option :all_users, type: :boolean, default: false,
+                           desc: 'Show users with this role through a group membership'
+      def user_ls(role_id)
+        if options[:help]
+          invoke :help, ['user_ls']
+        else
+          require_relative 'role/user_ls'
+          Gzr::Commands::Role::UserLs.new(role_id,options).execute
+        end
+      end
+
+      desc 'rm ROLE_ID', 'Delete a role'
+      method_option :help, aliases: '-h', type: :boolean,
+                           desc: 'Display usage information'
+      def rm(role_id)
+        if options[:help]
+          invoke :help, ['rm']
+        else
+          require_relative 'role/rm'
+          Gzr::Commands::Role::Rm.new(role_id,options).execute
+        end
+      end
+
+      desc 'cat ROLE_ID', 'Output the JSON representation of a role to screen/file'
+      method_option :help, aliases: '-h', type: :boolean,
+                           desc: 'Display usage information'
+      method_option :dir,  type: :string,
+                           desc: 'Directory to get output file'
+      def cat(role_id)
+        if options[:help]
+          invoke :help, ['cat']
+        else
+          require_relative 'role/cat'
+          Gzr::Commands::Role::Cat.new(role_id,options).execute
+        end
+      end
+
+      desc 'ls', 'Display all roles'
+      method_option :help, aliases: '-h', type: :boolean,
+                           desc: 'Display usage information'
+      method_option :fields, type: :string, default: 'id,name,permission_set(id,name,permissions),model_set(id,name,models)',
+                           desc: 'Fields to display'
+      method_option :plain, type: :boolean, default: false,
+                           desc: 'print without any extra formatting'
+      method_option :csv, type: :boolean, default: false,
+                           desc: 'output in csv format per RFC4180'
+      def ls(*)
+        if options[:help]
+          invoke :help, ['ls']
+        else
+          require_relative 'role/ls'
+          Gzr::Commands::Role::Ls.new(options).execute
+        end
+      end
+    end
+  end
+end

data/lib/gzr/commands/role/cat.rb

@@ -0,0 +1,52 @@
+# The MIT License (MIT)
+
+# Copyright (c) 2018 Mike DeAngelo Looker Data Sciences, Inc.
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+# frozen_string_literal: true
+
+require_relative '../../command'
+require_relative '../../modules/role'
+require_relative '../../modules/filehelper'
+
+module Gzr
+  module Commands
+    class Role
+      class Cat < Gzr::Command
+        include Gzr::Role
+        include Gzr::FileHelper
+        def initialize(role_id,options)
+          super()
+          @role_id = role_id
+          @options = options
+        end
+
+        def execute(input: $stdin, output: $stdout)
+          say_warning("options: #{@options.inspect}") if @options[:debug]
+          with_session do
+            data = query_role(@role_id)
+            write_file(@options[:dir] ? "Role_#{data.id}_#{data.name}.json" : nil, @options[:dir], nil, output) do |f|
+              f.puts JSON.pretty_generate(data.to_attrs)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/gzr/commands/role/group_add.rb

@@ -0,0 +1,51 @@
+# The MIT License (MIT)
+
+# Copyright (c) 2018 Mike DeAngelo Looker Data Sciences, Inc.
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+# frozen_string_literal: true
+
+require_relative '../../command'
+require_relative '../../modules/role'
+
+module Gzr
+  module Commands
+    class Role
+      class GroupAdd < Gzr::Command
+        include Gzr::Role
+        def initialize(role_id,groups,options)
+          super()
+          @role_id = role_id
+          @groups = groups.collect { |g| g.to_i }
+          @options = options
+        end
+
+        def execute(input: $stdin, output: $stdout)
+          say_warning(@options) if @options[:debug]
+          
+          with_session do
+            groups = query_role_groups(@role_id, 'id').collect { |g| g.id }
+            groups += @groups
+            set_role_groups(@role_id,groups.uniq)
+          end
+        end
+      end
+    end
+  end
+end

data/lib/gzr/commands/role/group_ls.rb

@@ -0,0 +1,76 @@
+# The MIT License (MIT)
+
+# Copyright (c) 2018 Mike DeAngelo Looker Data Sciences, Inc.
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+# frozen_string_literal: true
+
+require_relative '../../command'
+require_relative '../../modules/role'
+require 'tty-table'
+
+module Gzr
+  module Commands
+    class Role
+      class GroupLs < Gzr::Command
+        include Gzr::Role
+        def initialize(role_id,options)
+          super()
+          @role_id = role_id
+          @options = options
+        end
+
+        def execute(input: $stdin, output: $stdout)
+          say_warning(@options) if @options[:debug]
+          with_session do
+            data = query_role_groups(@role_id,@options[:fields])
+            begin
+              say_ok "No role groups found"
+              return nil
+            end unless data && data.length > 0
+
+            table_hash = Hash.new
+            fields = field_names(@options[:fields])
+            table_hash[:header] = fields unless @options[:plain]
+            expressions = fields.collect { |fn| field_expression(fn) }
+            table_hash[:rows] = data.map do |row|
+              expressions.collect do |e|
+                v = eval "row.#{e}"
+                next (v.join "\n") if v.kind_of? Array
+                v
+              end
+            end
+            table = TTY::Table.new(table_hash)
+            alignments = fields.collect do |k|
+              next :left if k == "external_group_id"
+              (k =~ /(id|count)$/) ? :right : :left
+            end
+            begin
+              if @options[:csv] then
+                output.puts render_csv(table)
+              else
+                output.puts table.render(if @options[:plain] then :basic else :ascii end, multiline: true, alignments: alignments)
+              end
+            end if table
+          end
+        end
+      end
+    end
+  end
+end

data/lib/gzr/commands/role/group_rm.rb

@@ -0,0 +1,51 @@
+# The MIT License (MIT)
+
+# Copyright (c) 2018 Mike DeAngelo Looker Data Sciences, Inc.
+
+# Permission is hereby granted, free of charge, to any person obtaining a copy of
+# this software and associated documentation files (the "Software"), to deal in
+# the Software without restriction, including without limitation the rights to
+# use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
+# the Software, and to permit persons to whom the Software is furnished to do so,
+# subject to the following conditions:
+
+# The above copyright notice and this permission notice shall be included in all
+# copies or substantial portions of the Software.
+
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
+# FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
+# COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
+# IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+# CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+
+# frozen_string_literal: true
+
+require_relative '../../command'
+require_relative '../../modules/role'
+
+module Gzr
+  module Commands
+    class Role
+      class GroupRm < Gzr::Command
+        include Gzr::Role
+        def initialize(role_id,groups,options)
+          super()
+          @role_id = role_id
+          @groups = groups.collect { |g| g.to_i }
+          @options = options
+        end
+
+        def execute(input: $stdin, output: $stdout)
+          say_warning(@options) if @options[:debug]
+          
+          with_session do
+            groups = query_role_groups(@role_id, 'id').collect { |g| g.id }
+            groups -= @groups
+            set_role_groups(@role_id,groups.uniq)
+          end
+        end
+      end
+    end
+  end
+end