Class: DriveTime::Loader

Inherits:
Object
  • Object
show all
Defined in:
lib/drive_time/loader.rb

Overview

Load a Spreadsheet from Google Drive

Defined Under Namespace

Classes: SpreadsheetNotFoundError, WorksheetNotFoundError

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Loader

Returns a new instance of Loader.



9
10
11
 
# File 'lib/drive_time/loader.rb', line 9

def initialize
  begin_session
end

Instance Method Details

#load_file_direct(title) ⇒ Object 



13
14
15
 
# File 'lib/drive_time/loader.rb', line 13

def load_file_direct(title)
  return @session.file_by_title title
end

#load_spreadsheet(title, use_cache = true) ⇒ Object 



21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
 
# File 'lib/drive_time/loader.rb', line 21

def load_spreadsheet(title, use_cache=true)
  cached_directory = ENV['CACHED_DIR']
  spreadsheet_name = "#{title}.yml"
  spreadsheet_file_path = File.join(cached_directory, spreadsheet_name) if cached_directory
  spreadsheet = nil
  # Try and pull the file from the cache
  if cached_directory && use_cache

    if File.exist? spreadsheet_file_path
      File.open(spreadsheet_file_path, 'r') do |file|
        Logger.info "Pulling spreadsheet '#{title}' from cache"
        spreadsheet = YAML::load(file)
      end
    end
  end

  # If we haven't loaded a spreadsheet from cache, get it from drive
  unless spreadsheet
    Logger.info "Loading spreadsheet '#{title}' from Drive"
    spreadsheet = @session.spreadsheet_by_title(title)

    raise SpreadsheetNotFoundError, "Spreadsheet #{title} not found" if spreadsheet.nil?
    # Save the file to cache directory
    if cached_directory && use_cache
      # Save the spreadsheet
      Logger.info "Saving spreadsheet '#{title}' to cache"
      File.open(spreadsheet_file_path, 'w') do |file|
        file.puts YAML::dump(spreadsheet)
      end

      # Save its worksheets
      spreadsheet.worksheets.each do |worksheet|
        Logger.info "Saving worksheet '#{worksheet.title}' to cache"
        # Force worksheet to down of cells data
        #worksheet.reload
        worksheet_file_path = File.join(cached_directory, worksheet.title) + '.yml'
        File.open(worksheet_file_path, 'w') do |file|
          file.puts YAML::dump(worksheet)
        end
      end
    end
  end

  return spreadsheet
end

#load_spreadsheet_direct(title) ⇒ Object 



17
18
19
 
# File 'lib/drive_time/loader.rb', line 17

def load_spreadsheet_direct(title)
  @session.spreadsheet_by_title title
end

#load_worksheet_from_spreadsheet(spreadsheet, title, use_cache = true) ⇒ Object 



67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
 
# File 'lib/drive_time/loader.rb', line 67

def load_worksheet_from_spreadsheet(spreadsheet, title, use_cache=true)
  cached_directory = ENV['CACHED_DIR']
  worksheet_name = "#{title}.yml"

  worksheet = nil

  # Get the worksheet from the cache
  if cached_directory && use_cache
    worksheet_file_path = File.join(cached_directory, title)
    if File.exist? worksheet_file_path
      File.open(worksheet_file_path, 'r') do |file|
        Logger.info "Pulling worksheet '#{title}' from cache"
        worksheet = YAML::load(file)
      end
    end
  end

  # If we don't yet have a worksheet, pull it from Google Drive via the Spreadsheet
  unless worksheet
    worksheet = spreadsheet.worksheet_by_title title
    raise WorksheetNotFoundError, "Worksheet '#{title}'' not found in spreadsheet '#{spreadsheet.title}'" if worksheet.nil?
  end

  return worksheet
end