datahen 0.14.2 → 0.14.10

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 10e4e646ff403fbdb91e0be43e0a78a879c48f3686be251f26185cae961e4b4f
-  data.tar.gz: bc8a4f11cb3929e2cf1655d7463f02c6dd9166a1994a7cbdb69cddf327099f95
+  metadata.gz: 68b2f97e2b71ab8e79edacc9fa0b719c1d4b93bac684631ab084e46f44407795
+  data.tar.gz: 2e4a453502ac9a72c7cbea81b3f36100373adf9894e5c2ca4250cb1fa709c548
 SHA512:
-  metadata.gz: 7d57fc498d3f3b6bcf08a31a663a48f5ace6fff5f9fdeeb1b8f396710e6d7e96f31b1a0dc460aeab312d468d19098b759cc19d6c60ecee9286a409934cfd5872
-  data.tar.gz: ff1585a9b3be081898c597d5e696f0927b90f54d420de3bab39451b16dbd55d622a1aa7c9b631dac22cc9f84d1c5e60b86bf5284bde1cb0abb0a24ed69a5834e
+  metadata.gz: e73e87d091b590fd1efc4c31d5589b930e0fae458a8b8e5439c47dfae008c1db9186672f49ddcf41e7a0c32e31398b1703b45f1cde81a3a4e397c8f16008a4ea
+  data.tar.gz: 52fd87fe09082f7ebd3a6e96ce0fc699b8a474ceb867a2187f21fb54431ac91f4cffdaf3a0ede51feb2a18fa9085a13a00048ad83a362d63399f2c4b87739c78

data/lib/datahen/cli.rb

@@ -16,6 +16,7 @@ require 'datahen/cli/parser'
 require 'datahen/cli/seeder'
 require 'datahen/cli/finisher'
 require 'datahen/cli/env_var'
+require 'datahen/cli/account'
 
 
 
@@ -41,5 +42,8 @@ module Datahen
 
     desc "var SUBCOMMAND ...ARGS", "for environment variable related activities"
     subcommand "var", EnvVar
+
+    desc "account SUBCOMMAND ...ARGS", "for account related activities"
+    subcommand "account", Account
   end
 end

data/lib/datahen/cli/account.rb

@@ -0,0 +1,17 @@
+module Datahen
+  class CLI < Thor
+    class Account < Thor
+
+      desc "profile", "displays the account applied profile"
+      long_desc <<-LONGDESC
+        Displays the account applied profile
+      LONGDESC
+      def profile()
+        client = Client::Account.new(options)
+        puts "#{client.profile()}"
+      end
+
+    end
+  end
+
+end

data/lib/datahen/cli/env_var.rb

@@ -2,7 +2,7 @@ module Datahen
   class CLI < Thor
     class EnvVar < Thor
       desc "list", "List environment variables on the account"
-      
+
       long_desc <<-LONGDESC
         List all environment variables on the account.
       LONGDESC
@@ -19,7 +19,7 @@ module Datahen
           <name>: Var name can only consist of alphabets, numbers, underscores. Name must be unique to your account, otherwise it will be overwritten.\x5
           <value>: Value of variable.\x5
           LONGDESC
-      option :secret, type: :boolean, desc: 'Set true to make it decrypt the value. Default: false' 
+      option :secret, type: :boolean, desc: 'Set true to make it decrypt the value. Default: false'
       def set(name, value)
         # puts "options #{options}"
         client = Client::EnvVar.new(options)
@@ -38,10 +38,6 @@ module Datahen
         puts "#{client.unset(name)}"
       end
 
-
-      
-
-
     end
   end
 

data/lib/datahen/cli/job.rb

@@ -10,6 +10,7 @@ module Datahen
       desc "list", "gets a list of jobs"
       option :page, :aliases => :p, type: :numeric, desc: 'Get the next set of records by page.'
       option :per_page, :aliases => :P, type: :numeric, desc: 'Number of records per page. Max 500 per page.'
+      option :status, type: :string, desc: 'Returns jobs with a certain status'
       long_desc <<-LONGDESC
         List scrape jobs.
       LONGDESC

data/lib/datahen/cli/scraper.rb

@@ -1,8 +1,8 @@
 module Datahen
   class CLI < Thor
     class Scraper < Thor
-      desc "list", "List scrapers"
 
+      desc "list", "List scrapers"
       long_desc <<-LONGDESC
         List all scrapers.
       LONGDESC
@@ -29,6 +29,7 @@ module Datahen
       option :cancel_current_job, type: :boolean, desc: 'Set true to cancel currently active job if scheduler starts. Default: false'
       option :schedule, type: :string, desc: 'Set the schedule of the scraper to run. Must be in CRON format.'
       option :timezone, type: :string, desc: "Set the scheduler's timezone. Must be in IANA Timezone format. Defaults to \"America/Toronto\""
+      option :profile, type: :string, desc: 'Set the profiles (comma separated) to apply to the job. Default: default'
       def create(scraper_name, git_repository)
         # puts "options #{options}"
         client = Client::Scraper.new(options)
@@ -51,6 +52,7 @@ module Datahen
       option :cancel_current_job, type: :boolean, desc: 'Set true to cancel currently active job if scheduler starts. Default: false'
       option :schedule, type: :string, desc: 'Set the schedule of the scraper to run. Must be in CRON format.'
       option :timezone, type: :string, desc: "Set the scheduler's timezone. Must be in IANA Timezone format. Defaults to \"America/Toronto\""
+      option :profile, type: :string, desc: 'Set the profiles (comma separated) to apply to the job. Default: default'
       def update(scraper_name)
         client = Client::Scraper.new(options)
         puts "#{client.update(scraper_name, options)}"
@@ -164,15 +166,25 @@ module Datahen
       option :"max-timestamp", type: :string, desc: 'Ending timestamp point in time to query historic stats (inclusive)'
       option :"limit", type: :numeric, desc: 'Limit stats retrieved'
       option :"order", type: :numeric, desc: 'Order stats by timestamp [DESC]'
+      option :live, type: :boolean, desc: 'Get data from the live stats history, not cached stats history.'
+      option :filter, type: :string, desc: 'Filter results on `day` or `hour`, if not specified will return all records.'
       def history(scraper_name)
         client = Client::JobStat.new(options)
         if options[:job]
-          puts "#{client.job_stats_history(options[:job])}"
+          puts "#{client.job_stats_history(options[:job], options)}"
         else
-          puts "#{client.scraper_job_stats_history(scraper_name)}"
+          puts "#{client.scraper_job_stats_history(scraper_name, options)}"
         end
       end
 
+      desc "profile <scraper_name>", "displays the scraper applied profile"
+      long_desc <<-LONGDESC
+        Displays the account applied profile
+      LONGDESC
+      def profile(scraper_name)
+        client = Client::Scraper.new(options)
+        puts "#{client.profile(scraper_name)}"
+      end
 
       desc "job SUBCOMMAND ...ARGS", "manage scrapers jobs"
       subcommand "job", ScraperJob

data/lib/datahen/cli/scraper_job.rb

@@ -26,6 +26,7 @@ module Datahen
       LONGDESC
       option :page, :aliases => :p, type: :numeric, desc: 'Get the next set of records by page.'
       option :per_page, :aliases => :P, type: :numeric, desc: 'Number of records per page. Max 500 per page.'
+      option :status, type: :string, desc: 'Returns jobs with a certain status'
       def list(scraper_name)
         client = Client::ScraperJob.new(options)
         puts "#{client.all(scraper_name)}"
@@ -64,7 +65,7 @@ module Datahen
 
       desc "pause <scraper_name>", "pauses a scraper's current job"
       long_desc <<-LONGDESC
-        pauses a scraper's current job
+        Pauses a scraper's current job
       LONGDESC
       option :job, :aliases => :j, type: :numeric, desc: 'Set a specific job ID'
       def pause(scraper_name)
@@ -85,6 +86,7 @@ module Datahen
       option :workers, :aliases => :w, type: :numeric, desc: 'Set how many standard workers to use. Scraper job must be restarted(paused then resumed, or cancelled then resumed) for it to take effect. Default: 1. '
       option :browsers, type: :numeric, desc: 'Set how many browser workers to use. Scraper job must be restarted(paused then resumed, or cancelled then resumed) for it to take effect. Default: 0. '
       option :proxy_type, desc: 'Set the Proxy type. Default: standard'
+      option :profile, type: :string, desc: 'Set the profiles (comma separated) to apply to the job. Default: default'
       option :job, :aliases => :j, type: :numeric, desc: 'Set a specific job ID'
       def update(scraper_name)
         if options[:job]
@@ -96,6 +98,21 @@ module Datahen
         end
       end
 
+      desc "profile <scraper_name>", "displays a scraper's current job applied profile"
+      long_desc <<-LONGDESC
+        Displays a scraper's current job applied profile
+      LONGDESC
+      option :job, :aliases => :j, type: :numeric, desc: 'Set a specific job ID'
+      def profile(scraper_name)
+        if options[:job]
+          client = Client::Job.new(options)
+          puts "#{client.profile(options[:job])}"
+        else
+          client = Client::ScraperJob.new(options)
+          puts "#{client.profile(scraper_name)}"
+        end
+      end
+
       desc "var SUBCOMMAND ...ARGS", "for managing scraper's job variables"
       subcommand "var", ScraperJobVar
 

data/lib/datahen/client.rb

@@ -1,4 +1,5 @@
 require "datahen/client/base"
+require "datahen/client/account"
 require "datahen/client/auth_token"
 require "datahen/client/deploy_key"
 require 'datahen/client/export'
@@ -24,7 +25,6 @@ require "datahen/client/job_var"
 require "datahen/client/scraper_job_var"
 require "datahen/client/job_finisher"
 
-
 module Datahen
   module Client
   end

data/lib/datahen/client/account.rb

@@ -0,0 +1,13 @@
+module Datahen
+  module Client
+    class Account < Datahen::Client::Base
+
+      def profile(opts={})
+        params = @options.merge(opts)
+
+        self.class.get("/profile", params)
+      end
+
+    end
+  end
+end

data/lib/datahen/client/base.rb

@@ -5,6 +5,8 @@ module Datahen
     class Base
       include HTTParty
 
+      default_timeout 60
+
       def self.env_auth_token
         ENV['DATAHEN_TOKEN']
       end
@@ -55,6 +57,7 @@ module Datahen
         query[:"max-timestamp"] = opts[:"max-timestamp"] if opts[:"max-timestamp"]
         query[:limit] = opts[:limit] if opts[:limit]
         query[:order] = opts[:order] if opts[:order]
+        query[:filter] = opts[:filter] if opts[:filter]
 
         if opts[:query]
           if opts[:query].is_a?(Hash)

data/lib/datahen/client/job.rb

@@ -20,6 +20,7 @@ module Datahen
         body[:standard_worker_count] = opts[:workers] if opts[:workers]
         body[:browser_worker_count] = opts[:browsers] if opts[:browsers]
         body[:proxy_type] = opts[:proxy_type] if opts[:proxy_type]
+        body[:profile] = opts[:profile] if opts[:profile]
         params = @options.merge({body: body.to_json})
 
         self.class.put("/jobs/#{job_id}", params)
@@ -64,6 +65,12 @@ module Datahen
         self.class.put("/jobs/#{job_id}/finisher_update", params)
       end
 
+      def profile(job_id, opts={})
+        params = @options.merge(opts)
+
+        self.class.get("/jobs/#{job_id}/profile", params)
+      end
+
     end
 
   end

data/lib/datahen/client/job_stat.rb

@@ -18,12 +18,20 @@ module Datahen
         end
       end
 
-      def job_stats_history(job_id)
-        self.class.get("/jobs/#{job_id}/stats/history", @options)
+      def job_stats_history(job_id, opts={})
+        if opts[:live]
+          self.class.get("/jobs/#{job_id}/stats/history", @options)
+        else
+          self.class.get("/cached/jobs/#{job_id}/stats/history", @options)
+        end
       end
 
-      def scraper_job_stats_history(scraper_name)
-        self.class.get("/scrapers/#{scraper_name}/current_job/stats/history", @options)
+      def scraper_job_stats_history(scraper_name, opts={})
+        if opts[:live]
+          self.class.get("/scrapers/#{scraper_name}/current_job/stats/history", @options)
+        else
+          self.class.get("/cached/scrapers/#{scraper_name}/current_job/stats/history", @options)
+        end
       end
 
     end

data/lib/datahen/client/scraper.rb

@@ -25,6 +25,7 @@ module Datahen
         body[:cancel_current_job] = opts[:cancel_current_job] if opts[:cancel_current_job]
         body[:schedule] = opts[:schedule] if opts[:schedule]
         body[:timezone] = opts[:timezone] if opts[:timezone]
+        body[:profile] = opts[:profile] if opts[:profile]
         params = @options.merge({body: body.to_json})
         self.class.post("/scrapers", params)
       end
@@ -43,6 +44,7 @@ module Datahen
         body[:cancel_current_job] = opts[:cancel_current_job] if opts.has_key?("cancel_current_job") || opts.has_key?(:cancel_current_job)
         body[:schedule] = opts[:schedule] if opts[:schedule]
         body[:timezone] = opts[:timezone] if opts[:timezone]
+        body[:profile] = opts[:profile] if opts[:profile]
         params = @options.merge({body: body.to_json})
 
         self.class.put("/scrapers/#{scraper_name}", params)
@@ -52,6 +54,12 @@ module Datahen
         params = @options.merge(opts)
         self.class.delete("/scrapers/#{scraper_name}", params)
       end
+
+      def profile(scraper_name, opts={})
+        params = @options.merge(opts)
+
+        self.class.get("/scrapers/#{scraper_name}/profile", params)
+      end
     end
   end
 end

data/lib/datahen/client/scraper_job.rb

@@ -29,6 +29,7 @@ module Datahen
         body[:standard_worker_count] = opts[:workers] if opts[:workers]
         body[:browser_worker_count] = opts[:browsers] if opts[:browsers]
         body[:proxy_type] = opts[:proxy_type] if opts[:proxy_type]
+        body[:profile] = opts[:profile] if opts[:profile]
         params = @options.merge({body: body.to_json})
 
         self.class.put("/scrapers/#{scraper_name}/current_job", params)
@@ -48,6 +49,12 @@ module Datahen
         opts[:status] = 'paused'
         update(scraper_name, opts)
       end
+
+      def profile(scraper_name, opts={})
+        params = @options.merge(opts)
+
+        self.class.get("/scrapers/#{scraper_name}/current_job/profile", params)
+      end
     end
   end
 end

data/lib/datahen/version.rb

@@ -1,3 +1,3 @@
 module Datahen
-  VERSION = "0.14.2"
+  VERSION = "0.14.10"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: datahen
 version: !ruby/object:Gem::Version
-  version: 0.14.2
+  version: 0.14.10
 platform: ruby
 authors:
 - Parama Danoesubroto
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2020-07-28 00:00:00.000000000 Z
+date: 2020-08-10 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: thor
@@ -189,6 +189,7 @@ files:
 - exe/hen
 - lib/datahen.rb
 - lib/datahen/cli.rb
+- lib/datahen/cli/account.rb
 - lib/datahen/cli/env_var.rb
 - lib/datahen/cli/finisher.rb
 - lib/datahen/cli/global_page.rb
@@ -206,6 +207,7 @@ files:
 - lib/datahen/cli/scraper_var.rb
 - lib/datahen/cli/seeder.rb
 - lib/datahen/client.rb
+- lib/datahen/client/account.rb
 - lib/datahen/client/auth_token.rb
 - lib/datahen/client/backblaze_content.rb
 - lib/datahen/client/base.rb
@@ -264,7 +266,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
+rubygems_version: 3.1.2
 signing_key: 
 specification_version: 4
 summary: DataHen toolbelt for developers