cnvrg 1.5.6 → 1.5.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: cfc351ff0936f6dd53e58da8ced925f7000c68d2e5ffa3c687d15fc090faafc1
-  data.tar.gz: 2e32e0decc29ef4b41a08c82c74fdf8b197c8844d1d49f98efcb81284977b967
+  metadata.gz: 773594ee55ab2260bc979dea185389b81838ebe9fe735ca8f632ddd3e81e76d9
+  data.tar.gz: 31bf3acaccd8064830b7323ad3eb8501a3a9c831c4942b10e8c121904d692ae3
 SHA512:
-  metadata.gz: 71928a5eba809f59237be1ff75e58df1ac8b41203705278cf110b96acade3ab04c5073135a1e0eae3c58621d896d13f47fc2be9099a715be0d3e1f1c4a6e7d5f
-  data.tar.gz: 563db133a90bccdca1ca4eec07cf8dc7356a551647ebfd19a9936752ea65097386b8c20f429f690c75525d83a72130b14aa1c5e023961617672e1d4c1fb05ad2
+  metadata.gz: 61332fdcfd42e7d2f43dbc80c9d186f2e7bfe287c960998f4a8f55d7bf0b2705aff6cfca8737a8989ea86d38683f04498975ab7f24fbe1b60c940d1df656c94f
+  data.tar.gz: 93c4069fff847e2308e34b435fe4adbfa66a61b6487c64f83d400210831c7afb037d1e71241bb8f26da3f8397c20e43fc9465f42dc6a32131cef117f441d4ef3

data/lib/cnvrg/cli.rb

@@ -870,6 +870,7 @@ module Cnvrg
         if query.present?
           return clone_data_query(dataset_url, query)
         end
+        @executer = Cnvrg::Helpers::Executer.get_executer
 
         url_parts = dataset_url.split("/")
         project_index = Cnvrg::Helpers.look_for_in_path(dataset_url, "datasets")
@@ -920,6 +921,7 @@ module Cnvrg
           progressbar.finish
           if downloaded_files == files_count
             Dataset.verify_cnvrgignore_exist(dataset_name, false)
+            @executer.set_dataset_status(dataset: @dataset.slug, status: "cloned") if @executer
             if !read
               @dataset.write_idx(nil, commit) #nil means, generate idx
             end
@@ -945,6 +947,7 @@ module Cnvrg
     def clone_data_query(dataset_url,query=nil)
       begin
         verify_logged_in(false)
+        @executer = Cnvrg::Helpers::Executer.get_executer
         log_start(__method__, args, options)
         query = options["query"] || query
         if !query.present?
@@ -1000,6 +1003,7 @@ module Cnvrg
               end
 
             end
+            @executer.set_dataset_status(dataset: @dataset.slug, status: "cloned") if @executer.present?
           rescue Interrupt
             log_message("Couldn't download", Thor::Shell::Color::RED)
             exit(1)

data/lib/cnvrg/helpers/executer.rb

@@ -1,14 +1,22 @@
 class Cnvrg::Helpers::Executer
   def initialize(project: nil, job_type: nil, job_id: nil)
-    @project = project
-    @job_type = job_type
-    @job_id = job_id
-    if job_id.blank?
+    @project = project || Cnvrg::Project.new(owner: ENV['CNVRG_OWNER'], slug: ENV['CNVRG_PROJECT'])
+    @job_type = job_type || ENV['CNVRG_JOB_TYPE']
+    @job_id = job_id || ENV['CNVRG_JOB_ID']
+    if @job_id.blank?
       Cnvrg::CLI.log_message("Cant find job, exiting.", 'red')
       exit(1)
     end
   end
 
+
+  def self.get_executer
+    if ENV["CNVRG_JOB_ID"].blank?
+      return
+    end
+    self.new
+  end
+
   def fetch_commands(block: nil, key: nil)
     resp = Cnvrg::API.request("#{base_url}/commands", "GET", {block: block})
     commands = decrypt_commands(resp["commands"], resp["key"], resp["iv"])
@@ -73,6 +81,10 @@ class Cnvrg::Helpers::Executer
     commands.map{|k| k.with_indifferent_access}
   end
 
+  def set_dataset_status(dataset: nil, status: nil)
+    Cnvrg::API.request("#{base_url}/datasets/#{dataset}", "PUT", {status: status})
+  end
+
   private
   def execute_helper(cmd)
     case cmd[:type]

data/lib/cnvrg/version.rb

@@ -1,4 +1,4 @@
 module Cnvrg
-  VERSION = '1.5.6'
+  VERSION = '1.5.7'
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cnvrg
 version: !ruby/object:Gem::Version
-  version: 1.5.6
+  version: 1.5.7
 platform: ruby
 authors:
 - Yochay Ettun
@@ -10,7 +10,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2019-08-14 00:00:00.000000000 Z
+date: 2019-08-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -432,8 +432,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubyforge_project: 
-rubygems_version: 2.7.9
+rubygems_version: 3.0.3
 signing_key: 
 specification_version: 4
 summary: A CLI tool for interacting with cnvrg.io.