Module: Helper

Included in:
Challenge::Solution
Defined in:
lib/helper.rb

Instance Method Summary collapse

Instance Method Details

#export_data(file_name, deliverables) ⇒ Object 



12
13
14
15
16
17
18
 
# File 'lib/helper.rb', line 12

def export_data(file_name, deliverables)		
	CSV.open(file_name, "wb", headers: false) do |row|
		deliverables.each do |deliverable|
			row << [deliverable.delivery_id, deliverable.is_deliverable, deliverable.partner_id, deliverable.delivery_cost]
		end
	end
end

#import_capacities_data(file_name) ⇒ Object 



20
21
22
23
24
25
26
27
 
# File 'lib/helper.rb', line 20

def import_capacities_data(file_name)
	capacities_data = read_data(file_name)
	capacities = {}
	capacities_data.each do |capacity_data|
		capacities[capacity_data[0].strip] = capacity_data[1].strip.to_i
	end
	capacities
end

#import_partners_data(file_name) ⇒ Object 



29
30
31
32
33
34
35
36
37
 
# File 'lib/helper.rb', line 29

def import_partners_data(file_name)
	partners_data = read_data(file_name)
	partners = []
	partners_data.each do |partner_data|
		slabs = partner_data[1].strip.split('-')
		partners << Partner.new(partner_data[0].strip, slabs.first.to_i, slabs.last.to_i, partner_data[2].strip.to_i, partner_data[3].strip.to_i, partner_data[4].strip)
	end
	partners
end

#import_theatres_data(file_name) ⇒ Object 



39
40
41
42
43
44
45
46
 
# File 'lib/helper.rb', line 39

def import_theatres_data(file_name)
	theatres_data = read_data(file_name, false)
	theatres = []
	theatres_data.each do |theatre_data|
		theatres << Theatre.new(theatre_data[0].strip, theatre_data[1].strip.to_i, theatre_data[2].strip )
	end
	theatres
end

#read_data(file_name, have_headers = true) ⇒ Object 



6
7
8
9
10
 
# File 'lib/helper.rb', line 6

def read_data(file_name, have_headers=true)
	csv_text = File.read(file_name)
	csv_data = CSV.parse(csv_text, headers: have_headers)
	csv_data
end