Class: DataHandler

Inherits:
Object
  • Object
show all
Defined in:
lib/watir/datahandler.rb

Instance Method Summary collapse

Constructor Details

#initialize(datafile) ⇒ DataHandler

This allows you to put data in a csv file or an xls file and access it in your scripts for a data driven approach.

You can supply a file with extension csv or xls and it will just work

e.g. require ‘datahandler’ d = DataHandler.new(“data.csv”) loginId = d.data.LoginId –> line 1 of the csv file and column LoginId pin = d.data.Pin –> line 1 of the cvs file and column Pin

b = Datahandler.new(“data.xls”) loginId = b.data.LoginId –> line 1 of workbook 1 of the xls file with column LoginId

Remember that line 1 of the data is referenced by using 0 - and this excludes the header line.



29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
 
# File 'lib/watir/datahandler.rb', line 29

def initialize(datafile)
  
  # if the data files are in the same dir as the one you execute from this works

  directory = `dir`
  directory.match(/Directory of(.+)$/)
  working_directory = $1.to_s.gsub(/\\/,"/").gsub(/^\s/,"")
  file = "#{working_directory.chomp}\\#{datafile}"
  
  # otherwise it will take the location you specify

  if File.exists?(file)
    @datafile = file
  else
    @datafile = datafile
  end
  
  @excel = WIN32OLE::new('excel.Application')
  @workbook = @excel.Workbooks.Open(@datafile)
  @worksheet = @workbook.Worksheets(1)
  @excel['Visible'] = false
  
end

Instance Method Details

#csv_data ⇒ Object 



51
52
53
54
55
56
57
58
59
60
61
62
63
 
# File 'lib/watir/datahandler.rb', line 51

def csv_data
  csv_data = File.readlines(@datafile)
  header = csv_data[0].split(",")
  header.each{|z| z.chomp!}
  data = Struct.new("Data", *header)
  csv_data.shift
  data_array = []
  csv_data.each do |line|
    line.each{|z| line.chomp!}
    data_array << data.new(*line.split(","))
  end
return data_array
end

#data ⇒ Object 



98
99
100
101
102
103
104
105
 
# File 'lib/watir/datahandler.rb', line 98

def data
  extension = @datafile.split(".")[-1]
  if extension == "xls"
    xls_data
  elsif extension == "csv"
    csv_data
  end
end

#number_columns ⇒ Object 



65
66
67
68
69
70
71
72
73
74
75
 
# File 'lib/watir/datahandler.rb', line 65

def number_columns
  column = last_column = "a"
  while @worksheet.Range("a1:#{column}1")['Value']
    if !@worksheet.Range("#{column}1")['Value']
      last_column = column.pred
      break
    end
    column.succ!
  end
last_column
end

#xls_data ⇒ Object 



77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
 
# File 'lib/watir/datahandler.rb', line 77

def xls_data
  line = "1"
  data = []
    while @worksheet.Range("a#{line}")['Value']
       data << @worksheet.Range("a#{line}:#{number_columns}#{line}")['Value']
       line.succ!
    end
  
  header = data[0].flatten
  dataObject = Struct.new("DataObject", *header)
  data.shift
  data_array = []
  data.each do |line|
    data_array << dataObject.new(*line.flatten)
  end
@excel.Quit
@excel = nil
GC.start
data_array
end