Class: Cnvrg::Dataset
- Inherits:
-
Object
- Object
- Cnvrg::Dataset
- Defined in:
- lib/cnvrg/dataset.rb
Instance Attribute Summary collapse
-
#local_path ⇒ Object
readonly
Returns the value of attribute local_path.
-
#owner ⇒ Object
readonly
Returns the value of attribute owner.
-
#slug ⇒ Object
readonly
Returns the value of attribute slug.
-
#title ⇒ Object
readonly
Returns the value of attribute title.
-
#working_dir ⇒ Object
readonly
Returns the value of attribute working_dir.
Class Method Summary collapse
- .clone(owner, dataset_name, dataset_slug) ⇒ Object
- .init(owner, dataset_name, is_public = false) ⇒ Object
- .init_container(owner, dataset_slug, dataset_name) ⇒ Object
Instance Method Summary collapse
- #compare_commit(commit) ⇒ Object
- #compare_commits(commit) ⇒ Object
- #compare_idx(new_branch, commit = last_local_commit, local_idx = nil) ⇒ Object
- #create_volume ⇒ Object
- #downlowd_updated_data(current_commit) ⇒ Object
- #generate_idx ⇒ Object
- #get_idx ⇒ Object
- #get_ignore_list ⇒ Object
-
#initialize(project_home) ⇒ Dataset
constructor
A new instance of Dataset.
- #last_local_commit ⇒ Object
- #list(owner) ⇒ Object
- #list_commits ⇒ Object
- #revert(working_dir) ⇒ Object
- #snapshot ⇒ Object
- #update_idx(idx) ⇒ Object
- #update_idx_with_commit!(commit) ⇒ Object
- #update_idx_with_files_commits!(files, commit_time) ⇒ Object
- #update_ignore_list(new_ignore) ⇒ Object
- #url ⇒ Object
Constructor Details
#initialize(project_home) ⇒ Dataset
Returns a new instance of Dataset.
8 9 10 11 12 13 14 15 16 17 18 19 20 |
# File 'lib/cnvrg/dataset.rb', line 8 def initialize(project_home) begin config = YAML.load_file(project_home+"/.cnvrg/config.yml") @local_path = project_home @title = config[:dataset_name] @slug = config[:dataset_slug] @owner = config[:owner] @working_dir = project_home rescue => e end end |
Instance Attribute Details
#local_path ⇒ Object (readonly)
Returns the value of attribute local_path.
4 5 6 |
# File 'lib/cnvrg/dataset.rb', line 4 def local_path @local_path end |
#owner ⇒ Object (readonly)
Returns the value of attribute owner.
4 5 6 |
# File 'lib/cnvrg/dataset.rb', line 4 def owner @owner end |
#slug ⇒ Object (readonly)
Returns the value of attribute slug.
4 5 6 |
# File 'lib/cnvrg/dataset.rb', line 4 def slug @slug end |
#title ⇒ Object (readonly)
Returns the value of attribute title.
4 5 6 |
# File 'lib/cnvrg/dataset.rb', line 4 def title @title end |
#working_dir ⇒ Object (readonly)
Returns the value of attribute working_dir.
4 5 6 |
# File 'lib/cnvrg/dataset.rb', line 4 def working_dir @working_dir end |
Class Method Details
.clone(owner, dataset_name, dataset_slug) ⇒ Object
121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 |
# File 'lib/cnvrg/dataset.rb', line 121 def self.clone(owner, dataset_name,dataset_slug) begin list_dirs = [ dataset_name, "#{dataset_name}/.cnvrg" ] list_files = [ "#{dataset_name}/.cnvrgignore", "#{dataset_name}/.cnvrg/config.yml" ] config = {dataset_name: dataset_name, dataset_slug: dataset_slug, owner: owner} cnvrgignore = Helpers.cnvrgignore_content FileUtils.mkdir_p list_dirs FileUtils.touch list_files File.open("#{dataset_name}/.cnvrg/config.yml", "w+") { |f| f.write config.to_yaml } File.open("#{dataset_name}/.cnvrgignore", "w+") { |f| f.write cnvrgignore } unless File.exist? ".cnvrgignore" rescue => e puts "Exception in clone request:#{e.}" return false end return true end |
.init(owner, dataset_name, is_public = false) ⇒ Object
94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 |
# File 'lib/cnvrg/dataset.rb', line 94 def self.init(owner, dataset_name, is_public=false) list_dirs = [".cnvrg" ] list_files = [ ".cnvrgignore", ".cnvrg/config.yml" ] cnvrgignore = Helpers.cnvrgignore_content begin response = Cnvrg::API.request("cli/create_dataset", 'POST', {title: dataset_name, owner: owner, is_public: is_public}) Cnvrg::CLI.is_response_success(response) response = JSON.parse response["result"] dataset_slug = response["slug"] config = {dataset_name: dataset_name, dataset_slug: dataset_slug, owner: owner} FileUtils.mkdir_p list_dirs FileUtils.touch list_files File.open(".cnvrg/config.yml", "w+") { |f| f.write config.to_yaml } File.open(".cnvrgignore", "w+") { |f| f.write cnvrgignore } unless File.exist? ".cnvrgignore" rescue => e return false end return true end |
.init_container(owner, dataset_slug, dataset_name) ⇒ Object
148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 |
# File 'lib/cnvrg/dataset.rb', line 148 def self.init_container(owner, dataset_slug,dataset_name) cnvrgignore = Helpers.cnvrgignore_content begin list_dirs = [ ".cnvrg" ] list_files = [ ".cnvrgignore", ".cnvrg/config.yml" ] FileUtils.mkdir_p list_dirs FileUtils.touch list_files config = {dataset_name: dataset_name, dataset_slug: dataset_slug, owner: owner} File.open(".cnvrg/config.yml", "w+") { |f| f.write config.to_yaml } File.open(".cnvrgignore", "w+") { |f| f.write cnvrgignore } unless File.exist? ".cnvrgignore" rescue => e return false end return true end |
Instance Method Details
#compare_commit(commit) ⇒ Object
250 251 252 253 254 255 256 257 258 |
# File 'lib/cnvrg/dataset.rb', line 250 def compare_commit(commit) if commit.nil? or commit.empty? commit = last_local_commit end response = Cnvrg::API.request("users/#{self.owner}/projects/#{self.slug}/commit/compare", 'POST', {current_commit: commit}) CLI.is_response_success(response,false) update_is_new_branch(response["result"]["new_branch"]) return response["result"]["new_branch"] end |
#compare_commits(commit) ⇒ Object
244 245 246 247 248 |
# File 'lib/cnvrg/dataset.rb', line 244 def compare_commits(commit) response = Cnvrg::API.request("users/#{self.owner}/datasets/#{self.slug}/compare_commits", 'POST', {compare_commit: commit,current_commit:last_local_commit}) CLI.is_response_success(response,false) return response end |
#compare_idx(new_branch, commit = last_local_commit, local_idx = nil) ⇒ Object
231 232 233 234 235 236 237 238 239 240 |
# File 'lib/cnvrg/dataset.rb', line 231 def compare_idx(new_branch, commit=last_local_commit,local_idx=nil) if local_idx.nil? local_idx = self.generate_idx end response = Cnvrg::API.request("users/#{self.owner}/datasets/#{self.slug}/status", 'POST', {idx: local_idx, new_branch: new_branch, current_commit: commit}) CLI.is_response_success(response,false) return response end |
#create_volume ⇒ Object
219 220 221 222 223 |
# File 'lib/cnvrg/dataset.rb', line 219 def create_volume response = Cnvrg::API.request("users/#{self.owner}/datasets/#{self.slug}/volumes/create", 'POST') CLI.is_response_success(response) return response end |
#downlowd_updated_data(current_commit) ⇒ Object
225 226 227 228 229 230 |
# File 'lib/cnvrg/dataset.rb', line 225 def downlowd_updated_data(current_commit) response = Cnvrg::API.request("users/#{self.owner}/datasets/#{self.slug}/downlowd_updated_data", 'POST', {current_commit: current_commit}) CLI.is_response_success(response,false) return response end |
#generate_idx ⇒ Object
181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 |
# File 'lib/cnvrg/dataset.rb', line 181 def generate_idx if File.exists? "#{self.local_path}/.cnvrg/idx.yml" old_idx = YAML.load_file("#{self.local_path}/.cnvrg/idx.yml") else old_idx = nil end tree_idx = Hash.new(0) list = Dir.glob("#{self.local_path}/**/**", File::FNM_DOTMATCH).reject { |x| (x =~ /\/\.{1,2}$/) or (x =~ /^#{self.local_path}\/\.cnvrg\/*/) } list_ignore = self.get_ignore_list() list.each do |e| label = e.gsub(self.local_path + "/", "") if File.directory? e if list_ignore.include? label next end tree_idx[label+"/"] = nil else if list_ignore.include? label next end sha1 = Digest::SHA1.file(e).hexdigest if old_idx.nil? or old_idx.to_h["tree"].nil? tree_idx[label] = {sha1: sha1, commit_time: nil} elsif old_idx["tree"][label].nil? or old_idx["tree"][label]["sha1"] != sha1 tree_idx[label] = {sha1: sha1, commit_time: nil} else tree_idx[label] = old_idx["tree"][label] end end end idx = {commit: old_idx.to_h[:commit], tree: tree_idx} File.open("#{self.local_path}/.cnvrg/idx.yml", 'w') { |f| f.write idx.to_yaml } return YAML.load_file("#{self.local_path}/.cnvrg/idx.yml") end |
#get_idx ⇒ Object
174 175 176 |
# File 'lib/cnvrg/dataset.rb', line 174 def get_idx YAML.load_file("#{self.local_path}/.cnvrg/idx.yml") end |
#get_ignore_list ⇒ Object
70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
# File 'lib/cnvrg/dataset.rb', line 70 def get_ignore_list ignore_list = [] File.open(self.local_path+"/.cnvrgignore", "r").each_line do |line| line = line.strip if line.start_with? "#" or ignore_list.include? line or line.empty? next end if line.end_with? "/" or File.directory?(line) ignore_list << line all_sub = Dir.glob("#{line}/**/*", File::FNM_DOTMATCH).flatten ignore_list << all_sub.flatten elsif line.include? "*" regex_list = Dir.glob("**/*#{line}", File::FNM_DOTMATCH).flatten ignore_list << regex_list else ignore_list << line end end return ignore_list.flatten end |
#last_local_commit ⇒ Object
22 23 24 25 |
# File 'lib/cnvrg/dataset.rb', line 22 def last_local_commit idx = YAML.load_file(@local_path + "/.cnvrg/idx.yml") return idx[:commit] end |
#list(owner) ⇒ Object
34 35 36 37 38 39 |
# File 'lib/cnvrg/dataset.rb', line 34 def list(owner) response = Cnvrg::API.request("users/#{owner}/datasets/list", 'GET') CLI.is_response_success(response) return response end |
#list_commits ⇒ Object
40 41 42 43 44 45 |
# File 'lib/cnvrg/dataset.rb', line 40 def list_commits response = Cnvrg::API.request("users/#{self.owner}/datasets/#{self.slug}/list_commits", 'GET') CLI.is_response_success(response) return response end |
#revert(working_dir) ⇒ Object
285 286 287 288 289 290 291 |
# File 'lib/cnvrg/dataset.rb', line 285 def revert(working_dir) FileUtils.rm_rf working_dir # response = Cnvrg::API.request("users/#{self.owner}/projects/#{self.slug}/revert", 'GET') # CLI.is_response_success(response) end |
#snapshot ⇒ Object
27 28 29 30 31 32 33 |
# File 'lib/cnvrg/dataset.rb', line 27 def snapshot commit = last_local_commit response = Cnvrg::API.request("users/#{self.owner}/datasets/#{self.slug}/volumes/create", 'POST', {data_commit: commit}) CLI.is_response_success(response) return response end |
#update_idx(idx) ⇒ Object
270 271 272 273 274 |
# File 'lib/cnvrg/dataset.rb', line 270 def update_idx(idx) File.open("#{self.local_path}/.cnvrg/idx.yml", 'w') { |f| f.write idx.to_yaml } return true end |
#update_idx_with_commit!(commit) ⇒ Object
277 278 279 280 281 282 283 |
# File 'lib/cnvrg/dataset.rb', line 277 def update_idx_with_commit!(commit) idx_hash = YAML.load_file("#{self.local_path}/.cnvrg/idx.yml") idx_hash[:commit] = commit File.open("#{self.local_path}/.cnvrg/idx.yml", 'w') { |f| f.write idx_hash.to_yaml } return true end |
#update_idx_with_files_commits!(files, commit_time) ⇒ Object
260 261 262 263 264 265 266 267 268 269 |
# File 'lib/cnvrg/dataset.rb', line 260 def update_idx_with_files_commits!(files, commit_time) idx_hash = YAML.load_file("#{self.local_path}/.cnvrg/idx.yml") files.each do |path| idx_hash[:tree].to_h[path].to_h[:commit_time] = commit_time end File.open("#{self.local_path}/.cnvrg/idx.yml", 'w') { |f| f.write idx_hash.to_yaml } return true end |
#update_ignore_list(new_ignore) ⇒ Object
51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
# File 'lib/cnvrg/dataset.rb', line 51 def update_ignore_list(new_ignore) if new_ignore.nil? or new_ignore.empty? return true end begin File.open(self.local_path+"/.cnvrgignore", "a+") do |f| f.puts("\n") new_ignore.each do |i| f.puts("#{i}\n") end end return true rescue return false end end |
#url ⇒ Object
46 47 48 49 |
# File 'lib/cnvrg/dataset.rb', line 46 def url url = Cnvrg::Helpers.remote_url "#{url}/#{self.owner}/projects/#{self.slug}" end |